Update app.py

app.py

@@ -1,40 +1,59 @@
 import gradio as gr
 import numpy as np
-import cv2
 from PIL import Image, ImageDraw, ImageFont
+import os
+
+# Save fonts locally for faster access
+FONTS_DIR = "fonts"
+os.makedirs(FONTS_DIR, exist_ok=True)
 
-# List of available handwriting-style fonts
 FONTS = {
-    "Cursive": "https://github.com/google/fonts/raw/main/ofl/dancingscript/DancingScript-Regular.ttf",
-    "School Notebook": "https://github.com/google/fonts/raw/main/ofl/permanentmarker/PermanentMarker-Regular.ttf",
-    "Elegant Calligraphy": "https://github.com/google/fonts/raw/main/ofl/sacramento/Sacramento-Regular.ttf"
+    "Cursive": "DancingScript-Regular.ttf",
+    "School Notebook": "PermanentMarker-Regular.ttf",
+    "Elegant Calligraphy": "Sacramento-Regular.ttf"
+}
+
+# Download fonts once and store them locally
+FONT_URLS = {
+    "DancingScript-Regular.ttf": "https://github.com/google/fonts/raw/main/ofl/dancingscript/DancingScript-Regular.ttf",
+    "PermanentMarker-Regular.ttf": "https://github.com/google/fonts/raw/main/ofl/permanentmarker/PermanentMarker-Regular.ttf",
+    "Sacramento-Regular.ttf": "https://github.com/google/fonts/raw/main/ofl/sacramento/Sacramento-Regular.ttf"
 }
 
+for font_name, url in FONT_URLS.items():
+    font_path = os.path.join(FONTS_DIR, font_name)
+    if not os.path.exists(font_path):  # Download only if not already present
+        import requests
+        with open(font_path, "wb") as f:
+            f.write(requests.get(url).content)
+
 # Function to convert text to handwriting
 def text_to_handwriting(text, font_choice, font_size, font_color):
     if not text.strip():
         return None  # Return nothing if text is empty
 
-    # Load font
-    font_path = FONTS[font_choice]
-    font = ImageFont.truetype(font_path, font_size)
+    font_path = os.path.join(FONTS_DIR, FONTS[font_choice])
+    font = ImageFont.truetype(font_path, int(font_size))
 
     # Estimate text size
+    max_width, max_height = 600, 300  # Limit size to prevent slow rendering
     text_width, text_height = font.getsize_multiline(text)
-    
-    # Create a blank image
-    img = Image.new("RGB", (text_width + 40, text_height + 40), (255, 255, 255))
+
+    # Create image with limited size
+    img_width = min(text_width + 40, max_width)
+    img_height = min(text_height + 40, max_height)
+    img = Image.new("RGB", (img_width, img_height), "white")
     draw = ImageDraw.Draw(img)
-    
-    # Draw the text on the image
+
+    # Draw text
     draw.text((20, 20), text, font=font, fill=font_color)
 
     return img
 
 # Gradio UI
 with gr.Blocks() as demo:
-    gr.Markdown("# 🖋️ Text to Handwriting Converter")
-    gr.Markdown("Convert your typed text into **realistic handwriting styles!**")
+    gr.Markdown("# 🖋️ Text to Handwriting Converter (Fast Version)")
+    gr.Markdown("Convert your typed text into **realistic handwriting styles instantly!**")
 
     text_input = gr.Textbox(label="Enter Your Text", placeholder="Type something...")
     font_selector = gr.Dropdown(list(FONTS.keys()), label="Select Handwriting Style")
@@ -51,4 +70,4 @@ with gr.Blocks() as demo:
         outputs=output_image
     )
 
-demo.launch()
+demo.launch()