Update app.py

app.py

@@ -1,11 +1,58 @@
+# interface.py
 import gradio as gr
 import subprocess
 import json
 import os
+import zipfile
+from PIL import ImageFont
+import matplotlib.font_manager
 
 # Path to the compiled engine executable
 ENGINE_EXECUTABLE = "./engine"  # File thực thi trên Linux
 
+def extract_and_load_fonts_from_directory(directory="fontfile", extract_to="extracted_fonts"):
+    if not os.path.exists(extract_to):
+        os.makedirs(extract_to)
+    
+    fonts = []
+    
+    # Extract fonts from zip files
+    for root, dirs, files in os.walk(directory):
+        for file in files:
+            if file.endswith(".zip"):
+                zip_path = os.path.join(root, file)
+                try:
+                    with zipfile.ZipFile(zip_path, 'r') as zip_ref:
+                        zip_ref.extractall(extract_to)
+                    print(f"Extracted: {zip_path}")
+                except Exception as e:
+                    print(f"Failed to extract {zip_path}: {e}")
+    
+    # Collect all .ttf and .shx fonts
+    for root, dirs, files in os.walk(extract_to):
+        for file in files:
+            if file.endswith(".ttf") or file.endswith(".shx"):
+                fonts.append(os.path.join(root, file))
+    
+    return fonts
+
+def get_system_fonts():
+    fonts = []
+    for font in matplotlib.font_manager.findSystemFonts(fontpaths=None, fontext='ttf'):
+        fonts.append(font)
+    return fonts
+
+def get_available_fonts():
+    # Get system fonts
+    system_fonts = get_system_fonts()
+    
+    # Extract and load custom fonts
+    extracted_fonts = extract_and_load_fonts_from_directory()
+    
+    # Combine and deduplicate fonts
+    all_fonts = list(set(system_fonts + extracted_fonts))
+    return sorted(all_fonts)
+
 def call_engine(input_text, font_size, width, height, bg_color, text_color, mode, font_name, align, line_spacing, image_format):
     # Prepare input data as a dictionary
     input_data = {
@@ -16,22 +63,23 @@ def call_engine(input_text, font_size, width, height, bg_color, text_color, mode
         "bg_color": bg_color,
         "text_color": text_color,
         "mode": mode,
-        "font_path": font_name,
+        "font_path": font_name,  # Pass the selected font path
         "align": align,
         "line_spacing": line_spacing,
         "image_format": image_format
     }
 
-    # Convert input data to JSON string
-    input_json = json.dumps(input_data)
-
     # Call the engine executable with input data
     result = subprocess.run(
-        [ENGINE_EXECUTABLE, input_json],
+        [ENGINE_EXECUTABLE, json.dumps(input_data)],
         capture_output=True,
         text=True
     )
 
+    # Handle errors
+    if result.returncode != 0:
+        raise Exception(f"Engine failed with error: {result.stderr}")
+
     # Get the output image path from stdout
     output_path = result.stdout.strip()
 
@@ -50,7 +98,7 @@ with gr.Blocks() as demo:
 
     with gr.Row():
         font_size = gr.Slider(10, 100, value=30, label="Font Size")
-        font_name = gr.Dropdown(choices=[], value="", label="Font")  # Populate dynamically
+        font_name = gr.Dropdown(choices=get_available_fonts(), value="", label="Font")
         align = gr.Radio(["Left", "Center", "Right"], label="Text Alignment", value="Center")
 
     with gr.Row():