Update app.py

app.py

@@ -3,7 +3,7 @@ import gradio as gr
 import requests
 import inspect
 import pandas as pd
-from smolagents import GradioUI, CodeAgent, HfApiModel
+from langchain_google_genai import ChatGoogleGenerativeAI
 from tools import DuckDuckGoSearchTool, WeatherInfoTool
 
 # (Keep Constants as is)
@@ -16,7 +16,7 @@ class BasicAgent:
     def __init__(self):
         print("BasicAgent initialized.")
         # Initialize the Hugging Face model
-        model = HfApiModel()
+        model = ChatGoogleGenerativeAI(model="gemini-1.5-flash-latest")
         
         # Initialize the web search tool
         search_tool = DuckDuckGoSearchTool()
@@ -25,16 +25,18 @@ class BasicAgent:
         weather_info_tool = WeatherInfoTool()
 
         # Create agent
-        self.agent= CodeAgent(
-            tools=[weather_info_tool, search_tool], 
-            model=model,
-            add_base_tools=True,  # Add any additional base tools
-            planning_interval=3   # Enable planning every 3 steps
-        )
+        tools = [search_tool, weather_info_tool]
+        self.agent= llm.bind_tools(tools)
+        #= CodeAgent(
+        #    tools=[weather_info_tool, search_tool], 
+        #    model=model,
+        #    add_base_tools=True,  # Add any additional base tools
+        #    planning_interval=3   # Enable planning every 3 steps
+        #)
 
     def __call__(self, question: str) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
-        answer = self.agent.run(question)
+        answer = self.agent.invoke(question)
         print(f"Agent returning fixed answer: {fixed_answer}")
         return answer