fixed stream

app.py

@@ -36,8 +36,6 @@ with block:
             masks_annotated_image = gr.AnnotatedImage(label='Segments', height=512)
             pcl_figure = gr.Model3D(label="3-D Reconstruction", clear_color=[1.0, 1.0, 1.0, 1.0])
             n_samples = gr.Slider(minimum=1e3, maximum=1e6, step=1e3, default=1e3, label='Number of Samples')
-            #gr.Plot(label='3D Reconstruction')
-            #cutout_galary = gr.Gallery(label='Cutouts', object_fit='contain', height=512)  
         with gr.Row():
             with gr.Column(scale=1):
                 with gr.Row():
@@ -51,16 +49,6 @@ with block:
     components = {point_coords, point_labels, raw_image, masks, cutout_idx, input_image,
                   point_label_radio, text, reset_btn, sam_sgmt_everything_btn,
                   sam_decode_btn, depth_reconstruction_btn, masks_annotated_image, n_samples}
-    def on_depth_reconstruction_btn_click(inputs):
-        print("depth reconstruction")
-        image = inputs[input_image]
-        # depth reconstruction
-        #mask = inputs[masks][0][0]
-        path = dpt.generate_obj(image, inputs[n_samples])
-
-        return {pcl_figure: path}
-    
-    depth_reconstruction_btn.click(on_depth_reconstruction_btn_click, components, [pcl_figure], queue=False)
     
     # event - init coords
     def on_reset_btn_click(raw_image):
@@ -73,7 +61,7 @@ with block:
         sam.encode(input_image)
         print("encoding done")
         return input_image, point_coords_empty(), point_labels_empty(), None
-    input_image.upload(on_input_image_upload, [input_image], [raw_image, point_coords, point_labels], queue=False)
+    input_image.stream(on_input_image_upload, [input_image], [raw_image, point_coords, point_labels], queue=False)
 
     # event - set coords
     def on_input_image_select(input_image, point_coords, point_labels, point_label_radio, evt: gr.SelectData):
@@ -87,7 +75,6 @@ with block:
         return img, point_coords, point_labels
     input_image.select(on_input_image_select, [input_image, point_coords, point_labels, point_label_radio], [input_image, point_coords, point_labels], queue=False)
 
-
     def on_click_sam_dencode_btn(inputs):
         print("inferencing")
         image = inputs[raw_image]
@@ -96,6 +83,13 @@ with block:
         return {masks_annotated_image: (image, inputs[masks])}
     sam_decode_btn.click(on_click_sam_dencode_btn, components, [masks_annotated_image, masks, cutout_idx], queue=True)
 
+    def on_depth_reconstruction_btn_click(inputs):
+        print("depth reconstruction")
+        image = inputs[raw_image]
+        path = dpt.generate_obj(image, inputs[n_samples])
+        return {pcl_figure: path}
+    depth_reconstruction_btn.click(on_depth_reconstruction_btn_click, components, [pcl_figure], queue=False)
+
 
 if __name__ == '__main__':
     block.queue()