SAM encoding

app.py

@@ -33,6 +33,9 @@ with block:
     with gr.Column():
         with gr.Row():
             input_image = gr.Image(label='Input', type='pil', source='webcam', tool=None)
+            with gr.Column():
+                sam_encode_btn = gr.Button('Encode', variant='primary')
+                sam_encode_status = gr.Label('Not encoded yet')
             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])
             with gr.Column():
@@ -77,6 +80,15 @@ 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_encode_btn(inputs):
+        print("encoding")
+        # encode image on click
+        sam.encode(inputs[input_image])
+        print("encoding done")
+        return {sam_encode_status: 'Image Encoded!'}
+    
+    depth_reconstruction_btn.click(on_click_sam_encode_btn, components, [pcl_figure], queue=False)
+
     def on_click_sam_dencode_btn(inputs):
         print("inferencing")
         image = inputs[raw_image]