Update tools/YouTubeTool.py

tools/YouTubeTool.py

@@ -1,5 +1,6 @@
 from smolagents import Tool
-from youtube_transcript_api import YouTubeTranscriptApi
+from youtube_transcript_api import YouTubeTranscriptApi, NoTranscriptFound, TranscriptsDisabled
+import re
 
 class YouTubeTool(Tool):
     name = "youtube_reader"
@@ -8,10 +9,22 @@ class YouTubeTool(Tool):
     output_type = "string"
 
     def forward(self, video_url: str) -> str:
-        video_id = video_url.split("v=")[-1]  # basic extraction
         try:
-            transcript = YouTubeTranscriptApi.get_transcript(video_id)
-            return " ".join([t["text"] for t in transcript])
+            # Extract video ID
+            match = re.search(r"v=([A-Za-z0-9_-]{11})", video_url)
+            if not match:
+                return "Invalid YouTube URL."
+            video_id = match.group(1)
+
+            # Fetch transcript
+            transcript_list = YouTubeTranscriptApi.get_transcript(video_id)
+            transcript_text = " ".join([t["text"] for t in transcript_list])
+
+            return transcript_text
+
+        except NoTranscriptFound:
+            return "Transcript not available for this video."
+        except TranscriptsDisabled:
+            return "Transcripts are disabled for this video."
         except Exception as e:
-            return f"Error fetching YouTube transcript: {e}"
-    
+            return f"Error fetching transcript: {str(e)}"