Update gradio_demo.py

gradio_demo.py

@@ -1164,7 +1164,8 @@ with block:
                                 value=""
                             )
                     extract_button = gr.Button(value="(Option 2) Remove Background")
-                        #find_objects_button = gr.Button(value="Find Objects")
+                with gr.Row():
+                    extracted_objects = gr.Image(type="numpy", label="Extracted Foreground", height=480)
                     extracted_fg = gr.Image(type="numpy", label="Extracted Foreground", height=480)
 
                 with gr.Row():    
@@ -1310,13 +1311,13 @@ with block:
                 
         #     #     return mask_mover.create_composite(self.original_bg, x_pos, y_pos, scale)
             
-            # def update_position(background, x_pos, y_pos, scale):
-            #     if background is None:
-            #         return None
-            #     # Restore a fresh copy of the original background
-            #     fresh_bg = bg_manager.original_bg.copy()
-            #     # Composite the foreground once
-            #     return mask_mover.create_composite(fresh_bg, float(x_pos), float(y_pos), float(scale))
+            def update_position(background, x_pos, y_pos, scale):
+                if background is None:
+                    return None
+                # Restore a fresh copy of the original background
+                fresh_bg = bg_manager.original_bg.copy()
+                # Composite the foreground once
+                return mask_mover.create_composite(fresh_bg, float(x_pos), float(y_pos), float(scale))
 
   
 
@@ -1337,22 +1338,22 @@ with block:
         inputs=[input_bg],
         outputs=[input_bg],
         show_progress=False
-    )
+        )
 
         x_slider.change(
-            fn=lambda bg, x, y, scale: update_position(bg, x, y, scale),
+            fn=lambda bg, x, y, scale: bg_manager.update_position(bg, x, y, scale),
             inputs=[input_bg, x_slider, y_slider, fg_scale_slider],
             outputs=[input_bg]
         )
 
         y_slider.change(
-            fn=lambda bg, x, y, scale: update_position(bg, x, y, scale),
+            fn=lambda bg, x, y, scale: bg_manager.update_position(bg, x, y, scale),
             inputs=[input_bg, x_slider, y_slider, fg_scale_slider],
             outputs=[input_bg]
         )
 
         fg_scale_slider.change(
-            fn=lambda bg, x, y, scale: update_position(bg, x, y, scale),
+            fn=lambda bg, x, y, scale: bg_manager.update_position(bg, x, y, scale),
             inputs=[input_bg, x_slider, y_slider, fg_scale_slider],
             outputs=[input_bg]
         )