UI flow update

app.py

@@ -85,23 +85,25 @@ with block:
         img = Image.fromarray(img)
         point_coords.append([x,y])
         point_labels.append(point_label_radio)
-
-        return  [unedited_image,
-                 img,
-                 point_coords,
-                 point_labels,
-                 image_edit_trigger]
+        generated_mask, _, _ = sam.cond_pred(pts=np.array(point_coords), lbls=np.array(point_labels))
+        return  [(img, [(generated_mask, text)]),
+                  unedited_image,
+                  img,
+                  point_coords,
+                  point_labels,
+                  image_edit_trigger]
     
     masks_annotated_image.select(on_input_image_select,
                        [raw_image, input_image, image_edit_trigger, point_coords, point_labels, point_label_radio],
-                       [raw_image, input_image, point_coords, point_labels, image_edit_trigger], queue=False)
+                       [masks_annotated_image, raw_image, input_image, point_coords, point_labels, image_edit_trigger], 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!'}
+        return {sam_encode_status: 'Image Encoded!',
+                masks_annotated_image: (inputs[input_image], [])}
     sam_encode_btn.click(on_click_sam_encode_btn, components, [sam_encode_status], queue=False)
 
     def on_click_sam_dencode_btn(inputs):