Update app.py

app.py

@@ -30,6 +30,7 @@ class BasicAgent:
     def __init__(self, model="google/gemma-7b"):
         #self.api_url = f"https://api-inference.huggingface.co/models/{model}"
         self.headers = {"Authorization": f"Bearer {os.getenv('HF_API_KEY')}"}
+        genai.configure(api_key=os.getenv(Gemini_API_KEY)  # Get from https://ai.google.dev/
         # Wikipedia setup (with proper User-Agent)
         self.wiki = wikipediaapi.Wikipedia(
             language='en',
@@ -46,6 +47,14 @@ class BasicAgent:
         print(f"Agent returning answer: {fixed_answer}")
         return fixed_answer
 
+    
+    def search_tool(self, prompt: str) -> str:
+        model = genai.GenerativeModel('gemini-pro')
+        response = model.generate_content(
+            f"Search the web for: {prompt}. Summarize the results."
+        )
+        return response.text
+
     def search_tool(self, prompt: str) -> str:
         # Initialize Gemini model
         llm = ChatGoogleGenerativeAI(model="gemini-1.5-flash")  # or "gemini-pro"