Update app.py

app.py

@@ -13,6 +13,13 @@ from tavily import TavilyClient
 from dotenv import load_dotenv
 load_dotenv()
 
+
+tavily_client = TavilyClient(api_key=os.getenv("TAVILY_API_KEY"))
+@tool
+def web_search(query: str) -> Dict[str, Any]:
+    """Search the web for information"""
+    return tavily_client.search(query)
+
 # (Keep Constants as is)
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
@@ -36,24 +43,22 @@ class BasicAgent:
             base_url="https://openrouter.ai/api/v1",
             api_key=os.getenv("OPENROUTER_API_KEY")
         )
-        self.tavily_client = TavilyClient(api_key=os.getenv("TAVILY_API_KEY"))
-        tools = [self.web_search]
+        tools = [web_search]
         self.agent = create_agent(
             model=self.llm,
             tools=tools,
             system_prompt=SYSTEM_PROMPT
         )
         print("BasicAgent initialized.")
-    @tool
-    def web_search(self, query: str) -> Dict[str, Any]:
-        """Search the web for information"""
-        return self.tavily_client.search(query)
+    
     def __call__(self, question: str) -> str:
+        print(f"Agent received question: {question}")
         user_prompt = HumanMessage(content=question)
         response = self.agent.invoke(
             {"messages": [user_prompt]}
         )
         answer = response['messages'][-1].content.strip()
+        print(f"Agent generated answer: {answer}")
         return answer
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):