Update gradio_demo.py

gradio_demo.py

@@ -1174,7 +1174,7 @@ with block:
         
     with gr.Tab("Background", visible=True):
         # empty cache
-    
+        bg_manager = BackgroundManager()
         mask_mover = MaskMover()
         # with torch.no_grad():
         #     # Update the input channels to 12
@@ -1331,22 +1331,51 @@ with block:
             
         #     return mask_mover.create_composite(original_bg, x_pos, y_pos, scale)
 
+        # class BackgroundManager:
+        #     def __init__(self):
+        #         self.original_bg = None  # To store the original background
+        
+        #     def update_background(self, new_bg):
+        #         """Set a new background."""
+        #         if new_bg is not None:
+        #             self.original_bg = new_bg.copy()
+        #             print("Background updated successfully.")  # Debugging
+                    
+        #     def update_position(self, background, x_pos, y_pos, scale):
+        #         """Update composite when position changes."""
+        #         if self.original_bg is None:
+        #             print("No original background set.")
+        #             return None
+            
+        #         # Start from a clean copy of the original background
+        #         fresh_bg = self.original_bg.copy()
+                
+        #         # 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)
+                
+        #         return composite
+
         class BackgroundManager:
             def __init__(self):
                 self.original_bg = None  # To store the original background
+                self.current_mask = None  # To store the current mask applied
         
             def update_background(self, new_bg):
                 """Set a new background."""
                 if new_bg is not None:
                     self.original_bg = new_bg.copy()
+                    self.current_mask = None  # Reset current mask when background changes
                     print("Background updated successfully.")  # Debugging
-                    
-            def update_position(self, background, x_pos, y_pos, scale):
+        
+            def update_position(self, x_pos, y_pos, scale):
                 """Update composite when position changes."""
                 if self.original_bg is None:
                     print("No original background set.")
                     return None
-            
+                
                 # Start from a clean copy of the original background
                 fresh_bg = self.original_bg.copy()
                 
@@ -1356,10 +1385,12 @@ with block:
                 # Composite the foreground onto the fresh background
                 composite = mask_mover.create_composite(fresh_bg, x_pos, y_pos, scale)
                 
+                # Update the current mask
+                self.current_mask = composite
                 return composite
-    
+        
+
         # Create an instance of BackgroundManager
-        bg_manager = BackgroundManager()
         #bg_manager = gr.State(BackgroundManager())
 
         def update_position_wrapper(background, x_pos, y_pos, scale):
@@ -1376,31 +1407,58 @@ with block:
 
 
 
+        # input_bg.change(
+        #     fn=lambda new_bg: bg_manager.update_background(new_bg) or new_bg,
+        #     inputs=[input_bg],
+        #     outputs=[input_bg],
+        #     show_progress=False
+        # )
+
+        # x_slider.change(
+        #     fn=bg_manager.update_position,
+        #     inputs=[input_bg, x_slider, y_slider, fg_scale_slider],
+        #     outputs=[input_bg]
+        # )
+        
+        # y_slider.change(
+        #     fn=bg_manager.update_position,
+        #     inputs=[input_bg, x_slider, y_slider, fg_scale_slider],
+        #     outputs=[input_bg]
+        # )
+        
+        # fg_scale_slider.change(
+        #     fn=bg_manager.update_position,
+        #     inputs=[input_bg, x_slider, y_slider, fg_scale_slider],
+        #     outputs=[input_bg]
+        # )
+
+                # Update the input_bg.change function to reset the mask
         input_bg.change(
             fn=lambda new_bg: bg_manager.update_background(new_bg) or new_bg,
             inputs=[input_bg],
             outputs=[input_bg],
             show_progress=False
         )
-
+        
+        # Update the slider change functions to apply the current mask
         x_slider.change(
-            fn=bg_manager.update_position,
-            inputs=[input_bg, x_slider, y_slider, fg_scale_slider],
+            fn=lambda x_pos: bg_manager.update_position(x_pos, y_slider.value, fg_scale_slider.value),
+            inputs=[x_slider],
             outputs=[input_bg]
         )
         
         y_slider.change(
-            fn=bg_manager.update_position,
-            inputs=[input_bg, x_slider, y_slider, fg_scale_slider],
+            fn=lambda y_pos: bg_manager.update_position(x_slider.value, y_pos, fg_scale_slider.value),
+            inputs=[y_slider],
             outputs=[input_bg]
         )
         
         fg_scale_slider.change(
-            fn=bg_manager.update_position,
-            inputs=[input_bg, x_slider, y_slider, fg_scale_slider],
+            fn=lambda scale: bg_manager.update_position(x_slider.value, y_slider.value, scale),
+            inputs=[fg_scale_slider],
             outputs=[input_bg]
         )
-
+    
         # Update inputs list to include fg_scale_slider
 
     def process_relight_with_position(*args):