Update app.py

app.py

@@ -1,43 +1,53 @@
 import gradio as gr
 from PIL import Image
+import numpy as np
 
 def overlay_image(background, overlay):
-    # Resize overlay to fit the background
-    overlay = overlay.resize(background.size, Image.ANTIALIAS)
-
-    # Create a new image for the result
-    result = Image.new("RGBA", background.size)
-
-    # Paste the background and overlay
-    result.paste(background, (0, 0))
-    result.paste(overlay, (0, 0), overlay)
-
-    return result
+    try:
+        # Ensure both images have alpha channel
+        background = background.convert("RGBA")
+        overlay = overlay.convert("RGBA")
+        
+        # Resize overlay to match background
+        overlay = overlay.resize(background.size, Image.Resampling.LANCZOS)
+        
+        # Composite images
+        return Image.alpha_composite(background, overlay)
+    except Exception as e:
+        print(f"Error in overlay: {str(e)}")
+        return None
 
 def virtual_try_on(user_image, shirt_image):
-    # Convert images to RGBA
-    user_image = user_image.convert("RGBA")
-    shirt_image = Image.open(shirt_image).convert("RGBA")
-
-    # Overlay the shirt on the user image
-    result_image = overlay_image(user_image, shirt_image)
-
-    return user_image, result_image
+    try:
+        # Open and verify images
+        user_img = Image.fromarray(user_image) if isinstance(user_image, np.ndarray) else user_image
+        shirt_img = Image.open(shirt_image) if isinstance(shirt_image, str) else shirt_image
+        
+        result = overlay_image(user_img.convert("RGBA"), shirt_img.convert("RGBA"))
+        
+        if result is None:
+            raise gr.Error("Failed to process images. Please check your inputs.")
+            
+        return [user_img, result]
+    except Exception as e:
+        print(f"Error in try-on: {str(e)}")
+        raise gr.Error(f"Processing error: {str(e)}")
 
-# Gradio interface
 iface = gr.Interface(
     fn=virtual_try_on,
     inputs=[
-        gr.Image(type="pil", label="Upload your front-facing photo"),
-        gr.Image(type="filepath", label="Upload or choose a t-shirt design (PNG format)")
+        gr.Image(label="Your Photo", type="pil"),
+        gr.Image(label="T-Shirt Design", type="filepath")
     ],
     outputs=[
-        gr.Image(label="Original Photo"),
-        gr.Image(label="Try-On Preview")
+        gr.Image(label="Original"),
+        gr.Image(label="Virtual Try-On")
     ],
-    title="Virtual Try-On Tool",
-    description="Try on our t-shirts by uploading your photo and selecting a t-shirt design.",
-    theme="default"
+    title="YOURBRAND Virtual Try-On",
+    description="Upload your photo and a t-shirt design to see how it looks!",
+    examples=[
+        ["sample_person.jpg", "sample_shirt.png"]
+    ]
 )
 
-iface.launch()
+iface.launch(debug=True)