Update app.py

app.py

@@ -46,16 +46,15 @@ def run_segmentation_wrapper(image):
         print("!!", len(mask_np_list))  
         max_val = len(mask_np_list)-1
         sliderup = gr.Slider(value = 0, minimum=0, maximum=max_val, step=1, visible=True)
+        gr.Info('Segmentation finish. Select mask id and move to the next step.')
         return image, segmentation, mask_np_list, mask_label_list, image, sliderup, sliderup , 'Segmentation finish. Select mask id and move to the next step.'
     except:
         sliderup = gr.Slider(value = 0, minimum=0, maximum=1, step=1, visible=False)
+        gr.Warning('Please upload an image before proceeding.')
         return None,None,None,None,None, sliderup, sliderup , 'Please upload an image before proceeding.'
         
 
 def transparent_paste_with_mask(backimg, foreimg, mask_np,transparency = 128):
-    print(type(backimg))
-    print(type(foreimg))
-    print(type(mask_np))
     backimg_solid_np =  np.array(backimg)
     bimg = backimg.copy()
     fimg = foreimg.copy()
@@ -308,6 +307,7 @@ with gr.Blocks() as demo:
                             tgt_index = int(tgt_index)
                         )
                         run_edit_text()
+                        gr.Info('Image edit finish.')
                         return load_pil_img()