Update app.py

app.py

@@ -47,31 +47,28 @@ pipe = StableDiffusionXLFillPipeline.from_pretrained(
 
 pipe.scheduler = TCDScheduler.from_config(pipe.scheduler.config)
 
-def resize_and_pad(image, target_size):
-    aspect_ratio = image.width / image.height
-    target_aspect_ratio = target_size[0] / target_size[1]
-
-    if aspect_ratio > target_aspect_ratio:
-        # Image is wider than the target ratio
-        new_width = target_size[0]
-        new_height = int(new_width / aspect_ratio)
-    else:
-        # Image is taller than the target ratio
-        new_height = target_size[1]
-        new_width = int(new_height * aspect_ratio)
-
-    resized_image = image.resize((new_width, new_height), Image.LANCZOS)
+def resize_and_pad(image, target_size, resize_width=512):
+    # Calculate the new height to maintain aspect ratio
+    aspect_ratio = image.height / image.width
+    new_height = int(resize_width * aspect_ratio)
+    
+    # Resize the image
+    resized_image = image.resize((resize_width, new_height), Image.LANCZOS)
     
+    # Create a new white image with the target size
     new_image = Image.new('RGB', target_size, (255, 255, 255))
     
-    paste_x = (target_size[0] - new_width) // 2
+    # Calculate position to paste the resized image (center it)
+    paste_x = (target_size[0] - resize_width) // 2
     paste_y = (target_size[1] - new_height) // 2
     
+    # Paste the resized image onto the new image
     new_image.paste(resized_image, (paste_x, paste_y))
     
+    # Create a mask
     mask = Image.new('L', target_size, 255)
     mask_draw = ImageDraw.Draw(mask)
-    mask_draw.rectangle([paste_x, paste_y, paste_x + new_width, paste_y + new_height], fill=0)
+    mask_draw.rectangle([paste_x, paste_y, paste_x + resize_width, paste_y + new_height], fill=0)
     
     return new_image, mask
 
@@ -80,7 +77,7 @@ def infer(image, model_selection, width, height, overlap_width, num_inference_st
     target_size = (width, height)
     
     if expand_mode:
-        background, mask = resize_and_pad(image, target_size)
+        background, mask = resize_and_pad(image, target_size, resize_width=512)
         cnet_image = background.copy()
         cnet_image.paste(0, (0, 0), mask)
     else: