Update agent.py

agent.py

@@ -14,7 +14,7 @@ import pandas as pd
 from pypdf import PdfReader
 from langchain_community.tools import WikipediaQueryRun
 from langchain_community.utilities import WikipediaAPIWrapper
-from youtube_transcript_api import YouTubeTranscriptApi
+from langchain_community.document_loaders import YoutubeLoader
 
 @tool
 def web_search(query: str) -> str:
@@ -56,9 +56,10 @@ def youtube_transcript(video_url: str) -> str:
         video_url: YouTube video url
     """
     try:
-        video_id = video_url.split("v=")[-1].split("&")[0]
-        transcript = YouTubeTranscriptApi.get_transcript(video_id)
-        return " ".join([item["text"] for item in transcript])
+        loader = YoutubeLoader(video_url)
+        # video_id = video_url.split("v=")[-1].split("&")[0]
+        # transcript = YouTubeTranscriptApi.get_transcript(video_id)
+        return loader.load()
     except Exception as e:
         return f"Error fetching transcript: {str(e)}"