Update app.py

app.py

@@ -36,7 +36,31 @@ class BasicAgent:
             with open(path, "wb") as f:
                 f.write(resp.content)
             return path
+        @tool
+        def get_youtube_transcript(url: str) -> str:
+            """
+            Fetches the transcript/captions of a YouTube video.
+
+            Args:
+                 url: The full YouTube video URL.
 
+            Returns:
+                The transcript text, or an error message if unavailable.
+            """
+            from youtube_transcript_api import YouTubeTranscriptApi
+            import re
+
+             match = re.search(r"(?:v=|youtu\.be/)([\w-]{11})", url)
+             if not match:
+                 return "Could not extract video ID from URL."
+            video_id = match.group(1)
+
+            try:
+                transcript = YouTubeTranscriptApi.get_transcript(video_id)
+                 return " ".join(entry["text"] for entry in transcript)
+            except Exception as e:
+                return f"Transcript unavailable: {e}"
+                
         @tool
         def read_spreadsheet(file_path: str) -> str:
             """
@@ -81,6 +105,7 @@ class BasicAgent:
                 VisitWebpageTool(),
                 fetch_task_file,
                 read_spreadsheet,
+                get_youtube_transcript,
                 #os.environ["HF_TOKEN"] = "hf_token",
                 #image_analysis_tool,
             ],