Update app.py

app.py

@@ -49,18 +49,14 @@ def load_mask_ui(input_folder="example_tmp",load_edit = False):
     return mask_np_list, mask_label_list
 
 def load_image_ui(load_edit, input_folder="example_tmp"):
-    #try:
-    if 1:
-        image, mask_np_list, mask_label_list = load_mask_ui(input_folder, load_edit = load_edit)
-        #image = image.convert('RGB')
-        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)
-        return image, segmentation, mask_np_list, mask_label_list, image, sliderup, sliderup 
-    #except:
-    #    print("Image folder invalid: The folder should contain image.png")
-    #    return None, None, None, None, None, None
+
+    image, mask_np_list, mask_label_list = load_mask_ui(input_folder, load_edit = load_edit)
+    #image = image.convert('RGB')
+    segmentation = create_segmentation(mask_np_list) 
+    max_val = len(mask_np_list)-1
+    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, sliderup 
+
 
 
 def run_segmentation_wrapper(image):
@@ -157,15 +153,6 @@ def save_as_edit_mask(mask_np_list_updated, mask_label_list, input_folder="examp
     visualize_mask_list_clean(mask_np_list_updated, savepath)
   
 
-def image_change():
-    directory_path = "./example_tmp/"
-    for filename in os.listdir(directory_path):
-        file_path = os.path.join(directory_path, filename)
-        if os.path.isfile(file_path) or os.path.islink(file_path):
-            os.unlink(file_path)
-        elif os.path.isdir(file_path):
-            shutil.rmtree(file_path)
-    return gr.Button("1.2 Load original masks",visible = False)
 
 def button_clickable(is_clickable):
     return gr.Button(interactive=is_clickable)
@@ -181,9 +168,16 @@ if os.path.isdir("./example_tmp"):
     shutil.rmtree("./example_tmp")
 
 
+
+css = """
+#response { 
+    font-size: 25px !important; 
+}
+"""
+
 from segment import run_segmentation
 
-with gr.Blocks() as demo:
+with gr.Blocks(css=css) as demo:
     image = gr.State() # store mask
     image_loaded = gr.State()
     segmentation    = gr.State()
@@ -202,8 +196,8 @@ with gr.Blocks() as demo:
         with gr.Row():
             with gr.Column():
                 canvas = gr.Image(value = "./img.png", type="numpy",  label="Draw Mask", show_label=True, height=LENGTH, width=LENGTH, interactive=True)
-                
-                segment_button  = gr.Button("1.1 Run segmentation")
+                result_info0 = gr.Text(label="Response", elem_id="response")
+                segment_button  = gr.Button("Run segmentation")
                 
 
 
@@ -212,7 +206,6 @@ with gr.Blocks() as demo:
             # mask_np_list_updated.value = copy.deepcopy(mask_np_list.value) #!!
             mask_np_list_updated = mask_np_list
             with gr.Column():
-                result_info0 = gr.Text(label="Response")
                 gr.Markdown("""<p style="text-align: center; font-size: 20px">Edit Mask (Optional)</p>""")
                 slider =  gr.Slider(0, 20, step=1, label = 'mask id',  interactive=False)
                 label = gr.Text(label='label')
@@ -229,7 +222,7 @@ with gr.Blocks() as demo:
     with gr.Tab(label="2 Optimization"):
         with gr.Row():
             with gr.Column():
-                result_info = gr.Text(label="Response")
+                result_info = gr.Text(label="Response", elem_id="response")
                 
                 opt_flag = gr.State(0)
                 gr.Markdown("""<p style="text-align: center; font-size: 20px">Optimization settings (SD)</p>""")
@@ -274,13 +267,13 @@ with gr.Blocks() as demo:
                         )
                         run_optimization()
                         print('finish')
-                        return "Optimization finished!"
+                        return "Optimization finished! Move to the next step."
                     except:
                         return "CUDA out of memory, use a smaller batch size or try another picture."
 
                 
                 def immediate_update():
-                    return gr.Button("Run Optimization (Check Log for Completion).", interactive=True)
+                    return gr.Button("Run Optimization (Check log for progress.).", interactive=True)
                 
                 add_button.click(run_optimization_wrapper, 
                         inputs = [
@@ -300,7 +293,7 @@ with gr.Blocks() as demo:
                 add_button.click(fn=immediate_update, inputs=[], outputs=[add_button])
                 #add_button.update()
                 def change_text():
-                    return gr.Textbox("Optimization Finished!", interactive = False)
+                    return gr.Textbox("Optimization Finished! Move to the next step.", interactive = False)
 
 
     with gr.Tab(label="3 Editing"):
@@ -322,7 +315,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 (Check Log for Completion)")
+                    add_button  = gr.Button("Run Editing (Check log for progress.)")
                     def run_edit_text_wrapper(
                             mask_np_list,
                             mask_label_list,