Add reverse_text and analyze_youtube_video tools to agent.py for text manipulation and YouTube video analysis

agent.py

@@ -11,6 +11,7 @@ from langchain_core.tools import tool
 import requests
 from bs4 import BeautifulSoup
 import urllib.parse
+import re
 
 load_dotenv()
 
@@ -83,6 +84,57 @@ def arxiv_search(query: str) -> str:
     except Exception as e:
         return f"Error searching Arxiv: {str(e)}"
 
+@tool
+def reverse_text(text: str) -> str:
+    """Reverse the given text.
+    
+    Args:
+        text: The text to reverse."""
+    return text[::-1]
+
+@tool
+def analyze_youtube_video(video_url: str) -> str:
+    """Analyze a YouTube video for specific information.
+    
+    Args:
+        video_url: The URL of the YouTube video."""
+    try:
+        # Extract video ID from URL
+        video_id = re.search(r'(?:v=|\/)([0-9A-Za-z_-]{11}).*', video_url)
+        if not video_id:
+            return "Invalid YouTube URL"
+        
+        video_id = video_id.group(1)
+        
+        # Use YouTube API or web search to get video information
+        encoded_query = urllib.parse.quote(f"youtube video {video_id} transcript description")
+        url = f"https://html.duckduckgo.com/html/?q={encoded_query}"
+        
+        headers = {
+            'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36'
+        }
+        
+        response = requests.get(url, headers=headers)
+        response.raise_for_status()
+        
+        soup = BeautifulSoup(response.text, 'html.parser')
+        
+        results = []
+        for result in soup.find_all('div', class_='result__body'):
+            title = result.find('h2', class_='result__title')
+            snippet = result.find('a', class_='result__snippet')
+            
+            if title and snippet:
+                results.append(f"Title: {title.get_text()}\nSnippet: {snippet.get_text()}")
+            
+            if len(results) >= 3:
+                break
+        
+        return {"video_results": "\n\n".join(results) if results else "No results found"}
+            
+    except Exception as e:
+        return f"Error analyzing YouTube video: {str(e)}"
+
 # System prompt
 system_prompt = """You are a highly accurate question-answering assistant. Your task is to provide precise, direct answers to questions.
 
@@ -117,7 +169,10 @@ Key Rules:
    - For historical dates: Use the most widely accepted date
    - For scientific terms: Use the standard scientific notation
    - For geographical locations: Use official names without abbreviations
-   - For audio/video questions: Focus on the specific detail requested"""
+   - For audio/video questions: Focus on the specific detail requested
+   - For reversed text: Use the reverse_text tool to decode
+   - For YouTube videos: Use the analyze_youtube_video tool to get information
+   - For chess positions: Provide the move in standard algebraic notation"""
 
 # System message
 sys_msg = SystemMessage(content=system_prompt)
@@ -127,6 +182,8 @@ tools = [
     wiki_search,
     web_search,
     arxiv_search,
+    reverse_text,
+    analyze_youtube_video,
 ]
 
 def build_graph():