Create app.py

app.py

@@ -0,0 +1,64 @@
+import cv2
+import numpy as np
+from PIL import Image, ImageDraw, ImageFont
+import gradio as gr
+
+def detect_shirt_contour(image):
+    gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
+    blurred = cv2.GaussianBlur(gray, (5, 5), 0)
+    edges = cv2.Canny(blurred, 50, 150)
+
+    contours, _ = cv2.findContours(edges, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
+    contours = sorted(contours, key=cv2.contourArea, reverse=True)
+
+    if contours:
+        return cv2.boundingRect(contours[0])  
+    return None
+
+def warp_text(image, text, font_size, color):
+    pil_image = Image.fromarray(cv2.cvtColor(image, cv2.COLOR_BGR2RGB)).convert("RGBA")
+    draw = ImageDraw.Draw(pil_image)
+
+    font_path = "arial.ttf"  # Use a valid font
+    font = ImageFont.truetype(font_path, font_size)
+
+    text_width, text_height = draw.textsize(text, font=font)
+    text_overlay = Image.new("RGBA", (text_width, text_height), (255, 255, 255, 0))
+    text_draw = ImageDraw.Draw(text_overlay)
+    text_draw.text((0, 0), text, font=font, fill=color + (255,))
+
+    text_np = np.array(text_overlay)
+
+    shirt_bbox = detect_shirt_contour(image)
+    if not shirt_bbox:
+        return "Failed to detect shirt region."
+
+    x, y, w, h = shirt_bbox
+    src_points = np.float32([[0, 0], [text_width, 0], [0, text_height], [text_width, text_height]])
+    dst_points = np.float32([[x, y], [x + w, y], [x, y + h], [x + w, y + h]])
+
+    matrix = cv2.getPerspectiveTransform(src_points, dst_points)
+    warped_text = cv2.warpPerspective(text_np, matrix, (image.shape[1], image.shape[0]))
+
+    warped_text_pil = Image.fromarray(warped_text)
+    blended = Image.alpha_composite(pil_image, warped_text_pil).convert("RGB")
+
+    return blended
+
+def gradio_interface(image, text, font_size, color):
+    color_tuple = tuple(map(int, color.lstrip('#')[1:].lower()))
+    img = cv2.imdecode(np.frombuffer(image.read(), np.uint8), cv2.IMREAD_COLOR)
+    return warp_text(img, text, font_size, color_tuple)
+
+gr.Interface(
+    fn=gradio_interface,
+    inputs=[
+        gr.Image(type="file", label="Upload T-shirt Image"),
+        gr.Textbox(label="Enter Text"),
+        gr.Slider(10, 150, step=5, label="Font Size"),
+        gr.ColorPicker(label="Text Color")
+    ],
+    outputs=gr.Image(label="Warped T-shirt Design"),
+    title="Advanced T-shirt Text Warping",
+    description="Upload a t-shirt image and see the text realistically warp onto the fabric."
+).launch()