Update app.py

app.py

@@ -4,20 +4,24 @@ from PIL import Image
 
 def overlay_images(x, y, scale, design_img, target_img):
     # Convert inputs to PIL images
-    design_image = Image.fromarray(design_img)
-    target_image = Image.fromarray(target_img)
+    design_image = Image.fromarray(design_img).convert("RGBA")
+    target_image = Image.fromarray(target_img).convert("RGBA")
 
     # Resize the design image based on scale
     design_image = design_image.resize(
         (int(design_image.width * scale), int(design_image.height * scale)), Image.Resampling.LANCZOS
     )
 
+    # Ensure the design image has an alpha channel for transparency
+    if design_image.mode != 'RGBA':
+        design_image = design_image.convert('RGBA')
+    
     # Create a transparent overlay
     overlay = Image.new('RGBA', target_image.size, (255, 255, 255, 0))
     overlay.paste(design_image, (x, y), design_image)
-    
+
     # Combine overlay with target image
-    combined = Image.alpha_composite(target_image.convert('RGBA'), overlay)
+    combined = Image.alpha_composite(target_image, overlay)
     
     return np.array(combined)