Update app.py

app.py

@@ -25,7 +25,7 @@ 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, font_size, bg_color_name):
+def text_to_images_and_base64_generator(text_content, style, font_size, bg_color_name, aspect_ratio_str):
     """
     Converts text into images, returning both PIL objects for UI preview
     and base64 strings for an API/MCP client, with customizable options.
@@ -35,6 +35,7 @@ def text_to_images_and_base64_generator(text_content, style, font_size, bg_color
         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.
+        aspect_ratio_str (str): The string representing the desired aspect ratio.
 
     Returns:
         tuple: A tuple containing (list of PIL.Image.Image, list of base64 strings).
@@ -43,21 +44,25 @@ def text_to_images_and_base64_generator(text_content, style, font_size, bg_color
         gr.Warning("Input text is empty. Please enter some text to generate images.")
         return [], []
 
-    # --- Color Mapping ---
+    # --- Aspect Ratio and Color Mapping ---
+    aspect_ratio_map = {
+        "1:1 (Square)": (1080, 1080),
+        "16:9 (Widescreen)": (1080, 608),
+        "9:16 (Vertical)": (1080, 1920),
+        "4:3 (Standard)": (1080, 810)
+    }
     color_map = {
         "White": (255, 255, 255),
         "Light Gray": (240, 240, 240),
         "Beige": (245, 245, 220),
         "Pale Blue": (220, 230, 245)
     }
+    IMG_WIDTH, IMG_HEIGHT = aspect_ratio_map.get(aspect_ratio_str, (1080, 1080))
     BACKGROUND_COLOR = color_map.get(bg_color_name, (255, 255, 255))
 
     # --- Configuration ---
-    IMG_WIDTH = 1080
-    IMG_HEIGHT = 1080
     TEXT_COLOR = (10, 10, 10)
     STYLE_COLOR = (225, 225, 225)
-    
     PADDING_X = 80
     PADDING_Y = 80
     LINE_SPACING = 20
@@ -185,7 +190,8 @@ demo = gr.Interface(
         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")
+        gr.Radio(["White", "Light Gray", "Beige", "Pale Blue"], label="Background Color", value="White"),
+        gr.Radio(["1:1 (Square)", "16:9 (Widescreen)", "9:16 (Vertical)", "4:3 (Standard)"], label="Aspect Ratio", value="1:1 (Square)")
     ],
     outputs=[
         gr.Gallery(label="Generated Images (Preview)", show_label=True, preview=True),