Update app.py

app.py

@@ -503,6 +503,14 @@ def process(input_fg, prompt, image_width, image_height, num_samples, seed, step
 
     return pixels
 
+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)
+
 @torch.inference_mode()
 def process_bg(input_fg, input_bg, prompt, image_width, image_height, num_samples, seed, steps, a_prompt, n_prompt, cfg, highres_scale, highres_denoise, bg_source):
     clear_memory()