Update app.py

app.py

@@ -1,6 +1,5 @@
 import os
 import gradio as gr
-import tempfile
 from PIL import Image, ImageDraw, ImageFont
 
 # --- Image Generation Logic (from coverter.py) ---
@@ -28,8 +27,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 their file paths.
-    This version is adapted for Gradio to return image paths from a temporary directory.
+    Converts a given string of text into a series of images and returns them
+    as a list of PIL Image objects. This is compatible with both UI and MCP.
 
     Args:
         text_content (str): The text to be converted.
@@ -37,7 +36,7 @@ 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 file paths for the generated images.
+        list: A list of PIL.Image.Image objects.
     """
     if not text_content or not text_content.strip():
         # Return empty list and show a warning if there is no text
@@ -57,17 +56,12 @@ def text_to_images_generator(text_content, style='lines', font_path=None):
     FONT_SIZE = 48
     LINE_SPACING = 20
     
-    # Create a temporary directory to save images for the Gradio gallery
-    output_dir = tempfile.mkdtemp()
-
     # --- Font Loading ---
     font = None
     try:
-        # Prioritize provided font path if it exists
         if font_path and os.path.exists(font_path):
             font = ImageFont.truetype(font_path, FONT_SIZE)
         else:
-            # Otherwise, search for common system fonts
             font_paths_to_try = [
                 "Arial.ttf", "arial.ttf", "DejaVuSans.ttf",
                 "/System/Library/Fonts/Supplemental/Arial.ttf",
@@ -80,9 +74,8 @@ def text_to_images_generator(text_content, style='lines', font_path=None):
                     break
                 except IOError:
                     continue
-        # Fallback to default if no font is found
         if not font:
-            gr.Warning("Could not find a standard .ttf font. Falling back to the basic default font. Text may look pixelated.")
+            gr.Warning("Could not find a standard .ttf font. Falling back to the basic default font.")
             font = ImageFont.load_default()
     except Exception as e:
         print(f"An unexpected error occurred during font loading: {e}")
@@ -97,7 +90,6 @@ def text_to_images_generator(text_content, style='lines', font_path=None):
         words = paragraph.split()
         current_line = ""
         for word in words:
-            # Handle very long words by breaking them up
             if font.getlength(word) > drawable_width:
                 temp_word = ""
                 for char in word:
@@ -115,31 +107,26 @@ def text_to_images_generator(text_content, style='lines', font_path=None):
                 current_line = word
         all_lines_and_breaks.append(current_line.strip())
         
-        # Add a placeholder for paragraph breaks
         if i < len(paragraphs) - 1:
             all_lines_and_breaks.append(None)
 
     # --- Image Generation ---
-    generated_files = []
-    img_count = 0
+    generated_images = [] # Store PIL Image objects directly
     page_content = []
     y_text = PADDING_Y
     
     try:
-        # Get line height from font metrics if possible
         line_height = font.getbbox("A")[3] - font.getbbox("A")[1]
     except AttributeError:
-        # Fallback for default font
         line_height = 12
     
     PARAGRAPH_SPACING = line_height
 
-    def create_image_page(content, page_num):
-        """Helper function to create and save a single image page."""
+    def create_image_page(content):
+        """Helper function to create a single image page and return it."""
         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)
@@ -148,50 +135,39 @@ def text_to_images_generator(text_content, style='lines', font_path=None):
         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: # This is a paragraph break
+            else:
                 current_y += PARAGRAPH_SPACING
         
-        filename = os.path.join(output_dir, f"page_{page_num}.png")
-        img.save(filename)
-        generated_files.append(filename)
+        generated_images.append(img) # Add the PIL image to our list
 
-    # Iterate through all lines and create pages
     for item in all_lines_and_breaks:
         is_break = item is None
         item_height = PARAGRAPH_SPACING if is_break else line_height
         
-        # If adding the next line/break exceeds page height, create the current page
         if y_text + item_height > IMG_HEIGHT - PADDING_Y:
-            img_count += 1
-            create_image_page(page_content, img_count)
-            # Start a new page
+            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)
             
-    # Save the last page if it has content
     if page_content:
-        img_count += 1
-        create_image_page(page_content, img_count)
+        create_image_page(page_content)
 
-    return generated_files
+    return generated_images
 
 # --- Gradio Interface ---
 
-# Example text to pre-fill the textbox
 example_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.
 
 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."""
 
-# Define the Gradio interface
 demo = gr.Interface(
     fn=text_to_images_generator,
     inputs=[
@@ -206,6 +182,4 @@ demo = gr.Interface(
 
 # --- Main Execution ---
 if __name__ == "__main__":
-    # Launch the Gradio app as an MCP server
-    # The MCP server allows other applications to call this Gradio app as an API
     demo.launch(mcp_server=True)