Update gradio_demo.py

gradio_demo.py

@@ -1217,6 +1217,12 @@ with block:
         # Create an instance of BackgroundManager
         bg_manager = BackgroundManager()
 
+        input_bg.change(
+        fn=lambda new_bg: setattr(bg_manager, 'original_bg', None) or new_bg,
+        inputs=[input_bg],
+        outputs=[input_bg],
+        show_progress=False
+    )
 
         x_slider.change(
             fn=lambda bg, x, y, scale: bg_manager.update_position(bg, x, y, scale),
@@ -1289,26 +1295,39 @@ with block:
         # Create composite in cropped region
         fg_local_x = int(new_width/2 + crop_width*padding)
         fg_local_y = int(new_height/2 + crop_height*padding)
-        cropped_composite = mask_mover.create_composite(crop_region, fg_local_x, fg_local_y, scale)
+        #cropped_composite = mask_mover.create_composite(crop_region, fg_local_x, fg_local_y, scale)
         
         # Process the cropped region
+        # crop_args = list(args)
+        # crop_args[0] = cropped_composite
+        # crop_args[1] = crop_region
+        # crop_args[3] = crop_width
+        # crop_args[4] = crop_height
+        # crop_args = crop_args[:-3]  # Remove position and scale arguments
+
         crop_args = list(args)
-        crop_args[0] = cropped_composite
-        crop_args[1] = crop_region
+        crop_args[0] = None  # or an empty placeholder for fg if needed
+        crop_args[1] = crop_region  # keep the background as is
         crop_args[3] = crop_width
         crop_args[4] = crop_height
-        crop_args = crop_args[:-3]  # Remove position and scale arguments
+        crop_args = crop_args[:-3]  # remove position/scale arguments
         
         # Get relit result
         relit_crop = process_relight_bg(*crop_args)[0]
         
+        
         # Resize relit result to match crop dimensions if needed
         if relit_crop.shape[:2] != (crop_height, crop_width):
             relit_crop = resize_without_crop(relit_crop, crop_width, crop_height)
+
+        final_composite = mask_mover.create_composite(relit_crop, fg_local_x, fg_local_y, scale)
+
         
         # Place relit crop back into original background
         result = background.copy()
-        result[crop_y:crop_y+crop_height, crop_x:crop_x+crop_width] = relit_crop
+        #result[crop_y:crop_y+crop_height, crop_x:crop_x+crop_width] = relit_crop
+        result[crop_y:crop_y+crop_height, crop_x:crop_x+crop_width] = final_composite
+
         
         return result