Update app.py

app.py

@@ -6,7 +6,6 @@ import textwrap
 import os
 import matplotlib
 
-# Function to get available system fonts
 def get_system_fonts():
     fonts = []
     for font in matplotlib.font_manager.findSystemFonts(fontpaths=None, fontext='ttf'):
@@ -14,61 +13,56 @@ def get_system_fonts():
         fonts.append(font_name)
     return sorted(set(fonts))
 
-# 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.strip())) for c in color[:3])  # Convert to (R, G, B)
-    return color  # If already valid, return as is
+        return tuple(int(float(c.strip())) for c in color[:3])
+    return color
 
-# Function to render plain text as an image
 def render_plain_text_image(text, font_size, width, height, bg_color, text_color, font_name, align):
-    """Convert plain text to an image with the given parameters."""
     bg_color = parse_color(bg_color)
     text_color = parse_color(text_color)
 
-    # Create blank image
     img = Image.new("RGB", (width, height), color=bg_color)
     draw = ImageDraw.Draw(img)
 
-    # Load font
     try:
         font_path = matplotlib.font_manager.findfont(font_name)
         font = ImageFont.truetype(font_path, font_size)
     except Exception:
         font = ImageFont.load_default()
 
-    # Text wrapping
     margin = 10
     max_width = width - 2 * margin
     lines = []
     for line in text.split('\n'):
         lines.extend(textwrap.wrap(line, width=int(max_width / font_size * 1.8)))
 
-    # Calculate total text height
-    line_height = font.getsize('Ay')[1]
+    # Use getbbox() instead of getsize()
+    bbox = font.getbbox('Ay')
+    line_height = bbox[3] - bbox[1]
     total_text_height = line_height * len(lines)
 
-    # Starting position
     y = (height - total_text_height) // 2
 
     for line in lines:
-        line_width = font.getsize(line)[0]
+        # Use getbbox() to get line width
+        bbox = font.getbbox(line)
+        line_width = bbox[2] - bbox[0]
+        
         if align == 'Left':
             x = margin
         elif align == 'Center':
             x = (width - line_width) // 2
         else:  # Right alignment
             x = width - line_width - margin
+            
         draw.text((x, y), line, fill=text_color, font=font)
         y += line_height
 
     return img
 
-# Function to render math (LaTeX) as an image
 def render_math_image(text, font_size, width, height, bg_color, text_color):
-    """Convert LaTeX-formatted text to an image."""
     bg_color = parse_color(bg_color)
     text_color = parse_color(text_color)
 
@@ -76,7 +70,6 @@ def render_math_image(text, font_size, width, height, bg_color, text_color):
     ax.set_facecolor(bg_color)
     ax.axis('off')
 
-    # Ensure LaTeX formatting is correct
     if not (text.startswith(r"$") and text.endswith(r"$")):
         text = rf"${text}$"
 
@@ -89,9 +82,7 @@ def render_math_image(text, font_size, width, height, bg_color, text_color):
     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, font_name, align, image_format):
-    """Determine whether to use plain text or LaTeX rendering."""
     if mode == "Plain Text":
         img = render_plain_text_image(input_text, font_size, width, height, bg_color, text_color, font_name, align)
     elif mode == "LaTeX Math":
@@ -99,28 +90,22 @@ def text_to_image(input_text, font_size, width, height, bg_color, text_color, mo
     else:
         return "Invalid mode selected!"
 
-    # Save image to buffer
     buf = BytesIO()
     img.save(buf, format=image_format)
     buf.seek(0)
     return buf
 
-# Function to handle file upload using file path
 def handle_file_upload(file, font_size, width, height, bg_color, text_color, mode, font_name, align, image_format):
-    """Extract text from file and convert to an image."""
     if file is not None:
-        # 'file' is a list of file paths; use the first file path.
         file_path = file[0]
         with open(file_path, "r", encoding="utf-8") as f:
             text = f.read()
         return text_to_image(text, font_size, width, height, bg_color, text_color, mode, font_name, align, image_format)
     return "No file uploaded!"
 
-# Get list of system fonts
 font_list = get_system_fonts()
 default_font = "DejaVuSans.ttf" if "DejaVuSans.ttf" in font_list else font_list[0]
 
-# Gradio Interface
 with gr.Blocks() as demo:
     gr.Markdown("# 🖼️ Text to Image Converter")
 
@@ -169,4 +154,4 @@ with gr.Blocks() as demo:
         outputs=output_image
     )
 
-demo.launch()
+demo.launch()