Update app.py

app.py

@@ -1,120 +1,223 @@
 import gradio as gr
-from PIL import Image, ImageDraw, ImageFont
-import textwrap
-import io
 import os
+import zipfile
+import tempfile
+from PIL import Image, ImageDraw, ImageFont
+
+def draw_horizontal_lines(draw, width, height, spacing=60, color=(230, 230, 230)):
+    """Draws horizontal lines on the image."""
+    for y in range(0, height, spacing):
+        draw.line([(0, y), (width, y)], fill=color, width=2)
+
+def draw_dot_grid(draw, width, height, spacing=50, color=(220, 220, 220)):
+    """Draws a dot grid on the image."""
+    for x in range(0, width, spacing):
+        for y in range(0, height, spacing):
+            draw.ellipse([(x-2, y-2), (x+2, y+2)], fill=color)
+
+def draw_lattice_grid(draw, width, height, spacing=100, color=(235, 235, 235)):
+    """Draws a lattice/graph paper grid on the image."""
+    # Draw vertical lines
+    for x in range(0, width, spacing):
+        draw.line([(x, 0), (x, height)], fill=color, width=2)
+    # Draw horizontal lines
+    for y in range(0, height, spacing):
+        draw.line([(0, y), (width, y)], fill=color, width=2)
 
-def text_to_image(text_content, font_size_option):
+def text_to_images_mcp(text_content, style='lines', font_path=None):
     """
-    Convert text content to a formatted image with customizable font size.
+    Converts text to images and returns a zip file containing all generated images.
+    
     Args:
-        text_content (str): The input text to convert to image
-        font_size_option (str): Font size option (Small, Medium, Large)
+        text_content (str): The text to be converted.
+        style (str): The background style ('plain', 'lines', 'dots', 'grid').
+        font_path (str, optional): The path to a .ttf font file.
+    
     Returns:
-        PIL.Image: Generated image
+        str: Path to the zip file containing all generated images.
     """
     if not text_content.strip():
-        text_content = "Please enter some text to convert to image."
+        return None
     
-    # Font size mapping
-    font_sizes = {
-        "Small": 20,
-        "Medium": 28,
-        "Large": 36
-    }
-    font_size = font_sizes.get(font_size_option, 28)
+    # --- Configuration ---
+    IMG_WIDTH = 1080
+    IMG_HEIGHT = 1080
+    BACKGROUND_COLOR = (255, 255, 255)
+    TEXT_COLOR = (10, 10, 10)
+    STYLE_COLOR = (225, 225, 225)
     
-    # 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 = 50
-    line_spacing = font_size // 3  # Dynamic line spacing based on font size
+    PADDING_X = 80
+    PADDING_Y = 80
+    FONT_SIZE = 48
+    LINE_SPACING = 20
     
-    # Create image with higher quality
-    img = Image.new('RGB', (img_width, img_height), background_color)
-    draw = ImageDraw.Draw(img)
+    # Create temporary directory for images
+    temp_dir = tempfile.mkdtemp()
     
-    # Try to use a nice font with better quality, fallback to default if not available
+    # --- Font Loading ---
     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.load_default()
-        except:
-            # Fallback font
-            font = ImageFont.load_default()
-    
-    # Calculate text area with better precision
-    text_width = img_width - (padding * 2)
-    
-    # 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)
-            wrapped_lines.extend(lines)
-            wrapped_lines.append("")  # Add space between paragraphs
+    try:
+        if font_path and os.path.exists(font_path):
+            font = ImageFont.truetype(font_path, FONT_SIZE)
         else:
-            wrapped_lines.append("")
+            font_paths_to_try = [
+                "Arial.ttf", "arial.ttf",
+                "/System/Library/Fonts/Supplemental/Arial.ttf",
+                "/usr/share/fonts/truetype/liberation/LiberationSans-Regular.ttf",
+            ]
+            for f_path in font_paths_to_try:
+                try:
+                    font = ImageFont.truetype(f_path, FONT_SIZE)
+                    break
+                except IOError:
+                    continue
+        if not font:
+            font = ImageFont.load_default()
+    except Exception as e:
+        font = ImageFont.load_default()
+
+    # --- Text Wrapping Logic ---
+    drawable_width = IMG_WIDTH - 2 * PADDING_X
+    paragraphs = [p.strip() for p in text_content.strip().split('\n') if p.strip()]
+
+    if not paragraphs:
+        return None
+
+    all_lines_and_breaks = []
+    for i, paragraph in enumerate(paragraphs):
+        words = paragraph.split()
+        current_line = ""
+        for word in words:
+            if font.getlength(word) > drawable_width:
+                temp_word = ""
+                for char in word:
+                    if font.getlength(temp_word + char) > drawable_width:
+                        all_lines_and_breaks.append(temp_word)
+                        temp_word = char
+                    else:
+                        temp_word += char
+                word = temp_word
+            if font.getlength(current_line + " " + word) <= drawable_width:
+                current_line += " " + word
+            else:
+                all_lines_and_breaks.append(current_line.strip())
+                current_line = word
+        all_lines_and_breaks.append(current_line.strip())
+        if i < len(paragraphs) - 1:
+            all_lines_and_breaks.append(None)
+
+    # --- Image Generation ---
+    img_count = 0
+    page_content = []
+    y_text = PADDING_Y
     
