Update app.py

app.py

@@ -88,6 +88,7 @@ def get_available_fonts():
 
 def preview_layout(
     images,
+    background_video,
     arabic_font,
     arabic_y,
     translation_gap,
@@ -95,12 +96,52 @@ def preview_layout(
     translation_font_size
 ):
 
-    if not images:
-        raise Exception("Upload minimal 1 image")
+    if not images and not background_video:
+        raise Exception(
+            "Upload Images atau Background Video"
+        )
 
     preview_path = "temp/preview.jpg"
 
-    img = cv2.imread(images[0].name)
+    # =========================
+    # IMAGE MODE
+    # =========================
+
+    if images:
+
+        img = cv2.imread(
+            images[0].name
+        )
+
+    # =========================
+    # VIDEO MODE
+    # =========================
+
+    else:
+
+        frame_path = (
+            "temp/frame.jpg"
+        )
+
+        subprocess.run(
+            [
+                "ffmpeg",
+                "-y",
+
+                "-i",
+                background_video,
+
+                "-vframes",
+                "1",
+
+                frame_path
+            ],
+            check=True
+        )
+
+        img = cv2.imread(
+            frame_path
+        )
 
     h, w = img.shape[:2]
 
@@ -391,6 +432,7 @@ Generate cinematic Quran video automatically.
         fn=preview_layout,
         inputs=[
             images,
+            background_video,
             arabic_font,
             arabic_y,
             translation_gap,