Update app.py

app.py

@@ -4,12 +4,11 @@ import requests
 import inspect
 import pandas as pd
 #mode requirement
-from smolagents import DuckDuckGoSearchTool,load_tool, tool
+from smolagents import DuckDuckGoSearchTool,load_tool, tool,CodeAgent
 from typing import TypedDict, List, Dict, Any, Optional
 from langgraph.graph import StateGraph, END
 from langchain_openai import ChatOpenAI
 from langchain_core.messages import HumanMessage
-from langchain.agents import initialize_agent
 
 # (Keep Constants as is)
 # --- Constants ---
@@ -142,7 +141,7 @@ class BasicAgent:
         If you are given a task to create an image,you can use the image_generation_tool.
         """
         llm = ChatOpenAI(model="gpt-4o", temperature=0)
-        self.agent= initialize_agent(
+        self.agent= CodeAgent(
             tools = [add, subtract, multiply, divide, web_search, image_generation_tool],
             llm=llm,
         )