Update app.py

app.py

@@ -25,14 +25,16 @@ 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_images_and_base64_generator(text_content, style='lines'):
+def text_to_images_and_base64_generator(text_content, style, font_size, bg_color_name):
     """
     Converts text into images, returning both PIL objects for UI preview
-    and base64 strings for an API/MCP client.
+    and base64 strings for an API/MCP client, with customizable options.
 
     Args:
         text_content (str): The text to be converted.
-        style (str, optional): The background style ('plain', 'lines', 'dots', 'grid').
+        style (str): The background style ('plain', 'lines', 'dots', 'grid').
+        font_size (int): The font size for the text.
+        bg_color_name (str): The name of the background color.
 
     Returns:
         tuple: A tuple containing (list of PIL.Image.Image, list of base64 strings).
@@ -41,16 +43,23 @@ def text_to_images_and_base64_generator(text_content, style='lines'):
         gr.Warning("Input text is empty. Please enter some text to generate images.")
         return [], []
 
+    # --- Color Mapping ---
+    color_map = {
+        "White": (255, 255, 255),
+        "Light Gray": (240, 240, 240),
+        "Beige": (245, 245, 220),
+        "Pale Blue": (220, 230, 245)
+    }
+    BACKGROUND_COLOR = color_map.get(bg_color_name, (255, 255, 255))
+
     # --- Configuration ---
     IMG_WIDTH = 1080
     IMG_HEIGHT = 1080
-    BACKGROUND_COLOR = (255, 255, 255)
     TEXT_COLOR = (10, 10, 10)
     STYLE_COLOR = (225, 225, 225)
     
     PADDING_X = 80
     PADDING_Y = 80
-    FONT_SIZE = 48
     LINE_SPACING = 20
     
     # --- Font Loading ---
@@ -64,12 +73,12 @@ def text_to_images_and_base64_generator(text_content, style='lines'):
         ]
         for f_path in font_paths_to_try:
             try:
-                font = ImageFont.truetype(f_path, FONT_SIZE)
+                font = ImageFont.truetype(f_path, font_size)
                 break
             except IOError:
                 continue
         if not font:
-            font = ImageFont.load_default()
+            font = ImageFont.load_default() # Note: Default font doesn't support size
     except Exception as e:
         print(f"Font loading error: {e}")
         font = ImageFont.load_default()
@@ -112,8 +121,10 @@ def text_to_images_and_base64_generator(text_content, style='lines'):
     try:
         line_height = font.getbbox("A")[3] - font.getbbox("A")[1]
     except AttributeError:
-        line_height = 12
-    
+        # Fallback for default font which has no getbbox
+        bbox = font.getmask("A").getbbox()
+        line_height = bbox[3] - bbox[1] if bbox else 12
+
     PARAGRAPH_SPACING = line_height
 
     def create_and_encode_page(content):
@@ -123,7 +134,7 @@ def text_to_images_and_base64_generator(text_content, style='lines'):
 
         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)
+            draw_horizontal_lines(draw, IMG_WIDTH, IMG_HEIGHT, spacing=int(line_style_spacing), color=STYLE_COLOR)
         elif style == 'dots':
             draw_dot_grid(draw, IMG_WIDTH, IMG_HEIGHT, color=STYLE_COLOR)
         elif style == 'grid':
@@ -137,10 +148,8 @@ def text_to_images_and_base64_generator(text_content, style='lines'):
             else:
                 current_y += PARAGRAPH_SPACING
         
-        # 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")
@@ -173,8 +182,10 @@ One day, a young girl with eyes as curious as a cat's wandered into his shop. Sh
 demo = gr.Interface(
     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')
+        gr.Textbox(lines=10, label="Text Content", placeholder="Paste your long-form text here...", value=example_text),
+        gr.Radio(['lines', 'dots', 'grid', 'plain'], label="Background Style", value='lines'),
+        gr.Slider(24, 72, value=48, step=2, label="Font Size"),
+        gr.Radio(["White", "Light Gray", "Beige", "Pale Blue"], label="Background Color", value="White")
     ],
     outputs=[
         gr.Gallery(label="Generated Images (Preview)", show_label=True, preview=True),