UI update + removed "raw_image"

app.py

@@ -23,7 +23,6 @@ with block:
         return []
     def point_labels_empty():
         return []
-    raw_image = gr.Image(type='pil', visible=False)
     image_edit_trigger = gr.State(True)
     point_coords = gr.State(point_coords_empty)
     point_labels = gr.State(point_labels_empty)
@@ -38,10 +37,12 @@ with block:
                 sam_encode_btn = gr.Button('Encode', variant='primary')
                 sam_encode_status = gr.Label('Not encoded yet')
             prompt_image = gr.Image(label='Segments')
+        with gr.Row():
             with gr.Column():
                 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')
-                cube_size = gr.Slider(minimum=0.000001, maximum=0.001, step=0.000001, default=0.00001, label='Cube size')
+                with gr.Row():
+                    n_samples = gr.Slider(minimum=1e3, maximum=1e6, step=1e3, default=1e3, label='Number of Samples')
+                    cube_size = gr.Slider(minimum=0.000001, maximum=0.001, step=0.000001, default=0.00001, label='Cube size')
         with gr.Row():
             with gr.Column(scale=1):
                 with gr.Row():
@@ -52,14 +53,14 @@ with block:
                 sam_decode_btn = gr.Button('Predict using points!', variant = 'primary')
                 depth_reconstruction_btn = gr.Button('Depth Reconstruction', variant = 'primary')
     # components
-    components = {point_coords, point_labels, raw_image, image_edit_trigger, masks, cutout_idx, input_image,
+    components = {point_coords, point_labels, image_edit_trigger, masks, cutout_idx, input_image,
                   point_label_radio, text, reset_btn, sam_sgmt_everything_btn,
                   sam_decode_btn, depth_reconstruction_btn, prompt_image, n_samples, cube_size}
     
     # event - init coords
-    def on_reset_btn_click(raw_image):
-        return raw_image, point_coords_empty(), point_labels_empty(), None, []
-    reset_btn.click(on_reset_btn_click, [raw_image], [input_image, point_coords, point_labels], queue=False)
+    def on_reset_btn_click(input_image):
+        return input_image, point_coords_empty(), point_labels_empty(), None, []
+    reset_btn.click(on_reset_btn_click, [input_image], [input_image, point_coords, point_labels], queue=False)
 
     def on_input_image_upload(input_image):
         print("encoding")
@@ -67,7 +68,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.upload(on_input_image_upload, [input_image], [input_image, point_coords, point_labels], queue=False)
 
     # event - set coords
     def on_prompt_image_select(input_image, point_coords, point_labels, point_label_radio, evt: gr.SelectData):
@@ -100,7 +101,7 @@ with block:
 
     def on_click_sam_dencode_btn(inputs):
         print("inferencing")
-        image = inputs[raw_image]
+        image = inputs[input_image]
         generated_mask, _, _ = sam.cond_pred(pts=np.array(inputs[point_coords]), lbls=np.array(inputs[point_labels]))
         inputs[masks].append((generated_mask, inputs[text]))
         print(inputs[masks][0])
@@ -109,7 +110,7 @@ with block:
 
     def on_depth_reconstruction_btn_click(inputs):
         print("depth reconstruction")
-        image = inputs[raw_image]
+        image = inputs[input_image]
         path = dpt.generate_obj_masks(image=image, n_samples=inputs[n_samples], cube_size=inputs[cube_size], masks=inputs[masks])
         return {pcl_figure: path}
     depth_reconstruction_btn.click(on_depth_reconstruction_btn_click, components, [pcl_figure], queue=False)