Update app.py

app.py

@@ -4,6 +4,7 @@ import shutil
 import gradio as gr
 import subprocess
 import cv2
+import tempfile
 
 from fonts import setup_fonts
 from converter import convert_qurancaption_to_qvm
@@ -89,6 +90,7 @@ def get_available_fonts():
 def preview_layout(
     images,
     background_video,
+    timing_json,
     arabic_font,
     arabic_y,
     translation_gap,
@@ -96,74 +98,92 @@ def preview_layout(
     translation_font_size
 ):
 
-    if not images and not background_video:
-        raise Exception(
-            "Upload Images atau Background Video"
-        )
-
-    preview_path = "temp/preview.jpg"
+    os.makedirs("temp", exist_ok=True)
 
-    # =========================
-    # IMAGE MODE
-    # =========================
+    # =====================================
+    # BACKGROUND
+    # =====================================
 
     if images:
 
-        img = cv2.imread(
-            images[0].name
-        )
-
-    # =========================
-    # VIDEO MODE
-    # =========================
+        background_image = images[0].name
 
-    else:
+    elif background_video:
 
-        frame_path = (
-            "temp/frame.jpg"
-        )
+        background_image = "temp/preview_frame.jpg"
 
         subprocess.run(
             [
                 "ffmpeg",
                 "-y",
-
                 "-i",
                 background_video,
-
                 "-vframes",
                 "1",
-
-                frame_path
+                background_image
             ],
             check=True
         )
 
-        img = cv2.imread(
-            frame_path
+    else:
+
+        raise Exception(
+            "Upload Images atau Background Video"
         )
 
-    h, w = img.shape[:2]
+    # =====================================
+    # JSON -> QVM
+    # =====================================
+
+    qvm_json = "temp/preview_qvm.json"
 
-    scale = min(
-        1920 / w,
-        1080 / h
+    convert_qurancaption_to_qvm(
+        timing_json.name,
+        qvm_json
     )
 
-    img = cv2.resize(
-        img,
-        (
-            int(w * scale),
-            int(h * scale)
-        )
+    # =====================================
+    # ASS
+    # =====================================
+
+    preview_ass = "temp/preview.ass"
+
+    generate_ass(
+        qvm_json,
+        preview_ass,
+        arabic_font,
+        arabic_y,
+        translation_gap,
+        arabic_font_size,
+        translation_font_size
     )
 
-    cv2.imwrite(
-        preview_path,
-        img
+    # =====================================
+    # RENDER 1 FRAME
+    # =====================================
+
+    preview_output = "temp/preview.jpg"
+
+    subprocess.run(
+        [
+            "ffmpeg",
+            "-y",
+
+            "-loop", "1",
+            "-i", background_image,
+
+            "-vf",
+            f"ass={preview_ass}:fontsdir=fonts",
+
+            "-frames:v",
+            "1",
+
+            preview_output
+        ],
+        check=True
     )
 
-    return preview_path
+    return preview_output
     
 def process(
     images,
@@ -428,19 +448,20 @@ Generate cinematic Quran video automatically.
         label="Result"
     )
 
-    preview_btn.click(
-        fn=preview_layout,
-        inputs=[
-            images,
-            background_video,
-            arabic_font,
-            arabic_y,
-            translation_gap,
-            arabic_font_size,
-            translation_font_size
-        ],
-        outputs=preview_image
-    )
+preview_btn.click(
+    fn=preview_layout,
+    inputs=[
+        images,
+        background_video,
+        timing_json,
+        arabic_font,
+        arabic_y,
+        translation_gap,
+        arabic_font_size,
+        translation_font_size
+    ],
+    outputs=preview_image
+)
 
     generate_btn.click(
         fn=process,