Update app.py

app.py

@@ -6,7 +6,7 @@ import json
 import re
 import tempfile
 import logging
-from typing import List, Dict, Optional
+from typing import List, Dict, Optional, TypedDict, Annotated
 import numpy as np
 
 # Core ML/AI imports
@@ -19,10 +19,8 @@ from langgraph.graph import StateGraph, START, END
 from langgraph.graph.message import add_messages
 from langgraph.prebuilt import ToolNode, tools_condition
 from langgraph.checkpoint.memory import MemorySaver
-from typing import TypedDict, Annotated, List
 
 # File processing
-import pandas as pd
 import wikipedia
 from youtube_transcript_api import YouTubeTranscriptApi
 import speech_recognition as sr
@@ -53,10 +51,6 @@ logging.getLogger("ultralytics").setLevel(logging.ERROR)
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
-# Agent State Definition
-class AgentState(TypedDict):
-    messages: Annotated[List[AnyMessage], add_messages]
-
 # System prompt for the agent
 SYSTEM_PROMPT = """You are a precision research assistant for the GAIA benchmark. Your mission is EXTREME ACCURACY.
 
@@ -142,7 +136,9 @@ class GAIAAgent:
 
     def _setup_tools(self):
         """Setup all the tools for the agent"""
-        tools = []
+        
+        # Store reference to self for use in nested functions
+        agent_instance = self
         
         # Wikipedia tool
         @tool
@@ -164,10 +160,10 @@ class GAIAAgent:
         @tool  
         def web_search_tool(query: str) -> str:
             """Web search for current information"""
-            if not self.tavily_api_key:
+            if not agent_instance.tavily_api_key:
                 return "Tavily API key not available"
             try:
-                tavily_search = TavilySearchResults(api_key=self.tavily_api_key, max_results=5)
+                tavily_search = TavilySearchResults(api_key=agent_instance.tavily_api_key, max_results=5)
                 results = tavily_search.invoke(query)
                 formatted_results = []
                 for i, res in enumerate(results, 1):
@@ -180,11 +176,11 @@ class GAIAAgent:
         @tool
         def wolfram_alpha_tool(query: str) -> str:
             """Use Wolfram Alpha for computational questions"""
-            if not self.wolfram_api_key:
+            if not agent_instance.wolfram_api_key:
                 return "Wolfram API key not available"
             
             params = {
-                'appid': self.wolfram_api_key,
+                'appid': agent_instance.wolfram_api_key,
                 'input': query,
                 'format': 'plaintext',
                 'output': 'JSON'
@@ -260,18 +256,22 @@ class GAIAAgent:
         # Python REPL tool
         python_repl_tool = PythonREPLTool()
         
-        tools.extend([
+        tools = [
             wikipedia_tool,
             web_search_tool, 
             wolfram_alpha_tool,
             file_analyzer_tool,
             python_repl_tool
-        ])
+        ]
         
         return tools
 
     def _create_agent_runner(self):
         """Create the LangGraph agent runner"""
+        # Define AgentState locally
+        class AgentState(TypedDict):
+            messages: Annotated[List[AnyMessage], add_messages]
+            
         model_with_tools = self.llm.bind_tools(self.tools)
         
         def agent_node(state):