Update app.py

app.py

@@ -9,9 +9,12 @@ from langchain_core.messages import HumanMessage
 from langchain_openai import ChatOpenAI
 from langchain_community.tools import DuckDuckGoSearchResults
 from langchain_google_community import GoogleSearchAPIWrapper
+from langchain_community.document_loaders import YoutubeLoader
 import wikipedia
 import speech_recognition as sr
 import tempfile
+import ast
+
 # Or using AudioTranscriptTool.
 
 
@@ -39,7 +42,33 @@ def add(a: int|float, b: int|float) -> float:
 def subtract(a: int|float, b: int|float) -> float:
     """Subtract a with b."""
     return a - b
+    
+def youtube_transcript_tool(url: str) -> str:
+    """Load transcript from a YouTube video URL."""
+    loader = YoutubeLoader.from_youtube_url(url, add_video_info=True)
+    docs = loader.load()
+    return "\n".join([f"Title: {doc.metadata.get('title', 'Unknown')}\nTranscript: {doc.page_content}" for doc in docs])
+    
+def analyze_python_code(file_url: str) -> str:
+    """Downloads an python code from a URL, then analyze it and summarize its structure."""
+    try:
+        # Download file into temporary file
+        with tempfile.NamedTemporaryFile(suffix=".py", delete=True) as temp_file:
+            response = requests.get(file_url)
+            if response.status_code != 200:
+                return f"Failed to download file: {response.status_code}"
+                
+            temp_file.write(response.content)
+            temp_file.flush()  # Make sure data is written
+            # Transcribe from temp file
+            tree = ast.parse(temp_file.name)
+            functions = [node.name for node in ast.walk(tree) if isinstance(node, ast.FunctionDef)]
+            classes = [node.name for node in ast.walk(tree) if isinstance(node, ast.ClassDef)]
+            return f"Functions: {functions}\nClasses: {classes}"
 
+    except Exception as e:
+        return f"Reading failed: {str(e)}"
+        
 def transcribe_audio(file_url: str) -> str:
     """Downloads an audio file from a URL into a temporary file and transcribes it using SpeechRecognition."""
     try:
@@ -75,6 +104,9 @@ tools = [
     multiply,
     add,
     subtract,
+    youtube_transcript_tool,
+    transcribe_audio,
+    analyze_python_code,
     search_wikipedia,
     GoogleSearchAPIWrapper(k=10).run,
     DuckDuckGoSearchResults(),
@@ -86,9 +118,9 @@ from langgraph.graph.message import add_messages
 
 
 class AgentState(TypedDict):
-    # The input document
     messages: Annotated[list[AnyMessage], add_messages]
-    input_file: Optional[str]  # Contains file path, type (PNG)
+    # The input file download URL
+    input_file: Optional[str]  
     
 from langchain_core.messages import HumanMessage, SystemMessage
 
@@ -110,6 +142,20 @@ transcribe_audio(file_url: str) -> str:
     Returns:
         The transcribed text from the given audio.
 
+youtube_transcript_tool(url: str) -> str:
+    Load transcript from a YouTube video URL.
+    Args:
+        url, a string indicating the url of the given youtube video.
+    Returns:
+        The transcript of the selected youtube video.
+    
+analyze_python_code(file_url: str) -> str:
+    Downloads an python code from a URL, then analyze it and summarize its structure.
+    Args:
+        file_url, a string indicating the url of the given file.
+    Returns:
+        The structure and functions of the given code.
+        
 search_wikipedia(query: str) -> str:
     Search English Wikipedia for the given query and return a summary within 3 sentences.
     Returns: