Update app.py

app.py

@@ -106,7 +106,7 @@ def can_expand(source_width, source_height, target_width, target_height, alignme
         return False
     return True
 
-def prepare_image_and_mask(image, width, height, overlap_percentage, resize_option, custom_resize_percentage, alignment, overlap_left, overlap_right, overlap_top, overlap_bottom, x_offset, y_offset):
+def prepare_image_and_mask(image, width, height, overlap_percentage, resize_option, custom_resize_percentage, alignment, overlap_left, overlap_right, overlap_top, overlap_bottom, x_offset=None, y_offset=None):
     target_size = (width, height)
     scale_factor = min(target_size[0] / image.width, target_size[1] / image.height)
     new_width = int(image.width * scale_factor)
@@ -155,6 +155,10 @@ def prepare_image_and_mask(image, width, height, overlap_percentage, resize_opti
         margin_x = (target_size[0] - new_width) // 2
         margin_y = target_size[1] - new_height
     
+    # Ensure x_offset and y_offset are not None
+    x_offset = x_offset if x_offset is not None else 0
+    y_offset = y_offset if y_offset is not None else 0
+    
     margin_x = max(0, min(margin_x + x_offset, target_size[0] - new_width))
     margin_y = max(0, min(margin_y + y_offset, target_size[1] - new_height))
     
@@ -176,6 +180,7 @@ def prepare_image_and_mask(image, width, height, overlap_percentage, resize_opti
     ], fill=0)
 
     return background, mask
+    
 def preview_image_and_mask(image, width, height, overlap_percentage, resize_option, custom_resize_percentage, alignment, overlap_left, overlap_right, overlap_top, overlap_bottom, x_offset, y_offset):
     background, mask = prepare_image_and_mask(
         image,
@@ -591,15 +596,29 @@ with gr.Blocks(css=css, fill_height=True) as demo:
         outputs=[input_image_outpaint]
     )
     runout_button.click(
-        fn=clear_result,
-        inputs=None,
-        outputs=result_outpaint,
+    fn=clear_result,
+    inputs=None,
+    outputs=result_outpaint,
     ).then(
-        fn=infer,
-        inputs=[input_image_outpaint, width_slider, height_slider, overlap_percentage, num_inference_steps,
-                resize_option, custom_resize_percentage, prompt_input, alignment_dropdown,
-                overlap_left, overlap_right, overlap_top, overlap_bottom],
-        outputs=[result_outpaint],
+    fn=infer,
+    inputs=[
+        input_image_outpaint,
+        width_slider,
+        height_slider,
+        overlap_percentage,
+        num_inference_steps,
+        resize_option,
+        custom_resize_percentage,
+        prompt_input,
+        alignment_dropdown,
+        overlap_left,
+        overlap_right,
+        overlap_top,
+        overlap_bottom,
+        x_offset_slider,
+        y_offset_slider
+    ],
+    outputs=[result_outpaint],
     ).then(
         fn=lambda x, history: update_history(x[1], history),
         inputs=[result_outpaint, history_gallery],
@@ -609,15 +628,30 @@ with gr.Blocks(css=css, fill_height=True) as demo:
         inputs=None,
         outputs=[use_as_input_button_outpaint],
     )
+    
     prompt_input.submit(
         fn=clear_result,
         inputs=None,
         outputs=result_outpaint,
     ).then(
         fn=infer,
-        inputs=[input_image_outpaint, width_slider, height_slider, overlap_percentage, num_inference_steps,
-                resize_option, custom_resize_percentage, prompt_input, alignment_dropdown,
-                overlap_left, overlap_right, overlap_top, overlap_bottom, x_offset_slider, y_offset_slider],
+        inputs=[
+            input_image_outpaint,
+            width_slider,
+            height_slider,
+            overlap_percentage,
+            num_inference_steps,
+            resize_option,
+            custom_resize_percentage,
+            prompt_input,
+            alignment_dropdown,
+            overlap_left,
+            overlap_right,
+            overlap_top,
+            overlap_bottom,
+            x_offset_slider,
+            y_offset_slider
+        ],
         outputs=[result_outpaint],
     ).then(
         fn=lambda x, history: update_history(x[1], history),