Update app.py

app.py

@@ -12,14 +12,41 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 login(token=os.environ["HUGGINGFACEHUB_API_TOKEN"])
 
+## TOOL 1
 search_tool = DuckDuckGoSearchTool()
 
+## TOOL 2
+from smolagents import Tool
+from youtube_transcript_api import YouTubeTranscriptApi
+import re
+
+class YouTubeTranscriptTool(Tool):
+    name = "youtube_transcript"
+    description = "Use this to extract spoken content from a YouTube video given its URL."
+
+    def use(self, input_text: str) -> str:
+        try:
+            # Extract video ID from URL
+            video_id_match = re.search(r"(?:v=|youtu\.be/)([a-zA-Z0-9_-]{11})", input_text)
+            if not video_id_match:
+                return "Invalid YouTube URL provided."
+
+            video_id = video_id_match.group(1)
+            transcript = YouTubeTranscriptApi.get_transcript(video_id)
+            
+            # Combine into a single string
+            full_text = " ".join([entry['text'] for entry in transcript])
+            return full_text[:3000]  # limit to 3000 chars
+        except Exception as e:
+            return f"Error fetching transcript: {e}"
+transcript_tool = YouTubeTranscriptTool()
+
 async def run_and_submit_all(profile: gr.OAuthProfile | None):
     log_output = ""
 
     try:
         agent = ToolCallingAgent(
-            tools=[search_tool],
+            tools=[search_tool, transcript_tool],
             model=InferenceClientModel(model="deepseek-ai/DeepSeek-V3", provider="together"),
             max_steps=15,
             verbosity_level=0,