UI update

app.py

@@ -37,28 +37,31 @@ with block:
             with gr.Column():
                 input_image = gr.Image(label='Input', type='pil', tool=None) # mirror_webcam = False
                 sam_encode_btn = gr.Button('Encode', variant='primary')
-                sam_encode_status = gr.Label('Not encoded yet')
+                #sam_encode_status = gr.Label('Not encoded yet')
         with gr.Row():
-            prompt_image = gr.Image(label='Segments')
+            with gr.Column():
+                prompt_image = gr.Image(label='Segments')
+                with gr.Row():  
+                    point_label_radio = gr.Radio(label='Point Label', choices=[1,0], value=1)
+                    text = gr.Textbox(label='Mask Name')
+                    reset_btn = gr.Button('New Mask')
             prompt_lbl_image = gr.AnnotatedImage(label='Segment Labels')
             everything_image = gr.AnnotatedImage(label='Everything')
-            
+        selected_masks_image = gr.AnnotatedImage(label='Selected Masks')
+        sam_sgmt_everything_btn = gr.Button('Segment Everything!', variant = 'primary')
+
         with gr.Row():
             with gr.Column():
                 pcl_figure = gr.Model3D(label="3-D Reconstruction", clear_color=[1.0, 1.0, 1.0, 1.0])
                 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():
-            selected_masks_image = gr.AnnotatedImage(label='Selected Masks')
-            with gr.Column(scale=1):
-                with gr.Row():
-                    point_label_radio = gr.Radio(label='Point Label', choices=[1,0], value=1)
-                    text = gr.Textbox(label='Mask Name')
-                    reset_btn = gr.Button('New Mask')
-                sam_sgmt_everything_btn = gr.Button('Segment Everything!', variant = 'primary')
+                    depth_reconstruction_btn = gr.Button('Depth Reconstruction', variant = 'primary')
+       
+
+
                 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, image_edit_trigger, masks, cutout_idx, input_image,
                   point_label_radio, text, reset_btn, sam_sgmt_everything_btn,
@@ -112,9 +115,8 @@ with block:
         # encode image on click
         sam.encode(inputs[input_image])
         print("encoding done")
-        return {sam_encode_status: 'Image Encoded!',
-                prompt_image: inputs[input_image]}
-    sam_encode_btn.click(on_click_sam_encode_btn, components, [sam_encode_status, prompt_image], queue=False)
+        return {prompt_image: inputs[input_image]}
+    sam_encode_btn.click(on_click_sam_encode_btn, components, [prompt_image], queue=False)
 
     def on_click_sam_dencode_btn(inputs):
         print("inferencing")