Update app.py

app.py

@@ -7,7 +7,7 @@ import pandas as pd
 # Import our custom tools from their modules
 from huggingface_hub import login
 from smolagents import CodeAgent, InferenceClientModel
-from tools import web_search, visit_webpage, final_answer
+from tools import web_search, visit_webpage, final_answer, go_back, close_popups, search_item_ctrl_f
 from retriever import load_guest_dataset
 from functools import lru_cache
 
@@ -33,6 +33,12 @@ class BasicAgent:
         self.visit_webpage = visit_webpage
 
         self.final_answer = final_answer
+
+        self.go_back = go_back, 
+        
+        self.close_popups = close_popups, 
+        
+        self.search_item_ctrl_f = search_item_ctrl_f
         
         # Create the model
         self.model = InferenceClientModel(model_id="Qwen/Qwen2.5-72B-Instruct")
@@ -43,6 +49,7 @@ class BasicAgent:
         # Create the agent with tools
         self.agent = CodeAgent(
             tools=[self.web_search, self.visit_webpage, self.final_answer],
+            # tools=[self.web_search, self.visit_webpage, self.final_answer, go_back, close_popups, search_item_ctrl_f],
             model=self.model,
             max_steps=10  # Limit reasoning steps
         )
@@ -62,6 +69,7 @@ class BasicAgent:
             formatted_question = f"{question}\n\nOnly return the exact answer, no explanation."
             formatted_question = f"{self.system_prompt}\n\nQuestion: {question}\n\n"
             answer = self.agent.run(formatted_question)
+            answer = str(answer).strip() if answer is not None else "No answer produced."
             # answer = self.agent.run(question)
             print(f"Agent returning answer: {answer[:100]}...")
             fixed_answer = answer