Update app.py

app.py

@@ -39,13 +39,26 @@ class BasicAgent:
 
     def process_question(self, question:str) -> str:
         try:
-            search_results = cached_search(question) if self.tools and search_tool in self.tools else "No search results available."
-        
-            # Extract key information and provide direct answer
-            relevant_info = self._extract_key_info(search_results, question)
-            direct_answer = self._formulate_direct_answer(relevant_info, question)
-        
-            return direct_answer
+            # Check if this is a request about a YouTube video
+            youtube_patterns = ["youtube.com", "youtu.be", "watch youtube", "youtube video"]
+            use_youtube_tool = any(pattern in question.lower() for pattern in youtube_patterns)
+            
+            if use_youtube_tool and any(isinstance(tool, YouTubeVideoTool) for tool in self.tools):
+                # Extract potential YouTube URL or ID
+                url_match = re.search(r'(?:https?:\/\/)?(?:www\.)?(?:youtube\.com|youtu\.be)\/[^\s]+', question)
+                youtube_url = url_match.group(0) if url_match else question
+                
+                # Use YouTube tool
+                youtube_info = next(tool for tool in self.tools 
+                                   if isinstance(tool, YouTubeVideoTool))(youtube_url)
+                
+                relevant_info = self._extract_key_info(youtube_info, question)
+                return self._formulate_direct_answer(relevant_info, question)
+            else:
+                # Use regular search
+                search_results = cached_search(question) if self.tools and search_tool in self.tools else "No search results available."
+                relevant_info = self._extract_key_info(search_results, question)
+                return self._formulate_direct_answer(relevant_info, question)
         except Exception as e:
             if "too many requests" in str(e).lower():
                 time.sleep(2)