Update app.py

app.py

@@ -105,7 +105,7 @@ def slider_release(index, image,  mask_np_list_updated, mask_label_list):
         new_image = transparent_paste_with_mask(image, segmentation, mask_np, transparency = TRANSPARENCY)
     return new_image, mask_label
 def image_change():
-    return gr.Slider(value = 0, minimum=0, maximum=1, step=1, visible=False),gr.Button("Run Editing (Check log for progress.)",interactive = False)
+    return gr.Slider(value = 0, minimum=0, maximum=1, step=1, visible=False),gr.Button("Step 3. Run Editing (Check log for progress.)",interactive = False)
 
 def save_as_orig_mask(mask_np_list_updated, mask_label_list, input_folder="example_tmp"):
     print(mask_np_list_updated)
@@ -177,7 +177,7 @@ with gr.Blocks() as demo:
 
             with gr.Column():
                 result_info0 = gr.Text(label="Response")
-                segment_button  = gr.Button("Run segmentation")
+                segment_button  = gr.Button("Step 1. Run segmentation")
                 flag = gr.State(False)
 
                 # mask_np_list_updated.value = copy.deepcopy(mask_np_list.value) #!!
@@ -211,7 +211,7 @@ with gr.Blocks() as demo:
                     train_batch_size = gr.Number(value="20", label="Batch size", interactive= True )
                     gradient_accumulation_steps=gr.Number(value="2", label="Gradient accumulation", interactive= True )
                 
-                add_button  = gr.Button("Run optimization")
+                add_button  = gr.Button("Step 2. Run optimization")
                 def run_optimization_wrapper (
                         mask_np_list,
                         mask_label_list,
@@ -241,11 +241,11 @@ with gr.Blocks() as demo:
                         )
                         run_optimization()
                         gr.Info("Optimization Finished! Move to the next step.")
-                        return "Optimization finished! Move to the next step.",gr.Button("Run Editing (Check log for progress.)",interactive = True)
+                        return "Optimization finished! Move to the next step.",gr.Button("Step 3. Run Editing (Check log for progress.)",interactive = True)
                     except Exception as e:
                         print(e)
                         gr.Error("e")
-                        return "Error: use a smaller batch size or try latter.",gr.Button("Run Editing (Check log for progress.)",interactive = False)
+                        return "Error: use a smaller batch size or try latter.",gr.Button("Step 3. Run Editing (Check log for progress.)",interactive = False)
 
 
 
@@ -268,7 +268,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_button2  = gr.Button("Run Editing (Check log for progress.)",interactive = False)
+                    add_button2  = gr.Button("Step 3. Run Editing (Check log for progress.)",interactive = False)
                     def run_edit_text_wrapper(
                             mask_np_list,
                             mask_label_list,