Update app.py

app.py

@@ -46,7 +46,7 @@ def calculate_text_dimensions(text, font, max_width, margin):
     
     return lines, line_height, total_height
 
-def create_text_segment(lines, start_idx, max_lines, width, height, bg_color, text_color, font, align, margin):
+def create_text_segment(lines, start_idx, max_lines, width, height, bg_color, text_color, font, align, margin, line_spacing):
     img = Image.new("RGB", (width, height), color=bg_color)
     draw = ImageDraw.Draw(img)
     
@@ -69,11 +69,11 @@ def create_text_segment(lines, start_idx, max_lines, width, height, bg_color, te
             x = width - line_width - margin
             
         draw.text((x, y), line, fill=text_color, font=font)
-        y += line_height
+        y += line_height * line_spacing  # Adjust line spacing here
     
     return img, end_idx
 
-def render_plain_text_image(text, font_size, width, height, bg_color, text_color, font_path, align):
+def render_plain_text_image(text, font_size, width, height, bg_color, text_color, font_path, align, line_spacing):
     bg_color = parse_color(bg_color)
     text_color = parse_color(text_color)
     margin = 10
@@ -88,7 +88,7 @@ def render_plain_text_image(text, font_size, width, height, bg_color, text_color
     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
+    max_lines_per_segment = (height - 2 * margin) // (line_height * line_spacing)
     
     # Calculate number of segments needed
     num_segments = math.ceil(len(lines) / max_lines_per_segment)
@@ -100,7 +100,7 @@ def render_plain_text_image(text, font_size, width, height, bg_color, text_color
     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
+            width, height, bg_color, text_color, font, align, margin, line_spacing
         )
         segments.append(segment_img)
     
@@ -134,9 +134,9 @@ def render_math_image(text, font_size, width, height, bg_color, text_color):
     img = Image.open(buf)
     return img
 
-def text_to_image(input_text, font_size, width, height, bg_color, text_color, mode, font_path, align, image_format):
+def text_to_image(input_text, font_size, width, height, bg_color, text_color, mode, font_path, align, line_spacing, image_format):
     if mode == "Plain Text":
-        img = render_plain_text_image(input_text, font_size, width, height, bg_color, text_color, font_path, align)
+        img = render_plain_text_image(input_text, font_size, width, height, bg_color, text_color, font_path, align, line_spacing)
     elif mode == "LaTeX Math":
         img = render_math_image(input_text, font_size, width, height, bg_color, text_color)
     else:
@@ -173,6 +173,9 @@ with gr.Blocks() as demo:
         mode = gr.Radio(["Plain Text", "LaTeX Math"], label="Rendering Mode", value="Plain Text")
         image_format = gr.Radio(["PNG", "JPEG"], label="Image Format", value="PNG")
 
+    # Add line spacing slider
+    line_spacing = gr.Slider(1.0, 3.0, value=1.2, step=0.1, label="Line Spacing")
+
     output_image = gr.Image(label="Generated Image")
 
     with gr.Row():
@@ -183,7 +186,7 @@ with gr.Blocks() as demo:
         text_to_image,
         inputs=[
             input_text, font_size, width, height, bg_color, text_color,
-            mode, font_name, align, image_format
+            mode, font_name, align, line_spacing, image_format
         ],
         outputs=output_image
     )