Update app.py

delete other code

app.py

@@ -58,68 +58,12 @@ def load_image_ui(load_edit, input_folder="example_tmp"):
         segmentation = create_segmentation(mask_np_list)
         print("!!", len(mask_np_list))  
         max_val = len(mask_np_list)-1
-        sliderup = gr.Slider(value = 0, minimum=0, maximum=max_val, step=1,  interactive=True)
+        sliderup = gr.Slider(value = 0, minimum=0, maximum=max_val, step=1, interactive=True)
         return image, segmentation, mask_np_list, mask_label_list, image, sliderup
     except:
         print("Image folder invalid: The folder should contain image.png")
-        return None, None, None, None, None
-
-# def run_edit_text(
-#         num_tokens,
-#         num_sampling_steps,
-#         strength,
-#         edge_thickness,
-#         tgt_prompt,
-#         tgt_idx,
-#         guidance_scale,
-#         input_folder="example_tmp"
-#     ):
-#     subprocess.run(["python", 
-#                     "main.py" ,
-#                     "--text=True",
-#                     "--name={}".format(input_folder),
-#                     "--dpm={}".format("sd"),
-#                     "--resolution={}".format(512),
-#                     "--load_trained",
-#                     "--num_tokens={}".format(num_tokens),
-#                     "--seed={}".format(2024),
-#                     "--guidance_scale={}".format(guidance_scale),
-#                     "--num_sampling_step={}".format(num_sampling_steps),
-#                     "--strength={}".format(strength),
-#                     "--edge_thickness={}".format(edge_thickness),
-#                     "--num_imgs={}".format(2),
-#                     "--tgt_prompt={}".format(tgt_prompt) ,
-#                     "--tgt_index={}".format(tgt_idx)            
-#     ])
-    
-#     return Image.open(os.path.join(input_folder, "text", "out_text_0.png"))
-
-
-# def run_optimization(
-#         num_tokens,
-#         embedding_learning_rate, 
-#         max_emb_train_steps, 
-#         diffusion_model_learning_rate, 
-#         max_diffusion_train_steps,
-#         train_batch_size,
-#         gradient_accumulation_steps,
-#         input_folder = "example_tmp"
-#     ):
-#     subprocess.run(["python", 
-#                     "main.py" ,
-#                     "--name={}".format(input_folder),
-#                     "--dpm={}".format("sd"),
-#                     "--resolution={}".format(512),
-#                     "--num_tokens={}".format(num_tokens),
-#                     "--embedding_learning_rate={}".format(embedding_learning_rate),
-#                     "--diffusion_model_learning_rate={}".format(diffusion_model_learning_rate),
-#                     "--max_emb_train_steps={}".format(max_emb_train_steps),
-#                     "--max_diffusion_train_steps={}".format(max_diffusion_train_steps),
-#                     "--train_batch_size={}".format(train_batch_size),
-#                     "--gradient_accumulation_steps={}".format(gradient_accumulation_steps)
-                    
-#     ])
-#     return 
+        return None, None, None, None, None, None
+
 
 
 def transparent_paste_with_mask(backimg, foreimg, mask_np,transparency = 128):
@@ -255,63 +199,26 @@ 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)    
-                
-                # 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])
-                #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 ])
-                
-                
+
             # mask_np_list_updated.value = copy.deepcopy(mask_np_list.value) #!!
             mask_np_list_updated = mask_np_list
             with gr.Column():
                 gr.Markdown("""<p style="text-align: center; font-size: 20px">Edit Mask (Optional)</p>""")
-                slider =  gr.Slider(0, 20, step=1,  interactive=False)
+                slider =  gr.Slider(0, 20, step=1, label = 'mask id',  interactive=False)
                 label = gr.Textbox()
                 slider.release(slider_release, 
                         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] , 
-                #         [] )  
-                
-                # 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] )
-
-                # 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] )
         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] )
+
 
         canvas.upload(image_change, inputs=[], outputs=[text_button])
                 
@@ -382,12 +289,7 @@ with gr.Blocks() as demo:
                 #add_button.update()
                 def change_text():
                     return gr.Textbox("Optimization Finished!", interactive = False)
-                '''txt_box.change(fn=lambda x: gr.Button.update(value="Optimization Finished!", interactive=True), 
-                       inputs=[txt_box], outputs=[add_button])
-                def change_text2():
-                    return gr.Textbox("Start optimization, check logs for progress...", interactive = False)
-                add_button.click(change_text2, [], txt_box)'''
-                #opt_flag.change(change_text, txt_box, txt_box)
+
 
     with gr.Tab(label="3 Editing"):
         with gr.Tab(label="3.1 Text-based editing"):
@@ -446,11 +348,7 @@ with gr.Blocks() as demo:
                         outputs = [canvas_text_edit],queue=True,
                     )
                     
-                    load_button  = gr.Button("Load results")
-                    load_button.click(load_pil_img, 
-                        inputs = [], 
-                        outputs = [canvas_text_edit]
-                    )
+