fix: Remove type parameter from Chatbot (default in Gradio 6)

- Remove deprecated type='messages' parameter (now default in Gradio 6)
- Add error handling for list_organizations API call
- Add retries and better error messages for video download
- Add timeout and error handling for ffmpeg audio extraction

app.py

@@ -43,10 +43,13 @@ def hello(profile: gr.OAuthProfile | None) -> str:
 def list_organizations(oauth_token: gr.OAuthToken | None) -> str:
     if oauth_token is None:
         return ""
-    org_names = [org["name"] for org in whoami(oauth_token.token)["orgs"]]
-    if org_names:
-        return f"You belong to: {', '.join(org_names)}"
-    return "You don't belong to any organizations."
+    try:
+        org_names = [org["name"] for org in whoami(oauth_token.token)["orgs"]]
+        if org_names:
+            return f"You belong to: {', '.join(org_names)}"
+        return "You don't belong to any organizations."
+    except Exception:
+        return ""
 
 
 def get_device():
@@ -77,41 +80,57 @@ CHAT_MODEL = "Qwen/Qwen2.5-72B-Instruct"
 def download_video(url: str, output_dir: str) -> list[dict]:
     """Download video from YouTube URL (video or playlist)."""
     ydl_opts = {
-        "format": "best[height<=720]",
+        "format": "best[height<=720]/best",
         "outtmpl": os.path.join(output_dir, "%(title)s.%(ext)s"),
         "quiet": True,
         "no_warnings": True,
+        "ignoreerrors": True,
+        "retries": 3,
     }
 
     downloaded = []
-    with yt_dlp.YoutubeDL(ydl_opts) as ydl:
-        info = ydl.extract_info(url, download=True)
-        if "entries" in info:
-            for entry in info["entries"]:
-                if entry:
-                    ext = entry.get("ext", "mp4")
-                    downloaded.append({
-                        "title": entry.get("title", "Unknown"),
-                        "path": os.path.join(output_dir, f"{entry['title']}.{ext}"),
-                        "duration": entry.get("duration", 0),
-                    })
-        else:
-            ext = info.get("ext", "mp4")
-            downloaded.append({
-                "title": info.get("title", "Unknown"),
-                "path": os.path.join(output_dir, f"{info['title']}.{ext}"),
-                "duration": info.get("duration", 0),
-            })
+    try:
+        with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+            info = ydl.extract_info(url, download=True)
+            if info is None:
+                raise ValueError("Could not extract video information")
+            if "entries" in info:
+                for entry in info["entries"]:
+                    if entry:
+                        ext = entry.get("ext", "mp4")
+                        downloaded.append({
+                            "title": entry.get("title", "Unknown"),
+                            "path": os.path.join(output_dir, f"{entry['title']}.{ext}"),
+                            "duration": entry.get("duration", 0),
+                        })
+            else:
+                ext = info.get("ext", "mp4")
+                downloaded.append({
+                    "title": info.get("title", "Unknown"),
+                    "path": os.path.join(output_dir, f"{info['title']}.{ext}"),
+                    "duration": info.get("duration", 0),
+                })
+    except Exception as e:
+        raise RuntimeError(f"Failed to download video: {e!s}") from e
     return downloaded
 
 
 def extract_audio(video_path: str, output_dir: str) -> str:
     """Extract audio from video file."""
     audio_path = os.path.join(output_dir, "audio.mp3")
-    subprocess.run([
-        "ffmpeg", "-i", video_path, "-vn", "-acodec", "libmp3lame",
-        "-q:a", "2", audio_path, "-y"
-    ], capture_output=True)
+    try:
+        result = subprocess.run(
+            ["ffmpeg", "-i", video_path, "-vn", "-acodec", "libmp3lame",
+             "-q:a", "2", audio_path, "-y"],
+            capture_output=True,
+            timeout=300,
+        )
+        if result.returncode != 0 and not os.path.exists(audio_path):
+            raise RuntimeError(f"FFmpeg failed: {result.stderr.decode()}")
+    except subprocess.TimeoutExpired as e:
+        raise RuntimeError("Audio extraction timed out") from e
+    except FileNotFoundError as e:
+        raise RuntimeError("FFmpeg not found. Please install FFmpeg.") from e
     return audio_path
 
 
@@ -430,7 +449,7 @@ with gr.Blocks() as demo:
 
         with gr.TabItem("Chat with Videos"):
             kb_stats = gr.Markdown()
-            chatbot = gr.Chatbot(label="Video Chat", type="messages")
+            chatbot = gr.Chatbot(label="Video Chat")
             chat_input = gr.Textbox(
                 label="Ask a question about your videos",
                 placeholder="What did the video say about...?",