Update app.py

app.py

@@ -932,7 +932,7 @@ def process_image(input_image, input_text):
             # Set extracted foreground for mask mover
             mask_mover.set_extracted_fg(cropped_rgba)
             
-            return annotated_frame, cropped_rgba, gr.update(visible=True), gr.update(visible=True)
+            return annotated_frame, cropped_rgba, gr.update(visible=False), gr.update(visible=False)
 
     
     else:
@@ -1061,7 +1061,7 @@ def process_image(input_image, input_text):
             # Set extracted foreground for mask mover
             mask_mover.set_extracted_fg(cropped_rgba)
             
-            return annotated_frame, cropped_rgba, gr.update(visible=True), gr.update(visible=True)
+            return annotated_frame, cropped_rgba, gr.update(visible=False), gr.update(visible=False)
         return annotated_frame, None, gr.update(visible=False), gr.update(visible=False)
         
 
@@ -1141,15 +1141,17 @@ with block:
             #     outputs=[result_gallery, output_bg],
             #     run_on_click=True, examples_per_page=1024
             # )
-        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]
+        ips = [extracted_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=[result_gallery])
         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)
+        
         find_objects_button.click(
             fn=process_image,
             inputs=[input_fg, text_prompt],
             outputs=[extracted_objects, extracted_fg]
             )
+        
         extract_button.click(
             fn=extract_foreground,
             inputs=[input_fg],