Update gradio_demo.py

gradio_demo.py

@@ -791,6 +791,15 @@ def get_depth(image):
 
 from PIL import Image
 
+def extract_foreground(image):
+            if image is None:
+                return None, gr.update(visible=True), gr.update(visible=True)
+            result, rgba = run_rmbg(image)
+            mask_mover.set_extracted_fg(rgba)
+            
+            return result, gr.update(visible=True), gr.update(visible=True)
+
+
 def compress_image(image):
     # Convert Gradio image (numpy array) to PIL Image
     img = Image.fromarray(image)
@@ -1006,7 +1015,8 @@ with block:
         ips = [input_fg, prompt, image_width, image_height, num_samples, seed, steps, a_prompt, n_prompt, cfg, highres_scale, highres_denoise, lowres_denoise, bg_source]
         relight_button.click(fn=process_relight, inputs=ips, outputs=[extracted_fg, result_gallery], show_progress=True)
         example_quick_prompts.click(lambda x, y: ', '.join(y.split(', ')[:2] + [x[0]]), inputs=[example_quick_prompts, prompt], outputs=prompt, show_progress=False, queue=False)
-        example_quick_subjects.click(lambda x: x[0], inputs=example_quick_subjects, outputs=prompt, show_progress=False, queue=False)
+        example_quick_subjects.click(lambda x: x[0], inputs=example_quick_subjects, outputs=prompt, show_progress=False, queue=False
+                                    
         extract_button.click(
             fn=extract_foreground,
             inputs=[input_fg],
@@ -1131,14 +1141,7 @@ with block:
             with gr.Column():
                 result_gallery = gr.Image(height=832, label='Outputs')
 
-        def extract_foreground(image):
-            if image is None:
-                return None, gr.update(visible=True), gr.update(visible=True)
-            result, rgba = run_rmbg(image)
-            mask_mover.set_extracted_fg(rgba)
-            
-            return result, gr.update(visible=True), gr.update(visible=True)
-
+        
 
         original_bg = None