Update app.py

app.py

@@ -36,27 +36,6 @@ def create_segmentation(mask_np_list):
     segmentation = Image.fromarray(np.uint8(segmentation*255))
     return segmentation
 
-def load_mask_ui(input_folder="example_tmp",load_edit = False):
-    if not load_edit:
-        mask_list, mask_label_list = load_mask(input_folder)
-    else:
-        mask_list, mask_label_list = load_mask_edit(input_folder) 
-        
-    mask_np_list = []
-    for  m  in mask_list:
-        mask_np_list. append( m.cpu().numpy())
-
-    return mask_np_list, mask_label_list
-
-def load_image_ui(load_edit, input_folder="example_tmp"):
-
-    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):
@@ -65,7 +44,7 @@ def run_segmentation_wrapper(image):
     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, visible=True)
     return image, segmentation, mask_np_list, mask_label_list, image, sliderup, sliderup , 'Segmentation finish. Select mask id and move to the next step.'
         
 
@@ -123,6 +102,8 @@ def slider_release(index, image,  mask_np_list_updated, mask_label_list):
         segmentation = create_segmentation(mask_np_list_updated)
         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)
 
 def save_as_orig_mask(mask_np_list_updated, mask_label_list, input_folder="example_tmp"):
     print(mask_np_list_updated)
@@ -202,17 +183,14 @@ with gr.Blocks() as demo:
             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, label = 'mask id',  interactive=False)
+                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]
                     )
 
-        
-
-
-                
+            canvas.upload(image_change, inputs=[], outputs=[slider])
                 
     with gr.Tab(label="2 Optimization"):
         with gr.Row():
@@ -303,7 +281,7 @@ with gr.Blocks() as demo:
                     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 )
-                    slider2 = gr.Slider(0, 20, step=1, label = 'mask id',  interactive=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="6", label="Editing: CFG guidance scale", interactive= True )
                     num_sampling_steps = gr.Number(value="50", label="Editing: Sampling steps", interactive= True )