Update app.py

app.py

@@ -28,8 +28,8 @@ def draw_lattice_grid(draw, width, height, spacing=100, color=(235, 235, 235)):
 
 def text_to_images_generator(text_content, style='lines', font_path=None):
     """
-    Converts a given string of text into a series of images and returns them
-    as a list of image file paths. This is compatible with both UI and MCP.
+    Converts a given string of text into a single combined image.
+    This is compatible with both UI and MCP.
 
     Args:
         text_content (str): The text to be converted.
@@ -37,12 +37,12 @@ def text_to_images_generator(text_content, style='lines', font_path=None):
         font_path (str, optional): The path to a .ttf font file.
 
     Returns:
-        list: A list of image file paths.
+        str: Path to the generated combined image file.
     """
     if not text_content or not text_content.strip():
-        # Return empty list and show a warning if there is no text
+        # Return None and show a warning if there is no text
         gr.Warning("Input text is empty. Please enter some text to generate images.")
-        return []
+        return None
 
     # --- Configuration ---
     IMG_WIDTH = 1080
@@ -112,10 +112,6 @@ def text_to_images_generator(text_content, style='lines', font_path=None):
             all_lines_and_breaks.append(None)
 
     # --- Image Generation ---
-    generated_image_paths = [] # Store image file paths
-    page_content = []
-    y_text = PADDING_Y
-    
     try:
         line_height = font.getbbox("A")[3] - font.getbbox("A")[1]
     except AttributeError:
@@ -123,10 +119,34 @@ def text_to_images_generator(text_content, style='lines', font_path=None):
     
     PARAGRAPH_SPACING = line_height
 
-    def create_image_page(content):
-        """Helper function to create a single image page and save it to a temporary file."""
-        img = Image.new('RGB', (IMG_WIDTH, IMG_HEIGHT), color=BACKGROUND_COLOR)
-        draw = ImageDraw.Draw(img)
+    # 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)
 
         if style == 'lines':
             line_style_spacing = line_height + LINE_SPACING
@@ -137,35 +157,22 @@ def text_to_images_generator(text_content, style='lines', font_path=None):
             draw_lattice_grid(draw, IMG_WIDTH, IMG_HEIGHT, color=STYLE_COLOR)
 
         current_y = PADDING_Y
-        for page_item in content:
+        for page_item in page_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
         
-        # Save image to temporary file
-        temp_file = tempfile.NamedTemporaryFile(delete=False, suffix='.png')
-        img.save(temp_file.name, format='PNG')
-        temp_file.close()
-        generated_image_paths.append(temp_file.name)
+        # Paste this page onto the combined image
+        combined_img.paste(page_img, (0, page_idx * IMG_HEIGHT))
 
-    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_image_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_image_page(page_content)
-
-    return generated_image_paths
+    # Save combined image to temporary file
+    temp_file = tempfile.NamedTemporaryFile(delete=False, suffix='.png')
+    combined_img.save(temp_file.name, format='PNG')
+    temp_file.close()
+    
+    return temp_file.name
 
 # --- Gradio Interface ---
 
@@ -179,9 +186,9 @@ demo = gr.Interface(
         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.Gallery(label="Generated Images", show_label=True, preview=True),
+    outputs=gr.Image(label="Generated Image", show_label=True),
     title="Text-to-Image Converter",
-    description="Transforms long-form text into a series of attractive, readable images. Paste your text, choose a style, and click 'Submit'. You can download the images from the gallery below.",
+    description="Transforms long-form text into a single combined image with multiple pages. Paste your text, choose a style, and click 'Submit'. You can download the image from below.",
     allow_flagging="never"
 )