Update app.py

app.py

@@ -4,47 +4,73 @@ import textwrap
 import io
 import os
 
-def text_to_image(text_content):
+def text_to_image(text_content, font_size_option):
     """
-    Convert text content to a formatted image with fixed styling.
+    Convert text content to a formatted image with customizable font size.
     Args:
         text_content (str): The input text to convert to image
+        font_size_option (str): Font size option (Small, Medium, Large)
     Returns:
         PIL.Image: Generated image
     """
     if not text_content.strip():
         text_content = "Please enter some text to convert to image."
     
+    # Font size mapping
+    font_sizes = {
+        "Small": 20,
+        "Medium": 28,
+        "Large": 36
+    }
+    font_size = font_sizes.get(font_size_option, 28)
+    
     # Fixed styling parameters
     img_width = 600  # 3:4 ratio width
     img_height = 800  # 3:4 ratio height
     background_color = "#F8F9FA"  # Light theme background
     text_color = "#000000"  # Black text
-    padding = 40
-    font_size = 24  # Medium font size
-    line_spacing = 8
+    padding = 50
+    line_spacing = font_size // 3  # Dynamic line spacing based on font size
     
-    # Create image
+    # Create image with higher quality
     img = Image.new('RGB', (img_width, img_height), background_color)
     draw = ImageDraw.Draw(img)
     
-    # Try to use a nice font, fallback to default if not available
-    try:
-        font = ImageFont.truetype("/System/Library/Fonts/Arial.ttf", font_size)
-    except:
+    # Try to use a nice font with better quality, fallback to default if not available
+    font = None
+    font_paths = [
+        "/System/Library/Fonts/Helvetica.ttc",  # macOS
+        "/System/Library/Fonts/Arial.ttf",      # macOS
+        "/Windows/Fonts/arial.ttf",             # Windows
+        "/usr/share/fonts/truetype/liberation/LiberationSans-Regular.ttf",  # Linux
+        "arial.ttf"
+    ]
+    
+    for font_path in font_paths:
+        try:
+            font = ImageFont.truetype(font_path, font_size)
+            break
+        except:
+            continue
+    
+    if font is None:
         try:
-            font = ImageFont.truetype("arial.ttf", font_size)
+            font = ImageFont.load_default()
         except:
+            # Fallback font
             font = ImageFont.load_default()
     
-    # Calculate text area
+    # Calculate text area with better precision
     text_width = img_width - (padding * 2)
     
-    # Wrap text to fit width
-    chars_per_line = text_width // (font_size // 2)  # Rough estimate
-    wrapped_lines = []
+    # More accurate text wrapping based on actual character width
+    # Test character width
+    test_char_width = draw.textbbox((0, 0), "W", font=font)[2]
+    chars_per_line = max(1, text_width // test_char_width)
     
+    wrapped_lines = []
     paragraphs = text_content.split('\n')
+    
     for paragraph in paragraphs:
         if paragraph.strip():
             lines = textwrap.wrap(paragraph, width=chars_per_line)
@@ -57,14 +83,15 @@ def text_to_image(text_content):
     while wrapped_lines and wrapped_lines[-1] == "":
         wrapped_lines.pop()
     
-    # Draw text
+    # Draw text with better positioning
     y_position = padding
     for line in wrapped_lines:
-        if y_position + font_size > img_height - padding:
+        if y_position + font_size + line_spacing > img_height - padding:
             # Add "..." if text is too long
             draw.text((padding, y_position), "...", font=font, fill=text_color)
             break
         
+        # Use textbbox for better text positioning
         draw.text((padding, y_position), line, font=font, fill=text_color)
         y_position += font_size + line_spacing
     
@@ -72,15 +99,22 @@ def text_to_image(text_content):
 
 demo = gr.Interface(
     fn=text_to_image,
-    inputs=[gr.Textbox(
-        label="Text Content", 
-        placeholder="Enter your text content here...",
-        lines=10,
-        value="Welcome to Text-to-Image Converter!\n\nThis tool converts your text into beautifully formatted images perfect for social media posts.\n\nSimply replace this text with your content and click generate!"
-    )],
+    inputs=[
+        gr.Textbox(
+            label="Text Content", 
+            placeholder="Enter your text content here...",
+            lines=10,
+            value="Welcome to Text-to-Image Converter!\n\nThis tool converts your text into beautifully formatted images perfect for social media posts.\n\nSimply replace this text with your content, choose your preferred font size, and click generate!"
+        ),
+        gr.Radio(
+            choices=["Small", "Medium", "Large"],
+            value="Medium",
+            label="Font Size"
+        )
+    ],
     outputs=[gr.Image(label="Generated Image", type="pil")],
     title="Text to Image Converter",
-    description="Convert your text content into attractive images with a clean, readable design. Perfect for social media posts and content sharing!"
+    description="Convert your text content into attractive images with a clean, readable design. Choose your font size and generate high-quality images perfect for social media!"
 )
 
 if __name__ == "__main__":