Update app.py

app.py

@@ -5,6 +5,7 @@ from PIL import Image, ImageDraw, ImageFont
 import textwrap
 import os
 import matplotlib
+import math
 
 def get_system_fonts():
     fonts = []
@@ -19,32 +20,29 @@ def parse_color(color):
         return tuple(int(float(c.strip())) for c in color[:3])
     return color
 
-def render_plain_text_image(text, font_size, width, height, bg_color, text_color, font_name, align):
-    bg_color = parse_color(bg_color)
-    text_color = parse_color(text_color)
-
-    img = Image.new("RGB", (width, height), color=bg_color)
-    draw = ImageDraw.Draw(img)
-
-    try:
-        font_path = matplotlib.font_manager.findfont(font_name)
-        font = ImageFont.truetype(font_path, font_size)
-    except Exception:
-        font = ImageFont.load_default()
-
-    margin = 10
-    max_width = width - 2 * margin
+def calculate_text_dimensions(text, font, max_width, margin):
     lines = []
     for line in text.split('\n'):
-        lines.extend(textwrap.wrap(line, width=int(max_width / font_size * 1.8)))
-
+        lines.extend(textwrap.wrap(line, width=int(max_width / font.size * 1.8)))
+    
     bbox = font.getbbox('Ay')
     line_height = bbox[3] - bbox[1]
-    total_text_height = line_height * len(lines)
-
-    y = (height - total_text_height) // 2
+    total_height = line_height * len(lines)
+    
+    return lines, line_height, total_height
 
-    for line in lines:
+def create_text_segment(lines, start_idx, max_lines, width, height, bg_color, text_color, font, align, margin):
+    img = Image.new("RGB", (width, height), color=bg_color)
+    draw = ImageDraw.Draw(img)
+    
+    bbox = font.getbbox('Ay')
+    line_height = bbox[3] - bbox[1]
+    
+    y = margin
+    end_idx = min(start_idx + max_lines, len(lines))
+    segment_lines = lines[start_idx:end_idx]
+    
+    for line in segment_lines:
         bbox = font.getbbox(line)
         line_width = bbox[2] - bbox[0]
         
@@ -57,8 +55,49 @@ def render_plain_text_image(text, font_size, width, height, bg_color, text_color
             
         draw.text((x, y), line, fill=text_color, font=font)
         y += line_height
+    
+    return img, end_idx
 
-    return img
+def render_plain_text_image(text, font_size, width, height, bg_color, text_color, font_name, align):
+    bg_color = parse_color(bg_color)
+    text_color = parse_color(text_color)
+    margin = 10
+
+    try:
+        font_path = matplotlib.font_manager.findfont(font_name)
+        font = ImageFont.truetype(font_path, font_size)
+    except Exception:
+        font = ImageFont.load_default()
+
+    # Calculate dimensions
+    max_width = width - 2 * margin
+    lines, line_height, total_text_height = calculate_text_dimensions(text, font, max_width, margin)
+    
+    # Calculate how many lines can fit in one segment
+    max_lines_per_segment = (height - 2 * margin) // line_height
+    
+    # Calculate number of segments needed
+    num_segments = math.ceil(len(lines) / max_lines_per_segment)
+    
+    # Create segments
+    segments = []
+    current_line = 0
+    
+    for i in range(num_segments):
+        segment_img, current_line = create_text_segment(
+            lines, current_line, max_lines_per_segment,
+            width, height, bg_color, text_color, font, align, margin
+        )
+        segments.append(segment_img)
+    
+    # Combine all segments vertically
+    total_height = len(segments) * height
+    final_image = Image.new("RGB", (width, total_height), color=bg_color)
+    
+    for i, segment in enumerate(segments):
+        final_image.paste(segment, (0, i * height))
+    
+    return final_image
 
 def render_math_image(text, font_size, width, height, bg_color, text_color):
     bg_color = parse_color(bg_color)
@@ -89,7 +128,6 @@ def text_to_image(input_text, font_size, width, height, bg_color, text_color, mo
     else:
         return "Invalid mode selected!"
     
-    # Return the PIL Image directly instead of BytesIO
     return img
 
 def handle_file_upload(file, font_size, width, height, bg_color, text_color, mode, font_name, align, image_format):
@@ -107,7 +145,7 @@ with gr.Blocks() as demo:
     gr.Markdown("# 🖼️ Text to Image Converter")
 
     with gr.Row():
-        input_text = gr.Textbox(label="Enter Text", placeholder="Type or paste text here...")
+        input_text = gr.Textbox(label="Enter Text", placeholder="Type or paste text here...", lines=5)
         file_input = gr.File(label="Upload a Text File", type="filepath")
 
     with gr.Row():
@@ -117,7 +155,7 @@ with gr.Blocks() as demo:
 
     with gr.Row():
         width = gr.Slider(200, 2000, value=800, label="Image Width")
-        height = gr.Slider(200, 2000, value=600, label="Image Height")
+        height = gr.Slider(200, 2000, value=600, label="Base Height")
 
     with gr.Row():
         bg_color = gr.ColorPicker(label="Background Color", value="#FFFFFF")