Update app.py

app.py

@@ -3,6 +3,9 @@ import datetime
 import requests
 import pytz
 import yaml
+import os
+from youtube_transcript_api import YouTubeTranscriptApi
+import re
 from tools.final_answer import FinalAnswerTool
 
 from Gradio_UI import GradioUI
@@ -10,31 +13,27 @@ from Gradio_UI import GradioUI
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 
 @tool
-def fetch_headlines(topic: str) -> str:
-    """Fetches recent news healines for a given topic using DuckDuckGo search.
+def get_video_transcript(url: str) -> str:
+    """Returns the transcript text of a give YouTube video URL.
 
     Args:
-        topic: The topic to search for (e.g., 'AI', 'Politics', 'Sport')
-
-    Returns:
-        A formatted string with up to 5 news headlines and their links.
+        url: YouTube video url given by a user.
     """
-    search_tool = DuckDuckGoSearchTool()
     try:
-        query = f"{topic} news"
-        results = search_tool(query)
-
-        headlines = []
-        for result in results[:5]:
-            title = result.get('title', 'No title')
-            link = result.get('link', 'No link')
-            headlines.append(f"{len(headlines) + 1}. {title} - {link}")
-
-        if not headlines:
-            return f"No headlines found for topic '{topic}'."
-        return "\n".join(headlines)
+        video_id = extract_video_id(url)
+        transctipt = YouTubeTranscriptApi.get_transcript(video_id)
+        text = " ".join([entry["text"] for entry in transctipt])
+        return text
     except Exception as e:
-        return f"Error fetching headlines: {str(e)}"
+        return f"Error: {str(e)}"
+
+def extract_video_id(url: str) -> str:
+    match = re.search(r"(?:v=|youtu\.be/)([^&\n?#]+)", url)
+    if match:
+        return match.grou(1)
+    else:
+        raise ValueError("Invalid YouTube URL.")
+        
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """A tool that fetches the current local time in a specified timezone.