Update tools.py

tools.py

@@ -6,6 +6,8 @@ import os
 import tempfile
 from urllib.parse import urlparse
 from openai import OpenAI
+from youtube_transcript_api import YouTubeTranscriptApi
+from pytube import extract
 
 @tool
 def current_date(_) -> str :
@@ -65,4 +67,25 @@ def transcribe_audio(audio_file: str, file_extension: str) -> str:
         return transcription.text
 
     except Exception as e:
-        return f"transcribe_audio failed: {e}"
+        return f"transcribe_audio failed: {e}"
+
+# TESTTTTT
+@tool
+def get_youtube_transcript(page_url: str) -> str:
+    """Get the transcript of a YouTube video
+    Args:
+        page_url (str): YouTube URL of the video
+    """
+    try:
+        # get video ID from URL
+        video_id = extract.video_id(page_url)
+
+        # get transcript
+        ytt_api = YouTubeTranscriptApi()
+        transcript = ytt_api.fetch(video_id)
+
+        # keep only text
+        txt = '\n'.join([s.text for s in transcript.snippets])
+        return txt
+    except Exception as e:
+        return f"get_youtube_transcript failed: {e}"