-    # Remove trailing empty lines
-    while wrapped_lines and wrapped_lines[-1] == "":
-        wrapped_lines.pop()
+    try:
+        line_height = font.getbbox("A")[3] - font.getbbox("A")[1]
+    except AttributeError:
+        line_height = 12
     
-    # Draw text with better positioning
-    y_position = padding
-    for line in wrapped_lines:
-        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
+    PARAGRAPH_SPACING = line_height
+    image_paths = []
+
+    def create_image_page(content, page_num):
+        """Helper function to create and save a single image page."""
+        img = Image.new('RGB', (IMG_WIDTH, IMG_HEIGHT), color=BACKGROUND_COLOR)
+        draw = ImageDraw.Draw(img)
+
+        # Draw the selected background style first
+        if style == 'lines':
+            line_style_spacing = line_height + LINE_SPACING
+            draw_horizontal_lines(draw, IMG_WIDTH, IMG_HEIGHT, spacing=line_style_spacing, color=STYLE_COLOR)
+        elif style == 'dots':
+            draw_dot_grid(draw, IMG_WIDTH, IMG_HEIGHT, color=STYLE_COLOR)
+        elif style == 'grid':
+            draw_lattice_grid(draw, IMG_WIDTH, IMG_HEIGHT, color=STYLE_COLOR)
+
+        # Draw the text on top of the background
+        current_y = PADDING_Y
+        for page_item in content:
+            if page_item is not None:
+                draw.text((PADDING_X, current_y), page_item, font=font, fill=TEXT_COLOR)
+                current_y += line_height + LINE_SPACING
+            else:
+                current_y += PARAGRAPH_SPACING
         
-        # Use textbbox for better text positioning
-        draw.text((padding, y_position), line, font=font, fill=text_color)
-        y_position += font_size + line_spacing
+        filename = os.path.join(temp_dir, f"page_{page_num}.png")
+        img.save(filename)
+        image_paths.append(filename)
+        return filename
+
+    for item in all_lines_and_breaks:
+        is_break = item is None
+        item_height = PARAGRAPH_SPACING if is_break else line_height
+        
+        if y_text + item_height > IMG_HEIGHT - PADDING_Y:
+            img_count += 1
+            create_image_page(page_content, img_count)
+
+            page_content = [item]
+            y_text = PADDING_Y + item_height + (0 if is_break else LINE_SPACING)
+        else:
+            page_content.append(item)
+            y_text += item_height + (0 if is_break else LINE_SPACING)
+            
+    if page_content:
+        img_count += 1
+        create_image_page(page_content, img_count)
+
+    # Create zip file
+    zip_path = os.path.join(tempfile.gettempdir(), f"text_to_images_{len(os.listdir(temp_dir))}_pages.zip")
+    with zipfile.ZipFile(zip_path, 'w') as zipf:
+        for img_path in image_paths:
+            zipf.write(img_path, os.path.basename(img_path))
     
-    return img
+    return zip_path
+
+# Sample text for demonstration
+sample_text = """In the heart of a bustling city, there lived a clockmaker named Alistair. His shop, a quaint corner of tranquility amidst the urban chaos, was filled with the gentle ticking of countless timepieces. Each clock was a masterpiece, a testament to his dedication and skill. But Alistair held a secret. One of his clocks, an old grandfather clock in the corner, did not just tell time. It told stories.
+
+Every midnight, as the city slept, the clock would chime, not with bells, but with whispers of forgotten tales. Stories of ancient kings, lost love, and adventures in lands woven from starlight and dreams. Alistair would sit by the fire, listening, his heart filled with the magic of the past. He was the keeper of time, and in turn, time had made him its confidant.
+
+One day, a young girl with eyes as curious as a cat's wandered into his shop. She wasn't interested in the shiny new watches but was drawn to the grandfather clock. "What's its story?" she asked, her voice soft. Alistair smiled, for he knew he had found the next guardian of the stories. The legacy of the whispering clock would live on."""
 
 demo = gr.Interface(
-    fn=text_to_image,
+    fn=text_to_images_mcp,
     inputs=[
         gr.Textbox(
-            label="Text Content", 
-            placeholder="Enter your text content here...",
+            value=sample_text,
             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!"
+            max_lines=20,
+            label="Text Content",
+            placeholder="Enter your long-form text here..."
+        ),
+        gr.Dropdown(
+            choices=['plain', 'lines', 'dots', 'grid'],
+            value='lines',
+            label="Background Style",
+            info="Choose the background style for your images"
         ),
-        gr.Radio(
-            choices=["Small", "Medium", "Large"],
-            value="Medium",
-            label="Font Size"
+        gr.Textbox(
+            value="",
+            label="Custom Font Path (Optional)",
+            placeholder="/path/to/your/font.ttf",
+            info="Leave empty to use system default font"
+        )
+    ],
+    outputs=[
+        gr.File(
+            label="Download Images (ZIP)",
+            file_count="single"
         )
     ],
-    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. Choose your font size and generate high-quality images perfect for social media!"
+    title="📖 Text to Images Converter",
+    description="Transform long-form text into a series of attractive, readable images. Simply paste your text, choose a background style, and download the generated images as a ZIP file.",
+    examples=[
+        [sample_text, 'lines', ''],
+        [sample_text, 'dots', ''],
+        [sample_text, 'grid', ''],
+        [sample_text, 'plain', '']
+    ],
+    cache_examples=False
 )
 
 if __name__ == "__main__":