Update app.py

app.py

@@ -3,17 +3,17 @@ import matplotlib.pyplot as plt
 from io import BytesIO
 from PIL import Image, ImageDraw, ImageFont
 
-# Function to parse color
-def parse_color(color):
-    """Convert RGBA string to RGB tuple if necessary."""
-    if isinstance(color, str) and color.startswith('rgba'):
-        color = color.replace('rgba', '').strip('()').split(',')
-        return tuple(int(float(c)) for c in color[:3])  # Convert to (R, G, B)
-    return color  # If already valid, return as is
-
-# Function to render plain text as an image
-def render_plain_text_image(text, font_size, width, height, bg_color, text_color):
-    """Convert plain text to an image with the given parameters."""
+# Function to convert text to an image
+def render_text_image(text, font_size, width, height, bg_color, text_color):
+    """Converts plain text to an image."""
+    
+    # Convert RGBA float colors to RGB tuples
+    def parse_color(color):
+        if isinstance(color, str) and "rgba" in color:
+            color = color.replace("rgba", "").strip("()").split(",")
+            return tuple(int(float(c)) for c in color[:3])  # Convert to (R, G, B)
+        return color
+
     bg_color = parse_color(bg_color)
     text_color = parse_color(text_color)
 
@@ -21,94 +21,85 @@ def render_plain_text_image(text, font_size, width, height, bg_color, text_color
     img = Image.new("RGB", (width, height), color=bg_color)
     draw = ImageDraw.Draw(img)
 
-    # Load font (default system font)
+    # Load a default font
     try:
         font = ImageFont.truetype("arial.ttf", font_size)
-    except:
+    except IOError:
         font = ImageFont.load_default()
 
     # Get text size
     text_width, text_height = draw.textsize(text, font=font)
 
-    # Position text in center
-    x = (width - text_width) // 2
-    y = (height - text_height) // 2
+    # Center text
+    text_x = (width - text_width) // 2
+    text_y = (height - text_height) // 2
 
-    draw.text((x, y), text, fill=text_color, font=font)
+    # Draw text on image
+    draw.text((text_x, text_y), text, font=font, fill=text_color)
 
     return img
 
-# Function to render math (LaTeX) as an image
+# Function to render LaTeX equations as an image
 def render_math_image(text, font_size, width, height, bg_color, text_color):
-    """Convert LaTeX-formatted text to an image."""
+    """Converts LaTeX math expressions into an image."""
     fig, ax = plt.subplots(figsize=(width / 100, height / 100))
     ax.set_axis_off()
 
-    # Ensure LaTeX formatting is correct
-    if "$" in text:
-        text = rf"{text}"
+    # Ensure proper LaTeX formatting
+    try:
+        formatted_text = rf"${text}$"
+    except Exception as e:
+        return f"Error parsing LaTeX: {e}"
 
-    ax.text(0.5, 0.5, text, fontsize=font_size, ha='center', va='center', color=text_color)
+    ax.text(0.5, 0.5, formatted_text, fontsize=font_size, ha='center', va='center', color=text_color)
 
     buf = BytesIO()
-    plt.savefig(buf, format='png', bbox_inches='tight', pad_inches=0.1)
+    plt.savefig(buf, format="png", bbox_inches="tight", pad_inches=0.1)
     plt.close(fig)
 
     buf.seek(0)
     return Image.open(buf)
 
-# Main function to handle input
-def text_to_image(input_text, font_size, width, height, bg_color, text_color, mode):
-    """Determine whether to use plain text or LaTeX rendering."""
-    if mode == "Plain Text":
-        return render_plain_text_image(input_text, font_size, width, height, bg_color, text_color)
-    elif mode == "LaTeX Math":
-        return render_math_image(input_text, font_size, width, height, bg_color, text_color)
+# File handler function
+def process_text_or_file(input_text, uploaded_file, font_size, width, height, bg_color, text_color):
+    """Handles text input or file upload to generate an image."""
+    
+    # Read text from uploaded file
+    if uploaded_file is not None:
+        text = uploaded_file.read().decode("utf-8")
     else:
-        return "Invalid mode selected!"
+        text = input_text
 
-# Function to handle file upload
-def handle_file_upload(file, font_size, width, height, bg_color, text_color, mode):
-    """Extract text from file and convert to an image."""
-    if file is not None:
-        text = file.read().decode("utf-8")  # Read and decode text file
-        return text_to_image(text, font_size, width, height, bg_color, text_color, mode)
-    return "No file uploaded!"
+    # Determine if the text is a LaTeX equation
+    if "$" in text or "\\" in text:
+        return render_math_image(text, font_size, width, height, bg_color, text_color)
+    else:
+        return render_text_image(text, font_size, width, height, bg_color, text_color)
 
-# Gradio Interface
+# Gradio UI
 with gr.Blocks() as demo:
-    gr.Markdown("# 🖼️ Text to Image Converter")
+    gr.Markdown("## 🖼️ Convert Text or Math to Image")
 
     with gr.Row():
-        input_text = gr.Textbox(label="Enter Text", placeholder="Type or paste text here...")
-        file_input = gr.File(label="Upload a Text File", type="file")
+        input_text = gr.Textbox(label="Enter Text or LaTeX", lines=3)
+        uploaded_file = gr.File(label="Upload a File (TXT, CSV)")
 
     with gr.Row():
-        font_size = gr.Slider(10, 100, value=30, label="Font Size")
-        width = gr.Slider(200, 1000, value=500, label="Image Width")
-        height = gr.Slider(200, 1000, value=300, label="Image Height")
+        font_size = gr.Slider(10, 100, value=30, step=2, label="Font Size")
+        width = gr.Slider(100, 1000, value=500, step=50, label="Image Width")
+        height = gr.Slider(100, 1000, value=200, step=50, label="Image Height")
 
     with gr.Row():
         bg_color = gr.ColorPicker(label="Background Color", value="#FFFFFF")
         text_color = gr.ColorPicker(label="Text Color", value="#000000")
 
-    mode = gr.Radio(["Plain Text", "LaTeX Math"], label="Rendering Mode", value="Plain Text")
-
-    output_image = gr.Image(label="Generated Image")
-
-    convert_button = gr.Button("Convert Text to Image")
-    file_convert_button = gr.Button("Convert File to Image")
-
-    convert_button.click(
-        text_to_image,
-        inputs=[input_text, font_size, width, height, bg_color, text_color, mode],
-        outputs=output_image
-    )
+    output_image = gr.Image(type="pil", label="Generated Image")
 
-    file_convert_button.click(
-        handle_file_upload,
-        inputs=[file_input, font_size, width, height, bg_color, text_color, mode],
+    gr.Button("Generate Image").click(
+        fn=process_text_or_file, 
+        inputs=[input_text, uploaded_file, font_size, width, height, bg_color, text_color],
         outputs=output_image
     )
 
+# Run Gradio app
 demo.launch()