Update app.py

app.py

@@ -16,19 +16,43 @@ 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.1)  # Adjust text to be proportional
+    font_size = int(min(width, height) * 0.2)  # Adjust text to be a larger proportion of the image
     try:
         font = ImageFont.truetype(font_path[style], size=font_size)
     except:
         font = ImageFont.load_default()
 
-    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))
+    # 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
 
     return img