segment everything!

app.py

@@ -39,9 +39,7 @@ with block:
         with gr.Row():
             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')
+            everything_image = gr.AnnotatedImage(label='Everything')
             
         with gr.Row():
             with gr.Column():
@@ -122,6 +120,13 @@ with block:
         return {pcl_figure: path}
     depth_reconstruction_btn.click(on_depth_reconstruction_btn_click, components, [pcl_figure], queue=False)
 
+    def on_sam_sgmt_everything_btn_click(inputs):
+        print("segmenting everything")
+        image = inputs[input_image]
+        annotated_image = sam.segment_everything(image)
+        return [annotated_image]
+    sam_sgmt_everything_btn.click(on_sam_sgmt_everything_btn_click, components, [everything_image], queue=False)
+
 
 if __name__ == '__main__':
     block.queue()