delete buttons

app.py

@@ -221,6 +221,11 @@ def button_clickable(is_clickable):
     return gr.Button.update(interactive=is_clickable)
 
 
+
+def load_pil_img():
+    from PIL import Image
+    return Image.open("example_tmp/text/out_text_0.png")
+
 import shutil
 if os.path.isdir("./example_tmp"):
     shutil.rmtree("./example_tmp")
@@ -250,13 +255,13 @@ with gr.Blocks() as demo:
                 
                 text_button  = gr.Button("Waiting 1.1 to complete",visible = False)
 
-                load_edit_button = gr.Button("Waiting 1.1 to complete",visible = False)    
+                # load_edit_button = gr.Button("Waiting 1.1 to complete",visible = False)    
                 
-                show_segment = gr.Checkbox(label = "Waiting 1.1 to complete",visible = False)
+                # show_segment = gr.Checkbox(label = "Waiting 1.1 to complete",visible = False)
                 flag = gr.State(False)
-                show_segment.select(show_segmentation,
-                                    [image_loaded, segmentation, flag], 
-                                    [canvas, flag])
+                # show_segment.select(show_segmentation,
+                #                     [image_loaded, segmentation, flag], 
+                #                     [canvas, flag])
                 #def show_more_buttons():
                 #    return gr.Button("1.2 Load original masks",visible = True), gr.Button("1.2 Load edited masks")   , gr.Checkbox(label = "Show Segmentation") 
                 #block_flag.change(show_more_buttons, [], [text_button,load_edit_button,show_segment ])
@@ -272,43 +277,43 @@ with gr.Blocks() as demo:
                         inputs = [slider, image_loaded,   mask_np_list_updated, mask_label_list], 
                         outputs= [canvas, label]
                     )
-                add_button  = gr.Button("Add")
-                add_button.click( edit_mask_add, 
-                        [canvas, image_loaded, slider, mask_np_list_updated] , 
-                        [mask_np_list_updated, canvas]
-                    )
-
-                save_button2  = gr.Button("Set and Save as edited masks")
-                save_button2.click( save_as_edit_mask, 
-                        [mask_np_list_updated,  mask_label_list] , 
-                        [] )  
+                # add_button  = gr.Button("Add")
+                # add_button.click( edit_mask_add, 
+                #         [canvas, image_loaded, slider, mask_np_list_updated] , 
+                #         [mask_np_list_updated, canvas]
+                #     )
+
+                # save_button2  = gr.Button("Set and Save as edited masks")
+                # save_button2.click( save_as_edit_mask, 
+                #         [mask_np_list_updated,  mask_label_list] , 
+                #         [] )  
                 
-                save_button  = gr.Button("Set and Save as original masks")
-                save_button.click( save_as_orig_mask, 
-                        [mask_np_list_updated,  mask_label_list] , 
-                        [] )  
+                # save_button  = gr.Button("Set and Save as original masks")
+                # save_button.click( save_as_orig_mask, 
+                #         [mask_np_list_updated,  mask_label_list] , 
+                #         [] )  
                 
-                back_button  = gr.Button("Back to current seg")
-                back_button.click( load_mask_ui, 
-                                [] , 
-                                [ mask_np_list_updated,mask_label_list] )
+                # back_button  = gr.Button("Back to current seg")
+                # back_button.click( load_mask_ui, 
+                #                 [] , 
+                #                 [ mask_np_list_updated,mask_label_list] )
 
-                add_mask_button = gr.Button("Add new empty mask")    
-                add_mask_button.click(add_mask, 
-                        [mask_np_list_updated, mask_label_list] , 
-                        [mask_np_list_updated, mask_label_list] )
+                # add_mask_button = gr.Button("Add new empty mask")    
+                # add_mask_button.click(add_mask, 
+                #         [mask_np_list_updated, mask_label_list] , 
+                #         [mask_np_list_updated, mask_label_list] )
 
 
         segment_button.click(run_segmentation, 
                 [canvas] ,
-                [text_button,load_edit_button,show_segment] )
+                [text_button] )
         text_button.click(load_image_ui, [false] , 
                         [image_loaded, segmentation,  mask_np_list, mask_label_list, canvas, slider] )
                         
-        load_edit_button.click(load_image_ui, [ true] , 
-                        [image_loaded, segmentation,  mask_np_list, mask_label_list, canvas, slider] )
+        # load_edit_button.click(load_image_ui, [ true] , 
+        #                 [image_loaded, segmentation,  mask_np_list, mask_label_list, canvas, slider] )
 
-        canvas.upload(image_change, inputs=[], outputs=[text_button,load_edit_button,show_segment])
+        canvas.upload(image_change, inputs=[], outputs=[text_button])
                 
                 
     with gr.Tab(label="2 Optimization"):
@@ -391,7 +396,7 @@ with gr.Blocks() as demo:
         
             with gr.Row():
                 with gr.Column():
-                    canvas_text_edit = gr.Image(value = None, type = "pil", label="Editing results", show_label=True)
+                    canvas_text_edit = gr.Image(value = None, type = "pil", label="Editing results", show_label=True,visible = True)
                     # canvas_text_edit = gr.Gallery(label = "Edited results")
                 
                 with gr.Column():
@@ -404,7 +409,7 @@ with gr.Blocks() as demo:
                     edge_thickness = gr.Number(value="10", label="Editing: Edge thickness", interactive= True )
                     strength = gr.Textbox(value="0.5", label="Editing: Mask strength", interactive= True )
                     
-                    add_button  = gr.Button("Run Editing")
+                    add_button  = gr.Button("Run Editing (Check Log for Completion)")
                     def run_edit_text_wrapper(
                             num_tokens,
                             guidance_scale,
@@ -428,7 +433,8 @@ with gr.Blocks() as demo:
                             tgt_prompt = tgt_prompt,
                             tgt_index = int(tgt_index)
                         )
-                        return run_edit_text()
+                        run_edit_text()
+                        return 0
                         
                     add_button.click(run_edit_text_wrapper, 
                         inputs = [num_tokens_global,
@@ -439,12 +445,9 @@ with gr.Blocks() as demo:
                                     tgt_prompt ,
                                     tgt_index
                                 ],        
-                        outputs = [canvas_text_edit],queue=True,
+                        outputs = [],queue=True,
                     )
                     
-                    def load_pil_img():
-                        from PIL import Image
-                        return Image.open("example_tmp/text/out_text_0.png")
                     
                     load_button  = gr.Button("Load results")
                     load_button.click(load_pil_img, 

segment.py

@@ -117,4 +117,4 @@ def run_segmentation(image, name="example_tmp", size = 512, noseg=False):
     draw_panoptic_segmentation(**panoptic_segmentation, save_folder = save_folder, noseg = noseg, model = model)
     print("Finish segment")
     #block_flag += 1
-    return  gr.Button.update("1.2 Load original masks",visible = True), gr.Button.update("1.2 Load edited masks",visible = True), gr.Checkbox.update(label = "Show Segmentation",visible = True) 
+    return  gr.Button.update("1.2 Load original masks",visible = True)#, gr.Button.update("1.2 Load edited masks",visible = True), gr.Checkbox.update(label = "Show Segmentation",visible = True)