Update app.py

app.py

@@ -12,7 +12,7 @@ from huggingface_hub import HfApi, HfFolder
 # Hugging Face repo and token
 HF_REPO = "ArrcttacsrjksX/Texttoimage"
 HF_ENGINE_URL = "https://huggingface.co/ArrcttacsrjksX/Texttoimage/resolve/main/engine"
-HF_TOKEN = os.getenv("HF_TOKEN")  # Get token from environment variable
+HF_TOKEN = os.getenv("HF_TOKEN")  # Lấy token từ biến môi trường
 ENGINE_EXECUTABLE = "./engine"
 
 def download_engine():
@@ -29,12 +29,10 @@ def download_engine():
             raise Exception("Failed to download engine")
 
 def ensure_executable(file_path):
-    """Ensure the file is executable."""
     if not os.access(file_path, os.X_OK):
         os.chmod(file_path, os.stat(file_path).st_mode | stat.S_IXUSR)
 
 def extract_and_load_fonts(directory="fontfile", extract_to="extracted_fonts"):
-    """Extract fonts from ZIP files and load them."""
     if not os.path.exists(extract_to):
         os.makedirs(extract_to)
     fonts = []
@@ -54,19 +52,11 @@ def extract_and_load_fonts(directory="fontfile", extract_to="extracted_fonts"):
     return fonts
 
 def get_system_fonts():
-    """Get system-installed fonts."""
     return matplotlib.font_manager.findSystemFonts(fontpaths=None, fontext='ttf')
 
 def get_available_fonts():
-    """Get a prioritized list of available fonts."""
     system_fonts = get_system_fonts()
     extracted_fonts = extract_and_load_fonts()
-    # Prioritize fonts with good Unicode support
-    preferred_fonts = ["DejaVuSans.ttf", "Arial.ttf", "LiberationSans-Regular.ttf"]
-    for font in preferred_fonts:
-        for path in system_fonts + extracted_fonts:
-            if font in path:
-                return [path] + sorted(set(system_fonts + extracted_fonts))
     return sorted(set(system_fonts + extracted_fonts))
 
 def upload_to_huggingface(file_path, text_content, timestamp_folder):
@@ -81,17 +71,10 @@ def upload_to_huggingface(file_path, text_content, timestamp_folder):
     os.remove("temp_text.txt")
 
 def call_engine(file_input, input_text, font_size, width, height, bg_color, text_color, mode, font_name, align, line_spacing, image_format):
-    """Call the engine to generate an image."""
     download_engine()
     ensure_executable(ENGINE_EXECUTABLE)
     if file_input:
         input_text = read_file_content(file_input)
-
-    # Validate LaTeX math syntax
-    if mode == "LaTeX Math":
-        if not input_text.startswith("$") or not input_text.endswith("$"):
-            input_text = f"${input_text}$"
-
     input_data = {
         "input_text": input_text,
         "font_size": font_size,
@@ -155,4 +138,4 @@ with gr.Blocks() as demo:
         outputs=output_image
     )
 
-demo.launch()
+demo.launch()