Update tools.py

tools.py

@@ -4,9 +4,10 @@ import requests
 import openai
 import os
 import tempfile
-from urllib.parse import urlparse
+from urllib.parse import urlparse, parse_qs
 from openai import OpenAI
 from youtube_transcript_api import YouTubeTranscriptApi
+from youtube_transcript_api._errors import TranscriptsDisabled, NoTranscriptFound, VideoUnavailable
 from pytube import extract
 from openai import OpenAI
 
@@ -128,13 +129,21 @@ def transcribe_youtube(youtube_url: str) -> str:
         youtube_url (str): youtube video's url
     """
     try:
-        video_id = extract.video_id(youtube_url)
-        transcript = YouTubeTranscriptApi.get_transcript(video_id)
-
+        query = urlparse(youtube_url).query
+        video_id = parse_qs(query)['v'][0]
+    except Exception:
+        return "invalid YouTube URL"
+        
+    try:
+        transcript_list = YouTubeTranscriptApi.list_transcripts(video_id)
+        transcript = transcript_list.find_transcript(['en']).fetch()
         # keep only text
-        text = '\n'.join([s['text'] for s in transcript])
+        text = '\n'.join([t['text'] for t in transcript])
         return text
-        
+
+    except (TranscriptsDisabled, NoTranscriptFound, VideoUnavailable) as e:
+        return f"transcript unavailable: {str(e)}"
+    
     except Exception as e:
         return f"transcribe_youtube failed: {e}"