Modified resize image

app.py

@@ -435,23 +435,33 @@ def resize_image(image, scaling_factor=1):
 
 def resize_to_square(image, size=1024):
 
-    # img = Image.open(image_path).convert("RGBA")
-    img = image
-    
+    # Load image if a file path is provided
+    if isinstance(image_path_or_img, str):
+        img = Image.open(image_path_or_img).convert("RGBA")
+    else:
+        img = image_path_or_img.convert("RGBA")  # If already an Image object
+
     # 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
+    # Extract the alpha channel as a mask
+    mask = img.split()[3] if img.mode == "RGBA" else None
+
+    # Paste the resized image onto the square canvas with the correct transparency mask
+    square_img.paste(img, (x_offset, y_offset), mask)
+
+    # Save if output_path is provided; otherwise, return the PIL Image object
+    if output_path:
+        square_img.save(output_path, "PNG")
+    else:
+        return square_img
 
 
 def encode_image(image):