dont clear selected points until `Clear points` button is pressed or input image is changed

app_gradio.py

@@ -159,9 +159,7 @@ def segment_with_points(
                        use_retina=use_retina,
                        withContours=withContours,)
 
-    global_points = []
-    global_point_label = []
-    return fig, None
+    return fig
 
 
 def get_points_with_draw(image, label, evt: gr.SelectData):
@@ -297,7 +295,7 @@ with gr.Blocks(css=css, title='Fast Segment Anything') as demo:
 
     segment_btn_p.click(segment_with_points,
                         inputs=[cond_img_p],
-                        outputs=[segm_img_p, cond_img_p])
+                        outputs=[segm_img_p])
 
     with gr.Tab("Text mode"):
         # Images
@@ -361,11 +359,20 @@ with gr.Blocks(css=css, title='Fast Segment Anything') as demo:
                         outputs=segm_img_t)
 
     def clear():
+        global global_points
+        global global_point_label
+        global_points = []
+        global_point_label = []
         return None, None
-    
+
     def clear_text():
         return None, None, None
 
+    everything_tab.select(clear, outputs=[cond_img_e, segm_img_e]) # reset when everything tab is selected
+    points_tab.select(clear, outputs=[cond_img_e, segm_img_e]) # reset when points tab is selected
+    text_tab.select(clear, outputs=[cond_img_e, segm_img_e]) # reset when text tab is selected
+    cond_img_p.clear(clear, outputs=[cond_img_e, segm_img_e]) # reset when input image is cleared
+    cond_img_p.input(clear, outputs=[cond_img_e, segm_img_e]) # reset when input image is changed
     clear_btn_e.click(clear, outputs=[cond_img_e, segm_img_e])
     clear_btn_p.click(clear, outputs=[cond_img_p, segm_img_p])
     clear_btn_t.click(clear_text, outputs=[cond_img_p, segm_img_p, text_box])