Update app.py

app.py

@@ -25,21 +25,21 @@ def draw_lattice_grid(draw, width, height, spacing=100, color=(235, 235, 235)):
         draw.line([(0, y), (width, y)], fill=color, width=2)
 
 
-def text_to_base64_images_generator(text_content, style='lines'):
+def text_to_images_and_base64_generator(text_content, style='lines'):
     """
-    Converts text into a series of images and returns them as a list
-    of base64 encoded strings, suitable for an API/MCP client.
+    Converts text into images, returning both PIL objects for UI preview
+    and base64 strings for an API/MCP client.
 
     Args:
         text_content (str): The text to be converted.
         style (str, optional): The background style ('plain', 'lines', 'dots', 'grid').
 
     Returns:
-        list: A list of base64 encoded PNG image strings.
+        tuple: A tuple containing (list of PIL.Image.Image, list of base64 strings).
     """
     if not text_content or not text_content.strip():
         gr.Warning("Input text is empty. Please enter some text to generate images.")
-        return []
+        return [], []
 
     # --- Configuration ---
     IMG_WIDTH = 1080
@@ -104,6 +104,7 @@ def text_to_base64_images_generator(text_content, style='lines'):
             all_lines_and_breaks.append(None)
 
     # --- Image Generation & Encoding ---
+    generated_images = []
     generated_base64_list = []
     page_content = []
     y_text = PADDING_Y
@@ -136,7 +137,10 @@ def text_to_base64_images_generator(text_content, style='lines'):
             else:
                 current_y += PARAGRAPH_SPACING
         
-        # Encode image to base64 string
+        # Store the PIL image object for the gallery
+        generated_images.append(img)
+        
+        # Encode image to base64 string for the API
         buffered = io.BytesIO()
         img.save(buffered, format="PNG")
         img_base64 = base64.b64encode(buffered.getvalue()).decode("utf-8")
@@ -158,7 +162,7 @@ def text_to_base64_images_generator(text_content, style='lines'):
     if page_content:
         create_and_encode_page(page_content)
 
-    return generated_base64_list
+    return generated_images, generated_base64_list
 
 # --- Gradio Interface ---
 
@@ -167,14 +171,17 @@ 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_base64_images_generator,
+    fn=text_to_images_and_base64_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.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.",
+    outputs=[
+        gr.Gallery(label="Generated Images (Preview)", show_label=True, preview=True),
+        gr.JSON(label="API Output (Base64 Strings)")
+    ],
+    title="Text-to-Image Server",
+    description="Transforms long-form text into images. A gallery is shown for UI preview, and a JSON list of base64 strings is returned for API/MCP clients.",
     allow_flagging="never"
 )