Update app.py

app.py

@@ -216,6 +216,17 @@ def retrieval_text(text, k, sim_th, filter_fn):
     
     prog.progress(1.0, "Idle")
 
+def generation_img(load_data, prompt, noise_scale, cfg_scale, steps):
+    pc = load_data(prog)
+    prog.progress(0.49, "Running Generation")
+    col2 = utils.render_pc(pc)
+    if torch.cuda.is_available():
+        with sys.clip_move_lock:
+            clip_model.cpu()
+
+    width = 640
+    height = 640
+
 try:
     f32 = numpy.float32
     half = torch.float16 if torch.cuda.is_available() else torch.bfloat16
@@ -279,12 +290,16 @@ try:
             'Choose the mode of generation', 
             ("PointCloud-to-Image", "PointCloud-to-Text")
         )
-        pc = st.sidebar.text_input("Input pc", key='rtextinput')
+        load_data = utils.input_3d_shape('rpcinput')
         if generation_mode == "PointCloud-to-Image":
             st.title("Image Generation")
             prog = st.progress(0.0, "Idle")
+            prompt = st.sidebar.text_input("Prompt (Optional)", key='sdtprompt')
+            noise_scale = st.sidebar.slider('Variation Level', 0, 5, 1)
+            cfg_scale = st.sidebar.slider('Guidance Scale', 0.0, 30.0, 10.0)
+            steps = st.sidebar.slider('Diffusion Steps', 8, 50, 25)
             if st.sidebar.button("submit"):
-                pc = st.text_input("Input pc", key='rtextinput')
+                generation_img(load_data, prompt, noise_scale, cfg_scale, steps)
         elif generation_mode == "PointCloud-to-Text":
             st.title("Text Generation")
             prog = st.progress(0.0, "Idle")