Update app.py

app.py

@@ -4,9 +4,24 @@ import os
 import shutil
 import random
 
+def process_transparent_image(image, padding_color):
+    """Handle images with transparency by converting transparent areas to a solid color."""
+    if image.mode in ("RGBA", "LA") or (image.mode == "P" and "transparency" in image.info):
+        alpha = image.convert("RGBA").getchannel("A")  # Extract the alpha channel
+        # Create a new image with the background color
+        background = Image.new("RGBA", image.size, padding_color + (255,))
+        background.paste(image, mask=alpha)  # Paste with transparency mask
+        return background.convert("RGB")  # Convert back to RGB
+    return image.convert("RGB")  # For non-transparent images
+
 def resize_and_pad(image, target_size):
     """Resize the image and pad it to match target size."""
-    img = image.convert("RGB")
+    # Randomly select black or white as the background color
+    padding_color = (255, 255, 255) if random.choice([True, False]) else (0, 0, 0)
+
+    # Handle transparent images
+    img = process_transparent_image(image, padding_color)
+
     aspect_ratio = img.width / img.height
     target_aspect_ratio = target_size[0] / target_size[1]
 
@@ -22,9 +37,6 @@ def resize_and_pad(image, target_size):
     # Use high-quality resampling for resizing
     img = img.resize((new_width, new_height), Image.Resampling.LANCZOS)
 
-    # Randomly select black or white as the background color
-    padding_color = (255, 255, 255) if random.choice([True, False]) else (0, 0, 0)
-
     # Create a new image with the determined background color
     new_img = Image.new("RGB", target_size, padding_color)
     offset = ((target_size[0] - new_width) // 2, (target_size[1] - new_height) // 2)