Update app.py

app.py

@@ -1,6 +1,5 @@
 import gradio as gr
 import os
-import zipfile
 import tempfile
 from PIL import Image, ImageDraw, ImageFont
 
@@ -26,7 +25,7 @@ def draw_lattice_grid(draw, width, height, spacing=100, color=(235, 235, 235)):
 
 def text_to_images_mcp(text_content, style='lines', font_path=None):
     """
-    Converts text to images and returns a zip file containing all generated images.
+    Converts text to images and returns a list of image objects.
     
     Args:
         text_content (str): The text to be converted.
@@ -34,10 +33,10 @@ def text_to_images_mcp(text_content, style='lines', font_path=None):
         font_path (str, optional): The path to a .ttf font file.
     
     Returns:
-        str: Path to the zip file containing all generated images.
+        list: List of PIL Image objects for display in Gradio.
     """
     if not text_content.strip():
-        return None
+        return []
     
     # --- Configuration ---
     IMG_WIDTH = 1080
@@ -51,9 +50,6 @@ def text_to_images_mcp(text_content, style='lines', font_path=None):
     FONT_SIZE = 48
     LINE_SPACING = 20
     
-    # Create temporary directory for images
-    temp_dir = tempfile.mkdtemp()
-    
     # --- Font Loading ---
     font = None
     try:
@@ -81,7 +77,7 @@ def text_to_images_mcp(text_content, style='lines', font_path=None):
     paragraphs = [p.strip() for p in text_content.strip().split('\n') if p.strip()]
 
     if not paragraphs:
-        return None
+        return []
 
     all_lines_and_breaks = []
     for i, paragraph in enumerate(paragraphs):
@@ -117,10 +113,10 @@ def text_to_images_mcp(text_content, style='lines', font_path=None):
         line_height = 12
     
     PARAGRAPH_SPACING = line_height
-    image_paths = []
+    generated_images = []
 
     def create_image_page(content, page_num):
-        """Helper function to create and save a single image page."""
+        """Helper function to create and return a single image page."""
         img = Image.new('RGB', (IMG_WIDTH, IMG_HEIGHT), color=BACKGROUND_COLOR)
         draw = ImageDraw.Draw(img)
 
@@ -142,10 +138,8 @@ def text_to_images_mcp(text_content, style='lines', font_path=None):
             else:
                 current_y += PARAGRAPH_SPACING
         
-        filename = os.path.join(temp_dir, f"page_{page_num}.png")
-        img.save(filename)
-        image_paths.append(filename)
-        return filename
+        generated_images.append(img)
+        return img
 
     for item in all_lines_and_breaks:
         is_break = item is None
@@ -165,13 +159,7 @@ def text_to_images_mcp(text_content, style='lines', font_path=None):
         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 zip_path
+    return generated_images
 
 # 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.
@@ -204,13 +192,19 @@ demo = gr.Interface(
         )
     ],
     outputs=[
-        gr.File(
-            label="Download Images (ZIP)",
-            file_count="single"
+        gr.Gallery(
+            label="Generated Images",
+            show_label=True,
+            elem_id="gallery",
+            columns=2,
+            rows=2,
+            object_fit="contain",
+            height="auto",
+            show_download_button=True
         )
     ],
     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.",
+    description="Transform long-form text into a series of attractive, readable images. Simply paste your text, choose a background style, and preview the generated images. You can download individual images by clicking on them.",
     examples=[
         [sample_text, 'lines', ''],
         [sample_text, 'dots', ''],