Update app.py

app.py

@@ -9,6 +9,7 @@ import math
 import tempfile
 from pathlib import Path
 
+# A list of common fonts to prioritize
 COMMON_FONTS = [
     "Times New Roman",
     "Arial",
@@ -149,10 +150,10 @@ def render_math_image(text, font_size, width, height, bg_color, text_color):
     return img
 
 def text_to_image(input_text, font_size, width, height, bg_color, text_color, 
-                 mode, font_name, align, image_format, preview_mode):
+                  mode, font_name, align, image_format, preview_mode):
     if mode == "Plain Text":
         img = render_plain_text_image(input_text, font_size, width, height, 
-                                    bg_color, text_color, font_name, align)
+                                      bg_color, text_color, font_name, align)
     elif mode == "LaTeX Math":
         img = render_math_image(input_text, font_size, width, height, bg_color, text_color)
     else:
@@ -164,18 +165,18 @@ def text_to_image(input_text, font_size, width, height, bg_color, text_color,
         return save_image_to_file(img, image_format)
 
 def handle_file_upload(file, font_size, width, height, bg_color, text_color, 
-                      mode, font_name, align, image_format, preview_mode):
+                       mode, font_name, align, image_format, preview_mode):
     if file is not None:
         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, preview_mode)
+                             mode, font_name, align, image_format, preview_mode)
     return "No file uploaded!"
 
+# Retrieve the system fonts and choose a default font.
 font_list, FONT_PATHS = get_system_fonts()
-default_font = next((f for f in font_list if "times" in f.lower() 
-                    or "arial" in f.lower()), font_list[0])
+default_font = next((f for f in font_list if "times" in f.lower() or "arial" in f.lower()), font_list[0])
 
 with gr.Blocks() as demo:
     gr.Markdown("# 🖼️ Text to Image Converter")
@@ -201,30 +202,27 @@ 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")
         preview_mode = gr.Checkbox(label="Preview Mode", value=True, 
-                                 info="Uncheck to get download link instead of preview")
+                                   info="Uncheck to get download link instead of preview")
 
-    output = gr.Variable()
+    # Replace gr.Variable with gr.State for maintaining state
+    output = gr.State()
     preview_image = gr.Image(label="Preview", visible=True)
     download_link = gr.File(label="Download Image", visible=False)
 
+    # The update function now returns a tuple of two outputs.
     def update_output(result, preview_mode):
         if preview_mode:
-            return {
-                preview_image: result,
-                download_link: gr.update(visible=False),
-                preview_image: gr.update(visible=True)
-            }
+            # When in preview mode, show the image preview and hide the download link.
+            return result, gr.update(visible=False)
         else:
-            return {
-                download_link: result,
-                preview_image: gr.update(visible=False),
-                download_link: gr.update(visible=True)
-            }
+            # Otherwise, hide the preview and show the download link.
+            return gr.update(visible=False), result
 
     with gr.Row():
         convert_button = gr.Button("Convert Text to Image")
         file_convert_button = gr.Button("Convert File to Image")
 
+    # When clicking the convert button, first run text_to_image and then update the outputs.
     convert_button.click(
         text_to_image,
         inputs=[
@@ -238,6 +236,7 @@ with gr.Blocks() as demo:
         outputs=[preview_image, download_link]
     )
 
+    # For file upload conversion:
     file_convert_button.click(
         handle_file_upload,
         inputs=[
@@ -257,4 +256,4 @@ with gr.Blocks() as demo:
         outputs=[preview_image, download_link]
     )
 
-demo.launch()
+demo.launch()