Resize the input image to square

app.py

@@ -433,6 +433,27 @@ def resize_image(image, scaling_factor=1):
         int(image.height * scaling_factor)))
     return image
 
+def resize_to_square(image, size=1024):
+
+    # img = Image.open(image_path).convert("RGBA")
+    img = image
+    
+    # Resize while maintaining aspect ratio
+    img.thumbnail((size, size), Image.LANCZOS)
+    
+    # Create a transparent square canvas
+    square_img = Image.new("RGBA", (size, size), (0, 0, 0, 0))
+    
+    # Calculate the position to paste the resized image (centered)
+    x_offset = (size - img.width) // 2
+    y_offset = (size - img.height) // 2
+    
+    # Paste the resized image onto the square canvas
+    square_img.paste(img, (x_offset, y_offset), img)
+
+    return square_img
+
+
 def encode_image(image):
     buffer = BytesIO()
     image.save(buffer, format="PNG")
@@ -527,6 +548,8 @@ def blend_details(input_image, relit_image, masked_image, scaling_factor=1):
 @spaces.GPU
 def generate_image(input_img, prompt):
 
+    input_img = resize_to_square(input_img)
+
     ai_gen_image = generate_ai_bg(input_img, prompt)
 
     mask_input_image = run_grounded_sam(input_img)