Create app.py

app.py

@@ -0,0 +1,48 @@
+import gradio as gr
+import numpy as np
+from PIL import Image, ImageOps
+import cv2
+
+def overlay_design(hoodie_img, design_img, x_offset, y_offset, scale, rotation):
+    if hoodie_img is None or design_img is None:
+        return None
+    
+    # Open images
+    hoodie = Image.open(hoodie_img).convert("RGBA")
+    design = Image.open(design_img).convert("RGBA")
+    
+    # Resize and rotate the design
+    width, height = design.size
+    design = design.resize((int(width * scale), int(height * scale)))
+    design = design.rotate(rotation, expand=True)
+    
+    # Create a blank image with same size as hoodie
+    temp_layer = Image.new("RGBA", hoodie.size, (0, 0, 0, 0))
+    
+    # Paste design onto temp_layer at specified position
+    temp_layer.paste(design, (int(x_offset), int(y_offset)), design)
+    
+    # Blend the layers
+    final_image = Image.alpha_composite(hoodie, temp_layer)
+    return final_image
+
+def generate_mockup(hoodie, design, x_offset=100, y_offset=200, scale=1.0, rotation=0):
+    return overlay_design(hoodie, design, x_offset, y_offset, scale, rotation)
+
+# Gradio UI
+demo = gr.Interface(
+    fn=generate_mockup,
+    inputs=[
+        gr.Image(type="filepath", label="Upload Hoodie Mockup"),
+        gr.Image(type="filepath", label="Upload Your Design"),
+        gr.Slider(0, 500, value=100, label="X Offset"),
+        gr.Slider(0, 500, value=200, label="Y Offset"),
+        gr.Slider(0.1, 2.0, value=1.0, label="Scale"),
+        gr.Slider(-180, 180, value=0, label="Rotation")
+    ],
+    outputs=gr.Image(type="pil", label="Generated Mockup"),
+    title="Rookus Hoodie Mockup Generator",
+    description="Upload a hoodie mockup and your design to generate a realistic preview."
+)
+
+demo.launch()