invert mask and UI update

app.py

@@ -97,6 +97,7 @@ with block:
                        [input_image, prompt_image, point_coords, point_labels, point_label_radio, text, pred_masks],
                        [prompt_image, lbl_image, point_coords, point_labels, pred_masks], queue=False)
     
+    
     def on_everything_image_select(input_image, pred_masks, masks, text, evt: gr.SelectData):
         i = evt.index
         mask = pred_masks[i][0]
@@ -109,6 +110,16 @@ with block:
     lbl_image.select(on_everything_image_select,
                        [input_image, pred_masks, masks, text],
                        [masks, selected_masks_image], queue=False)
+    
+    def on_selected_masks_image_select(input_image, masks, evt: gr.SelectData):
+        i = evt.index
+        del masks[i]
+        anno = (input_image, masks) 
+        return  [masks, anno]
+    
+    selected_masks_image.select(on_selected_masks_image_select,
+                                [input_image, masks],
+                                [masks, selected_masks_image], queue=False)
     #prompt_lbl_image.select(on_everything_image_select,
     #                   [input_image, prompt_masks, masks, text],
     #                   [masks, selected_masks_image], queue=False)

inference.py

@@ -64,7 +64,7 @@ class DepthPredictor:
         formatted = 255 - (output * 255 / np.max(output)).astype('uint8')
 
         #img = Image.fromarray(formatted)
-        return output
+        return 1 - output
     
     def generate_pcl(self, image):
         print(np.array(image).shape)