Update streamlit_app.py

streamlit_app.py

@@ -1,3 +1,4 @@
+# streamlit_app.py
 import os
 import time
 import json
@@ -72,7 +73,11 @@ def convert_video_to_mp4(video_path: str) -> str:
     target_path = str(Path(video_path).with_suffix(".mp4"))
     if os.path.exists(target_path):
         return target_path
-    ffmpeg.input(video_path).output(target_path).run(overwrite_output=True, quiet=True)
+    try:
+        ffmpeg.input(video_path).output(target_path).run(overwrite_output=True, quiet=True)
+    except Exception as e:
+        # If ffmpeg binary missing or conversion fails, re-raise with hint
+        raise RuntimeError(f"ffmpeg conversion failed: {e}")
     try:
         os.remove(video_path)
     except Exception:
@@ -89,10 +94,12 @@ def download_video_ytdlp(url: str, save_dir: str, video_password: str = None) ->
     if video_password:
         ydl_opts["videopassword"] = video_password
     with yt_dlp.YoutubeDL(ydl_opts) as ydl:
-        ydl.download([url])
-    video_id = url.rstrip("/").split("/")[-1] or url.rstrip("/").split("/")[-2]
-    matches = glob(os.path.join(save_dir, f"{video_id}.*"))
-    if not matches:
+        info = ydl.extract_info(url, download=True)
+    # Prefer returned id/filename
+    video_id = info.get("id") if isinstance(info, dict) else None
+    if video_id:
+        matches = glob(os.path.join(save_dir, f"{video_id}.*"))
+    else:
         matches = sorted(glob(os.path.join(save_dir, "*")), key=os.path.getmtime, reverse=True)[:1]
     if not matches:
         raise FileNotFoundError("Downloaded video not found")