Update app.py

app.py

@@ -164,32 +164,57 @@ def gradio_interface():
             if initial_image_pil is not None:
                 image_history = [initial_image_pil]
                 current_image_pil = initial_image_pil
-                return current_image_pil, image_history, initial_image_pil, {}, gr.Dropdown.update(choices=[], value=None), "Image loaded."
+                return current_image_pil, image_history, initial_image_pil, {}, gr.update(choices=[], value=None), "Image loaded."
             else:
-                return None, [], None, {}, gr.Dropdown.update(choices=[], value=None), "No image loaded."
+                return None, [], None, {}, gr.update(choices=[], value=None), "No image loaded."
 
         # When the initial image is uploaded, initialize the image history
-        initial_image.upload(fn=initialize_image, inputs=initial_image, outputs=[current_image_pil, image_history, current_image_display, masks_dict, segment_dropdown, status])
+        initial_image.upload(
+            fn=initialize_image,
+            inputs=initial_image,
+            outputs=[current_image_pil, image_history, current_image_display, masks_dict, segment_dropdown, status]
+        )
 
         # Segment button click
         def segment_image_wrapper(current_image_pil, prompts):
             if current_image_pil is None:
-                return "No image uploaded.", {}, gr.Dropdown.update(choices=[], value=None)
+                return "No image uploaded.", {}, gr.update(choices=[], value=None)
             masks = extract_masks(current_image_pil, prompts)
             if not masks:
-                return "No masks detected for the given prompts.", {}, gr.Dropdown.update(choices=[], value=None)
+                return "No masks detected for the given prompts.", {}, gr.update(choices=[], value=None)
             dropdown_choices = list(masks.keys())
-            return "Segmentation completed.", masks, gr.Dropdown.update(choices=dropdown_choices, value=dropdown_choices[0])
+            return "Segmentation completed.", masks, gr.update(choices=dropdown_choices, value=dropdown_choices[0])
 
-        segment_button.click(fn=segment_image_wrapper, inputs=[current_image_pil, prompts], outputs=[status, masks_dict, segment_dropdown])
+        segment_button.click(
+            fn=segment_image_wrapper,
+            inputs=[current_image_pil, prompts],
+            outputs=[status, masks_dict, segment_dropdown]
+        )
 
         # Apply button click
-        apply_button.click(fn=process_image,
-                           inputs=[current_image_pil, segment_dropdown, masks_dict, replacement_image, color_ref_image, apply_replacement, apply_color_grading, apply_color_to_full_image, blending_amount, image_history],
-                           outputs=[current_image_pil, status, image_history, current_image_display])
+        apply_button.click(
+            fn=process_image,
+            inputs=[
+                current_image_pil,
+                segment_dropdown,
+                masks_dict,
+                replacement_image,
+                color_ref_image,
+                apply_replacement,
+                apply_color_grading,
+                apply_color_to_full_image,
+                blending_amount,
+                image_history
+            ],
+            outputs=[current_image_pil, status, image_history, current_image_display]
+        )
 
         # Undo button click
-        undo_button.click(fn=undo, inputs=image_history, outputs=[current_image_pil, image_history, current_image_display])
+        undo_button.click(
+            fn=undo,
+            inputs=image_history,
+            outputs=[current_image_pil, image_history, current_image_display]
+        )
 
     demo.launch(share=True)