fix

app.py

@@ -37,12 +37,9 @@ with block:
                 sam_encode_btn = gr.Button('Encode', variant='primary')
                 sam_encode_status = gr.Label('Not encoded yet')
         with gr.Row():
-            with gr.Tab("Select with points"):
-                with gr.Column():
-                    prompt_image = gr.Image(label='Segments')
-                    prompt_lbl_image = gr.AnnotatedImage(label='Segment Labels')
+            prompt_image = gr.Image(label='Segments')
+            prompt_lbl_image = gr.AnnotatedImage(label='Segment Labels')
                     
-                
             with gr.Tab("Select from segmented map"):
                 everything_image = gr.AnnotatedImage(label='Everything')
             
@@ -64,7 +61,7 @@ with block:
     # components
     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}
+                  sam_decode_btn, depth_reconstruction_btn, prompt_image, prompt_lbl_image, n_samples, cube_size}
     
     # event - init coords
     def on_reset_btn_click(input_image):
@@ -83,15 +80,15 @@ with block:
     def on_prompt_image_select(input_image, prompt_image, point_coords, point_labels, point_label_radio, evt: gr.SelectData):
         x, y = evt.index
         color = red if point_label_radio == 0 else blue
-        if prompt is None:
-            prompt = np.array(input_image.copy())
+        if prompt_image is None:
+            prompt_image = np.array(input_image.copy())
 
-        cv2.circle(prompt, (x, y), 5, color, -1)
+        cv2.circle(prompt_image, (x, y), 5, color, -1)
         point_coords.append([x,y])
         point_labels.append(point_label_radio)
         generated_mask, _, _ = sam.cond_pred(pts=np.array(point_coords), lbls=np.array(point_labels))
 
-        return  [ prompt,
+        return  [ prompt_image,
                   (input_image, [(generated_mask, "Mask")]),
                   point_coords,
                   point_labels ]