Update app.py

app.py

@@ -8,7 +8,9 @@ import requests
 from typing import List, Dict, Union
 import pandas as pd
 import wikipediaapi
-from google.genarativeai import genai
+from typing import Optional
+from termax.agent import Agent  # Assuming you're using a specific agent framework
+from termax.tool import google_search  # Assuming this is your search tool import
 
 load_dotenv()
 
@@ -34,10 +36,33 @@ class BasicAgent:
 
     def __call__(self, question: str) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
-        fixed_answer = self.answer_question(question)
+        fixed_answer = self.search_tool(question)
         print(f"Agent returning answer: {fixed_answer}")
         return fixed_answer
 
+
+    def search_tool(self, prompt: str) -> str:
+        """
+        Searches for information using a Gemini-powered agent with Google Search capability.
+        
+        Args:
+            prompt: The user's query to search for
+            
+        Returns:
+            str: The response from the search agent
+        """
+        root_agent = Agent(
+            name="search_assistant",
+            model="gemini-2.0-flash",  # Or your preferred Gemini model
+            instruction="You are a helpful assistant. Answer user questions using Google Search when needed.",
+            description="An assistant that can search the web.",
+            tools=[google_search]
+        )
+        
+        # Actually invoke the agent with the prompt
+        response = root_agent.run(prompt)
+        return response
+
     def answer_question(self, prompt: str) -> str:
         from google.genarativeai import genai