Update gradio_demo.py

gradio_demo.py

@@ -785,6 +785,8 @@ class MaskMover:
         
         return np.array(result.convert('RGB'))  # Convert back to 'RGB' if needed
 
+
+
 @spaces.GPU(duration=60)
 @torch.inference_mode()
 def get_depth(image):
@@ -1358,9 +1360,10 @@ with block:
         #         return composite
 
         class BackgroundManager:
-            def __init__(self):
+            def __init__(self, mask_mover):
                 self.original_bg = None  # To store the original background
                 self.current_mask = None  # To store the current mask applied
+                self.mask_mover = mask_mover  # Reference to the MaskMover instance
         
             def update_background(self, new_bg):
                 """Set a new background."""
@@ -1375,14 +1378,14 @@ with block:
                     print("No original background set.")
                     return None
                 
-                # Start from a clean copy of the original background
-                fresh_bg = self.original_bg.copy()
+                # Use the original background from MaskMover
+                bg = self.mask_mover.original_fg  # Assuming original_fg is the background
                 
                 # Debugging
                 print(f"Updating position: x={x_pos}, y={y_pos}, scale={scale}")
                 
-                # Composite the foreground onto the fresh background
-                composite = mask_mover.create_composite(fresh_bg, x_pos, y_pos, scale)
+                # Composite the foreground onto the original background
+                composite = self.mask_mover.create_composite(bg, x_pos, y_pos, scale)
                 
                 # Update the current mask
                 self.current_mask = composite
@@ -1391,7 +1394,7 @@ with block:
 
         # Create an instance of BackgroundManager
         #bg_manager = gr.State(BackgroundManager())
-        bg_manager = BackgroundManager()
+        bg_manager = BackgroundManager(mask_mover)
 
 
         def update_position_wrapper(background, x_pos, y_pos, scale):