Update app.py

app.py

@@ -1,10 +1,10 @@
 import os
-import tempfile
-import base64
 import gradio as gr
 from PIL import Image, ImageDraw, ImageFont
+import base64
+import io
 
-# --- Image Generation Logic (from coverter.py) ---
+# --- Image Generation Logic ---
 
 def draw_horizontal_lines(draw, width, height, spacing=60, color=(230, 230, 230)):
     """Draws horizontal lines on the image."""
@@ -19,39 +19,37 @@ def draw_dot_grid(draw, width, height, spacing=50, color=(220, 220, 220)):
 
 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_images_generator(text_content, style='lines'):
+def text_to_base64_images_generator(text_content, style='lines'):
     """
-    Converts a given string of text into a single combined image and returns the file path.
-    This is compatible with both UI and MCP.
+    Converts text into a series of images and returns them as a list
+    of base64 encoded strings, suitable for an API/MCP client.
 
     Args:
         text_content (str): The text to be converted.
-        style (str): The background style ('plain', 'lines', 'dots', 'grid').
+        style (str, optional): The background style ('plain', 'lines', 'dots', 'grid').
 
     Returns:
-        str: Message with the path to the generated combined image file.
+        list: A list of base64 encoded PNG image strings.
     """
     if not text_content or not text_content.strip():
-        return "Error: Input text is empty. Please enter some text to generate images."
+        gr.Warning("Input text is empty. Please enter some text to generate images.")
+        return []
 
     # --- Configuration ---
     IMG_WIDTH = 1080
     IMG_HEIGHT = 1080
     BACKGROUND_COLOR = (255, 255, 255)
     TEXT_COLOR = (10, 10, 10)
-    STYLE_COLOR = (225, 225, 225) # Color for lines/dots/grid
+    STYLE_COLOR = (225, 225, 225)
     
     PADDING_X = 80
     PADDING_Y = 80
-
     FONT_SIZE = 48
     LINE_SPACING = 20
     
@@ -73,10 +71,10 @@ def text_to_images_generator(text_content, style='lines'):
         if not font:
             font = ImageFont.load_default()
     except Exception as e:
-        print(f"An unexpected error occurred during font loading: {e}")
+        print(f"Font loading error: {e}")
         font = ImageFont.load_default()
 
-    # --- Text Wrapping Logic ---
+    # --- Text Wrapping ---
     drawable_width = IMG_WIDTH - 2 * PADDING_X
     paragraphs = [p.strip() for p in text_content.strip().split('\n') if p.strip()]
     
@@ -105,7 +103,11 @@ def text_to_images_generator(text_content, style='lines'):
         if i < len(paragraphs) - 1:
             all_lines_and_breaks.append(None)
 
-    # --- Image Generation ---
+    # --- Image Generation & Encoding ---
+    generated_base64_list = []
+    page_content = []
+    y_text = PADDING_Y
+    
     try:
         line_height = font.getbbox("A")[3] - font.getbbox("A")[1]
     except AttributeError:
@@ -113,34 +115,10 @@ def text_to_images_generator(text_content, style='lines'):
     
     PARAGRAPH_SPACING = line_height
 
-    # Calculate pages and total height needed
-    pages_content = []
-    current_page = []
-    y_text = PADDING_Y
-
-    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:
-            pages_content.append(current_page)
-            current_page = [item]
-            y_text = PADDING_Y + item_height + (0 if is_break else LINE_SPACING)
-        else:
-            current_page.append(item)
-            y_text += item_height + (0 if is_break else LINE_SPACING)
-            
-    if current_page:
-        pages_content.append(current_page)
-
-    # Create a single combined image with all pages
-    total_height = len(pages_content) * IMG_HEIGHT
-    combined_img = Image.new('RGB', (IMG_WIDTH, total_height), color=BACKGROUND_COLOR)
-    
-    for page_idx, page_content in enumerate(pages_content):
-        # Create individual page
-        page_img = Image.new('RGB', (IMG_WIDTH, IMG_HEIGHT), color=BACKGROUND_COLOR)
-        draw = ImageDraw.Draw(page_img)
+    def create_and_encode_page(content):
+        """Helper to create a PIL image, draw text, and encode to base64."""
+        img = Image.new('RGB', (IMG_WIDTH, IMG_HEIGHT), color=BACKGROUND_COLOR)
+        draw = ImageDraw.Draw(img)
 
         if style == 'lines':
             line_style_spacing = line_height + LINE_SPACING
@@ -151,43 +129,36 @@ def text_to_images_generator(text_content, style='lines'):
             draw_lattice_grid(draw, IMG_WIDTH, IMG_HEIGHT, color=STYLE_COLOR)
 
         current_y = PADDING_Y
-        for page_item in page_content:
+        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
         
-        # Paste this page onto the combined image
-        combined_img.paste(page_img, (0, page_idx * IMG_HEIGHT))
-
-    # Save combined image to temporary file and also encode as base64
-    temp_file = tempfile.NamedTemporaryFile(delete=False, suffix='.png')
-    combined_img.save(temp_file.name, format='PNG')
-    temp_file.close()
-    
-    # Convert to base64 for inline viewing
-    import io
-    img_buffer = io.BytesIO()
-    combined_img.save(img_buffer, format='PNG')
-    img_buffer.seek(0)
-    img_base64 = base64.b64encode(img_buffer.getvalue()).decode('utf-8')
-    
-    return f"""Image successfully generated and saved to: {temp_file.name}
-
-Image details:
-- Total pages: {len(pages_content)}
-- Image dimensions: {IMG_WIDTH} x {total_height} pixels
-- Style: {style}
-- Format: PNG
+        # Encode image to base64 string
+        buffered = io.BytesIO()
+        img.save(buffered, format="PNG")
+        img_base64 = base64.b64encode(buffered.getvalue()).decode("utf-8")
+        generated_base64_list.append(img_base64)
 
-Base64 encoded image (you can copy this and paste into a base64 image viewer):
-data:image/png;base64,{img_base64}
+    # --- Page Creation Loop ---
+    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:
+            create_and_encode_page(page_content)
+            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:
+        create_and_encode_page(page_content)
 
-To view the image:
-1. Copy the entire data:image/png;base64,... string above
-2. Paste it into your browser's address bar, or
-3. Use an online base64 image viewer like: https://base64.guru/converter/decode/image"""
+    return generated_base64_list
 
 # --- Gradio Interface ---
 
@@ -196,14 +167,14 @@ example_text = """In the heart of a bustling city, there lived a clockmaker name
 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 in the corner. "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_images_generator,
+    fn=text_to_base64_images_generator,
     inputs=[
         gr.Textbox(lines=15, label="Text Content", placeholder="Paste your long-form text here...", value=example_text),
         gr.Radio(['lines', 'dots', 'grid', 'plain'], label="Background Style", value='lines')
     ],
-    outputs=gr.Textbox(label="Result", show_label=True),
-    title="Text-to-Image Converter",
-    description="Transforms long-form text into a single combined image with multiple pages. Paste your text, choose a style, and click 'Submit'. The result will show the file path where your image was saved.",
+    outputs=gr.JSON(label="Base64 Encoded Images"),
+    title="Text-to-Image API Server",
+    description="Transforms long-form text into a series of images and returns a JSON list of base64-encoded strings.",
     allow_flagging="never"
 )