Update app.py

app.py

@@ -16,43 +16,19 @@ def generate_text_image(text, style, width, height):
     draw = ImageDraw.Draw(img)
 
     # Dynamically calculate font size based on image dimensions
-    font_size = int(min(width, height) * 0.2)  # Adjust text to be a larger proportion of the image
+    font_size = int(min(width, height) * 0.1)  # Adjust text to be proportional
     try:
         font = ImageFont.truetype(font_path[style], size=font_size)
     except:
         font = ImageFont.load_default()
 
-    # Word wrapping to fit text within the image
-    max_width = width - 20  # Leave some padding
-    words = text.split()
-    lines = []
-    current_line = ""
-
-    for word in words:
-        test_line = f"{current_line} {word}".strip()
-        test_width, _ = draw.textsize(test_line, font=font)
-        if test_width <= max_width:
-            current_line = test_line
-        else:
-            lines.append(current_line)
-            current_line = word
-
-    if current_line:
-        lines.append(current_line)
-
-    # Calculate total text block height
-    line_height = draw.textsize("Test", font=font)[1]
-    total_text_height = line_height * len(lines)
-
-    # Start position for vertically centered text
-    y_position = (height - total_text_height) // 2
-
-    # Draw each line of text
-    for line in lines:
-        text_width, _ = draw.textsize(line, font=font)
-        x_position = (width - text_width) // 2
-        draw.text((x_position, y_position), line, font=font, fill=(0, 0, 0, 255))
-        y_position += line_height
+    text_bbox = draw.textbbox((0, 0), text, font=font)
+    text_width = text_bbox[2] - text_bbox[0]
+    text_height = text_bbox[3] - text_bbox[1]
+
+    position = ((width - text_width) // 2, (height - text_height) // 2)
+
+    draw.text(position, text, font=font, fill=(0, 0, 0, 255))
 
     return img
 
@@ -76,6 +52,26 @@ def apply_displacement_map(text_img, clothing_img, strength=20):
 
     return text_warped
 
+def apply_displacement_map(text_img, clothing_img, strength=20):
+    """Apply displacement map to blend text onto clothing."""
+    gray = cv2.cvtColor(clothing_img, cv2.COLOR_BGR2GRAY)
+    grad_x = cv2.Sobel(gray, cv2.CV_32F, 1, 0, ksize=5)
+    grad_y = cv2.Sobel(gray, cv2.CV_32F, 0, 1, ksize=5)
+
+    grad_x = cv2.normalize(grad_x, None, 0, 1, cv2.NORM_MINMAX)
+    grad_y = cv2.normalize(grad_y, None, 0, 1, cv2.NORM_MINMAX)
+
+    displacement_map = np.zeros_like(clothing_img, dtype=np.float32)
+    displacement_map[:, :, 0] = grad_x * strength
+    displacement_map[:, :, 1] = grad_y * strength
+
+    text_warped = cv2.remap(text_img, 
+                            displacement_map[:, :, 0].astype(np.float32),
+                            displacement_map[:, :, 1].astype(np.float32), 
+                            interpolation=cv2.INTER_LINEAR)
+
+    return text_warped
+
 def overlay_text_on_clothing(clothing_image, text_input, style, strength=20, alpha=0.7, photo=None, blend_alpha=0.5):
     """Blend generated text and optional photo onto the clothing image."""
     clothing_img = cv2.imread(clothing_image)  # Read image from file path