Update tools.py

tools.py

@@ -9,6 +9,7 @@ import fitz  # PyMuPDF
 import pandas as pd
 from imdb import IMDb
 from youtube_transcript_api import YouTubeTranscriptApi
+import yt_dlp
 import whisper
 from bs4 import BeautifulSoup
 import re
@@ -251,16 +252,57 @@ def run_python_code(code: str) -> str:
 @tool
 def search_youtube_transcript(video_id: str) -> str:
     """
-    Fetch transcript text from a YouTube video using its ID.
-    Use this for questions asking what was said in a specific video.
-    Returns full transcript as plain text.
+    Try to fetch transcript from a YouTube video using its ID.
+    If transcript is not available, fall back to title/description metadata using yt-dlp.
     """
     print(f"🛠️ search_youtube_transcript called with: {video_id}")
+
     try:
         transcript = YouTubeTranscriptApi.get_transcript(video_id)
-        return " ".join([entry["text"] for entry in transcript])
+        if not transcript:
+            return "Transcript is empty or unavailable for this video."
+
+        # Limit to first 1000 characters to prevent token overflow
+        text = " ".join([entry["text"] for entry in transcript])
+        return f"✅ Transcript found (preview):\n\n{text[:1000]}..."
+
+    except Exception as e:
+        error_msg = str(e)
+        print(f"⚠️ Transcript fetch failed: {error_msg}")
+
+        # Handle common errors
+        if "not available" in error_msg.lower() or "no transcript" in error_msg.lower():
+            fallback_info = fetch_video_metadata(video_id)
+            return f"⚠️ No transcript available.\n\n🔎 Here's some metadata instead:\n{fallback_info}"
+        elif "video unavailable" in error_msg.lower():
+            return "❌ This video is unavailable. Please check the link or video ID."
+        elif "404" in error_msg or "Not Found" in error_msg:
+            return "❌ The video could not be found. Please check the video ID."
+        else:
+            return f"❌ Error fetching transcript: {error_msg}"
+
+
+def fetch_video_metadata(video_id: str) -> str:
+    """
+    Fallback to extract title and description if transcript is not available.
+    """
+    try:
+        url = f"https://www.youtube.com/watch?v={video_id}"
+        ydl_opts = {
+            'quiet': True,
+            'no_warnings': True,
+            'extract_flat': True,
+            'no_playlist': True
+        }
+
+        with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+            info = ydl.extract_info(url, download=False, process=False)
+            title = info.get("title", "Unknown Title")
+            description = info.get("description", "No description found.")
+            return f"📺 Title: {title}\n📝 Description: {description[:500]}..."  # Truncate for LLM
+
     except Exception as e:
-        return f"Error fetching transcript: {e}"
+        return f"❌ Error fetching metadata: {str(e)}"
 
 @tool
 def search_baseball_stats(player: str) -> str: