Update app.py

app.py

@@ -1,6 +1,6 @@
 import os
 import requests
-from smolagents import LiteLLMModel, ToolCallingAgent, Tool
+from smolagents import LiteLLMModel, CodeAgent, ToolCallingAgent, Tool
 import wikipedia
 import gradio as gr
 import pandas as pd
@@ -96,7 +96,7 @@ model = LiteLLMModel(
 )
 
 # --- Initialize Tool-Calling Agent ---
-agent = ToolCallingAgent(
+agent = CodeAgent(
     tools=[WebSearchTool(), WikipediaTool(), WeatherTool(), LoadCsvTool()],
     model=model,
     max_steps=10,