Update app.py

app.py

@@ -95,16 +95,27 @@ def edit_mask_add(canvas,  image, idx, mask_np_list):
     return mask_np_list_updated, image_edit
 
 def slider_release(index, image,  mask_np_list_updated, mask_label_list):
-
     if index > len(mask_np_list_updated)-1:
-        return image, "out of range"
+        return image, "out of range", ""
     else:
         mask_np = mask_np_list_updated[index]
         mask_label = mask_label_list[index]
+        index = mask_label.rfind('-')
+        mask_label = mask_label[:index]
+        if mask_label == 'handbag':
+            mask_prompt = "white handbag"
+        elif mask_label == 'person':
+            mask_prompt = "young man"
+        elif mask_label == 'wall-other-merged':
+            mask_prompt = "white wall"
+        elif mask_label == 'table-merged':
+            mask_prompt = "table"
+        else:
+            mask_prompt = mask_label
         segmentation = create_segmentation(mask_np_list_updated)
         new_image = transparent_paste_with_mask(image, segmentation, mask_np, transparency = TRANSPARENCY)
     gr.Info('Edit '+ mask_label)
-    return new_image, mask_label
+    return new_image, mask_label, mask_prompt
 def image_change():
     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)
 
@@ -186,10 +197,6 @@ with gr.Blocks() as demo:
                 gr.Markdown("""<p style="text-align: center; font-size: 20px">Edit Mask (Optional)</p>""")
                 slider =  gr.Slider(0, 20, step=1, label = 'mask id',  visible=False)
                 label = gr.Text(label='label')
-                slider.release(slider_release, 
-                        inputs = [slider, image_loaded,   mask_np_list_updated, mask_label_list], 
-                        outputs= [canvas, label]
-                    )
 
             
                 
@@ -260,10 +267,9 @@ with gr.Blocks() as demo:
                 
                 with gr.Column():
                     gr.Markdown("""<p style="text-align: center; font-size: 20px">Editing setting (SD)</p>""")
-                    tgt_prompt =  gr.Textbox(value="White bag", label="Editing: Text prompt", interactive= True )
+                    tgt_prompt =  gr.Textbox(value="text prompt", label="Editing: Text prompt", interactive= True )
                     with gr.Accordion(label="Advanced settings", open=False):
                         slider2 = gr.Slider(0, 20, step=1, label = 'mask id',  visible=False)
-                        #tgt_index = gr.Number(value="0", label="Editing: Object index", interactive= True )
                         guidance_scale = gr.Textbox(value="5", label="Editing: CFG guidance scale", interactive= True )
                         num_sampling_steps = gr.Number(value="20", label="Editing: Sampling steps", interactive= True )
                         edge_thickness = gr.Number(value="10", label="Editing: Edge thickness", interactive= True )
@@ -333,7 +339,11 @@ with gr.Blocks() as demo:
                                     slider2
                                 ],        
                         outputs = [canvas_text_edit],queue=True)
-                    
+
+        slider.release(slider_release, 
+                        inputs = [slider, image_loaded,   mask_np_list_updated, mask_label_list], 
+                        outputs= [canvas, label,tgt_prompt])
+
         slider.change(
             lambda x: x,
             inputs=[slider],