Update app.py

app.py

@@ -38,7 +38,6 @@ def render_plain_text_image(text, font_size, width, height, bg_color, text_color
     for line in text.split('\n'):
         lines.extend(textwrap.wrap(line, width=int(max_width / font_size * 1.8)))
 
-    # Use getbbox() instead of getsize()
     bbox = font.getbbox('Ay')
     line_height = bbox[3] - bbox[1]
     total_text_height = line_height * len(lines)
@@ -46,7 +45,6 @@ def render_plain_text_image(text, font_size, width, height, bg_color, text_color
     y = (height - total_text_height) // 2
 
     for line in lines:
-        # Use getbbox() to get line width
         bbox = font.getbbox(line)
         line_width = bbox[2] - bbox[0]
         
@@ -80,7 +78,8 @@ def render_math_image(text, font_size, width, height, bg_color, text_color):
     plt.close(fig)
 
     buf.seek(0)
-    return Image.open(buf)
+    img = Image.open(buf)
+    return img
 
 def text_to_image(input_text, font_size, width, height, bg_color, text_color, mode, font_name, align, image_format):
     if mode == "Plain Text":
@@ -89,11 +88,9 @@ def text_to_image(input_text, font_size, width, height, bg_color, text_color, mo
         img = render_math_image(input_text, font_size, width, height, bg_color, text_color)
     else:
         return "Invalid mode selected!"
-
-    buf = BytesIO()
-    img.save(buf, format=image_format)
-    buf.seek(0)
-    return buf
+    
+    # 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):
     if file is not None: