Update app.py

app.py

@@ -14,6 +14,8 @@ import torch
 import bitsandbytes
 import scipy
 from googletrans import Translator
+import re
+import subprocess
 
 ZipFile("ffmpeg.zip").extractall()
 st = os.stat('ffmpeg')
@@ -31,13 +33,35 @@ whisper_model = WhisperModel("large-v2", device="cuda", compute_type="float16")
 print("cwd", os.getcwd())
 print(os.listdir())
 
+
+def get_video_duration(video_path):
+    result = subprocess.run(["ffmpeg", "-i", video_path, "-hide_banner"], capture_output=True, text=True, stderr=subprocess.STDOUT)
+    output = result.stdout
+    match = re.search(r"Duration: (\d+:\d+:\d+.\d+)", output)
+    if match:
+        duration_str = match.group(1)
+        h, m, s = map(float, re.split('[:.]', duration_str))
+        duration = h * 3600 + m * 60 + s
+        return duration
+    else:
+        return None
+        
 def process_video(Video, target_language):
     current_path = os.getcwd()
     print("Iniciando process_video")
     common_uuid = uuid.uuid4()
     print("Checking FFmpeg availability...")
     run(["ffmpeg", "-version"])
+    duration = get_video_duration(Video)
+    if duration is None:
+        print("Could not determine video duration.")
+        return None
 
+    # Check duration
+    if duration > 900:
+        print("Video is longer than 15 minutes.")
+        gr.Warning("Warning: Video is longer than 15 minutes.")
+        return None
     
     audio_file = f"{common_uuid}.wav"
     run(["ffmpeg", "-i", Video, audio_file])