Update app.py

app.py

@@ -1,16 +1,19 @@
-import gradio as gr
 import os
-import shutil
 import json
+import shutil
+import gradio as gr
 
 from fonts import setup_fonts
 from converter import convert_qurancaption_to_qvm
-from background import generate_background
 from subtitle import generate_ass
+from background import generate_background
 from renderer import render_final_video
 
 
-# download font saat startup
+# --------------------------------------------------
+# Startup
+# --------------------------------------------------
+
 setup_fonts()
 
 os.makedirs("uploads", exist_ok=True)
@@ -18,6 +21,41 @@ os.makedirs("temp", exist_ok=True)
 os.makedirs("outputs", exist_ok=True)
 
 
+# --------------------------------------------------
+# Fonts
+# --------------------------------------------------
+
+def get_available_fonts():
+
+    fonts = []
+
+    if os.path.exists("fonts"):
+
+        for f in os.listdir("fonts"):
+
+            if (
+                f.endswith(".ttf")
+                or f.endswith(".otf")
+            ):
+
+                fonts.append(
+                    f.replace(".ttf", "")
+                     .replace(".otf", "")
+                )
+
+    if not fonts:
+
+        fonts = [
+            "NotoNaskhArabic-Regular"
+        ]
+
+    return sorted(fonts)
+
+
+# --------------------------------------------------
+# Main Process
+# --------------------------------------------------
+
 def process(
     images,
     audio,
@@ -25,50 +63,75 @@ def process(
     arabic_font
 ):
 
-    # cleanup
-    for folder in ["uploads", "temp"]:
+    # clean temp folders
+    for folder in [
+        "uploads",
+        "temp"
+    ]:
 
         if os.path.exists(folder):
             shutil.rmtree(folder)
 
         os.makedirs(folder)
 
+    # --------------------
+    # Images
+    # --------------------
+
     image_paths = []
 
-    # save uploaded images
     for i, img in enumerate(images):
 
         ext = os.path.splitext(
             img.name
         )[1]
 
-        dst = f"uploads/{i:03d}{ext}"
+        dst = (
+            f"uploads/{i:03d}{ext}"
+        )
 
         shutil.copy(
             img.name,
             dst
         )
 
-        image_paths.append(dst)
+        image_paths.append(
+            dst
+        )
 
-    # save audio
-    audio_path = "uploads/audio.mp3"
+    # --------------------
+    # Audio
+    # --------------------
+
+    audio_path = (
+        "uploads/audio.mp3"
+    )
 
     shutil.copy(
         audio,
         audio_path
     )
 
-    # save json
-    json_path = "uploads/timing.json"
+    # --------------------
+    # JSON
+    # --------------------
+
+    json_path = (
+        "uploads/timing.json"
+    )
 
     shutil.copy(
         timing_json.name,
         json_path
     )
 
-    # convert QuranCaption
-    qvm_json = "temp/qvm_timing.json"
+    # --------------------
+    # Convert QuranCaption
+    # --------------------
+
+    qvm_json = (
+        "temp/qvm_timing.json"
+    )
 
     convert_qurancaption_to_qvm(
         json_path,
@@ -95,16 +158,26 @@ def process(
             preview[0]
         )
 
-    # generate background video
-    background_path = "temp/background.mp4"
+    # --------------------
+    # Background Video
+    # --------------------
+
+    background_path = (
+        "temp/background.mp4"
+    )
 
     generate_background(
         image_paths,
         background_path
     )
 
-    # generate subtitle ass
-    subtitle_path = "temp/subtitles.ass"
+    # --------------------
+    # ASS Subtitle
+    # --------------------
+
+    subtitle_path = (
+        "temp/subtitles.ass"
+    )
 
     generate_ass(
         qvm_json,
@@ -112,8 +185,13 @@ def process(
         arabic_font
     )
 
-    # render final video
-    final_video = "outputs/final.mp4"
+    # --------------------
+    # Final Render
+    # --------------------
+
+    final_video = (
+        "outputs/final.mp4"
+    )
 
     render_final_video(
         background_path,
@@ -125,6 +203,10 @@ def process(
     return final_video
 
 
+# --------------------------------------------------
+# UI
+# --------------------------------------------------
+
 with gr.Blocks() as demo:
 
     gr.Markdown(
@@ -141,16 +223,14 @@ Generate cinematic Quran video automatically.
 """
     )
 
-    with gr.Row():
-
-        images = gr.Files(
-            label="Upload Images",
-            file_count="multiple"
-        )
+    images = gr.Files(
+        label="Upload Images",
+        file_count="multiple"
+    )
 
     audio = gr.Audio(
         type="filepath",
-        label="Upload Audio"
+        label="Upload Audio MP3"
     )
 
     timing_json = gr.File(
@@ -158,25 +238,21 @@ Generate cinematic Quran video automatically.
     )
 
     arabic_font = gr.Dropdown(
-    [
-        "Noto Naskh Arabic",
-        "Noto Kufi Arabic",
-        "Noto Sans Arabic"
-    ],
-    value="Noto Naskh Arabic",
-    label="Arabic Font"
+        choices=get_available_fonts(),
+        value=get_available_fonts()[0],
+        label="Arabic Font"
     )
 
-    btn = gr.Button(
+    generate_btn = gr.Button(
         "Generate Video",
         variant="primary"
     )
 
-    output = gr.Video(
+    output_video = gr.Video(
         label="Result"
     )
 
-    btn.click(
+    generate_btn.click(
         fn=process,
         inputs=[
             images,
@@ -184,7 +260,8 @@ Generate cinematic Quran video automatically.
             timing_json,
             arabic_font
         ],
-        outputs=output
+        outputs=output_video
     )
 
+
 demo.launch()