Update app.py

app.py

@@ -3,70 +3,56 @@ import numpy as np
 from PIL import Image, ImageDraw, ImageFont
 import gradio as gr
 
-
-# Function to detect fabric texture and generate a displacement map
 def generate_displacement_map(image):
     gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
     laplacian = cv2.Laplacian(gray, cv2.CV_32F)  # Detect edges and texture
     displacement_map = cv2.normalize(laplacian, None, alpha=0, beta=1, norm_type=cv2.NORM_MINMAX)
     return displacement_map
 
-
-# Function to blend text with the cloth image
-def blend_text_with_cloth(cloth_image, text, font_size=32, font_color=(255, 0, 0), x=50, y=50):
-    # Convert cloth image to BGR for OpenCV processing
+def apply_text_warping(cloth_image, text, font_size=32, font_color=(255, 255, 255), x=50, y=50):
     cloth_image = np.array(cloth_image)
-    cloth_bgr = cv2.cvtColor(cloth_image, cv2.COLOR_RGBA2BGRA)
-
-    # Generate displacement map
+    cloth_bgr = cv2.cvtColor(cloth_image, cv2.COLOR_RGBA2BGR)
     displacement_map = generate_displacement_map(cloth_bgr)
 
-    # Render text
     img_pil = Image.fromarray(cloth_image)
     draw = ImageDraw.Draw(img_pil)
-
-    # Load a font (use a default font if custom font isn't available)
     try:
         font = ImageFont.truetype("arial.ttf", font_size)
     except:
         font = ImageFont.load_default()
-
-    draw.text((x, y), text, font=font, fill=font_color)
-    rendered_text = np.array(img_pil)
-
-    # Convert rendered text image to BGR
-    text_bgr = cv2.cvtColor(rendered_text, cv2.COLOR_RGBA2BGRA)
-
-    # Blend text with cloth image using displacement map
-    blended = cv2.addWeighted(cloth_bgr, 0.8, text_bgr, 0.5, 0)
-    final_image = cv2.cvtColor(blended, cv2.COLOR_BGRA2RGBA)
-
+    
+    text_img = Image.new("RGBA", img_pil.size, (255, 255, 255, 0))
+    text_draw = ImageDraw.Draw(text_img)
+    text_draw.text((x, y), text, font=font, fill=font_color+(255,))
+    text_img = np.array(text_img)
+    text_gray = cv2.cvtColor(text_img, cv2.COLOR_RGBA2GRAY)
+    _, text_mask = cv2.threshold(text_gray, 1, 255, cv2.THRESH_BINARY)
+    text_mask = cv2.GaussianBlur(text_mask, (5, 5), 3)
+    text_mask = cv2.cvtColor(text_mask, cv2.COLOR_GRAY2BGRA) / 255.0
+    cloth_bgr = cv2.cvtColor(cloth_image, cv2.COLOR_RGBA2BGRA)
+    blended = cloth_bgr * (1 - text_mask) + text_img * text_mask
+    final_image = cv2.cvtColor(blended.astype(np.uint8), cv2.COLOR_BGRA2RGBA)
     return Image.fromarray(final_image)
 
-
-# Gradio function
 def process_image(image, text, font_size, font_color, x, y):
-    font_color = tuple(map(int, font_color.strip("()").split(",")))  # Convert string to tuple
-    result = blend_text_with_cloth(image, text, font_size, font_color, x, y)
+    font_color = tuple(map(int, font_color.strip("()").split(",")))
+    result = apply_text_warping(image, text, font_size, font_color, x, y)
     return result
 
-
-# Gradio Interface
 interface = gr.Interface(
     fn=process_image,
     inputs=[
         gr.Image(type="pil", label="Upload Cloth Image"),
-        gr.Textbox(label="Text to Blend", value="Sample Text"),  # Use `value` instead of `default`
+        gr.Textbox(label="Text to Blend", value="Your Text Here"),
         gr.Slider(10, 100, step=2, label="Font Size", value=32),
-        gr.Textbox(label="Font Color (RGB)", value="(255, 0, 0)"),  # Use `value` instead of `default`
+        gr.Textbox(label="Font Color (RGB)", value="(255,255,255)"),
         gr.Slider(0, 1000, step=10, label="X Coordinate", value=50),
         gr.Slider(0, 1000, step=10, label="Y Coordinate", value=50),
     ],
     outputs=gr.Image(type="pil", label="Blended Output"),
-    title="Text-Cloth Blending",
-    description="Upload a cloth image and add text that blends naturally into the fabric. Adjust parameters like font size, color, and position.",
+    title="Advanced Text-Cloth Blending",
+    description="Upload a cloth image and add naturally blended, warped text.",
 )
 
-# Launch the app
 if __name__ == "__main__":
-    interface.launch(server_name="0.0.0.0", server_port=7860)
+    interface.launch()