Update gradio_demo.py

gradio_demo.py

@@ -1372,19 +1372,20 @@ with block:
                 
         #         return composite
 
-        class BackgroundManager:
+       class BackgroundManager:
             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.current_bg = None  # Store current background
                 self.mask_mover = mask_mover  # Reference to the MaskMover instance
         
             def update_background(self, new_bg):
-                """Set a new background."""
+                """Update the current background without affecting the mask"""
                 if new_bg is not None:
-                    self.original_bg = new_bg.copy()
-                    self.current_mask = None  # Reset current mask when background changes
+                    if self.original_bg is None:  # Store the original background only once
+                        self.original_bg = new_bg.copy()
+                    self.current_bg = new_bg.copy()  # Overwrite the current background
                     print("Background updated successfully.")  # Debugging
-
+        
             def reset_background(self):
                 """Reset the background to its original state"""
                 if self.original_bg is not None:
@@ -1462,24 +1463,32 @@ with block:
         #     outputs=[input_bg],
         #     show_progress=False
         # )
-        
-        # Update the slider change functions to apply the current mask
+
+        # Update the input_bg.change function to reset the mask and update the background
+        input_bg.change(
+            fn=lambda new_bg: bg_manager.update_background(new_bg),  # Update the background
+            inputs=[input_bg],
+            outputs=[input_bg],  # You may want to update other outputs as needed
+            show_progress=False
+        )
+
+        # Slider change functions
         x_slider.change(
             fn=lambda x_pos: bg_manager.update_position(x_pos, y_slider.value, fg_scale_slider.value),
             inputs=[x_slider],
-            outputs=[input_bg]
+            outputs=[input_bg]  # This should not reset the background
         )
         
         y_slider.change(
             fn=lambda y_pos: bg_manager.update_position(x_slider.value, y_pos, fg_scale_slider.value),
             inputs=[y_slider],
-            outputs=[input_bg]
+            outputs=[input_bg]  # This should not reset the background
         )
         
         fg_scale_slider.change(
             fn=lambda scale: bg_manager.update_position(x_slider.value, y_slider.value, scale),
             inputs=[fg_scale_slider],
-            outputs=[input_bg]
+            outputs=[input_bg]  # This should not reset the background
         )
     
         # Update inputs list to include fg_scale_slider
@@ -1585,13 +1594,15 @@ with block:
         outputs=[result_gallery], show_progress=True
     )
 
+
+    # Connect the reset button to the reset functionality
     reset_button.click(
         fn=lambda: bg_manager.reset_background(),  # Call the reset function
         inputs=[],  # No inputs needed for reset
         outputs=[input_bg],  # Update the displayed background
         show_progress=True
     )
-    
+        
     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)