Update app.py

app.py

@@ -3,15 +3,17 @@ import gradio as gr
 import requests
 import inspect
 import pandas as pd
-from smolagents import ToolCallingAgent, LiteLLMModel, DuckDuckGoSearchTool
+from smolagents import CodeAgent, LiteLLMModel, DuckDuckGoSearchTool
 
 # (Keep Constants as is)
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 
-# tools to use
-search_tool = DuckDuckGoSearchTool()
+# import custom tools to use
+from tools import SerperSearchTool
+
+search_tool = SerperSearchTool()
 
 # Initialize the watsonx model
 llm = LiteLLMModel(
@@ -21,11 +23,18 @@ llm = LiteLLMModel(
     num_ctx=8162
 )
 
-# --- Basic Agent Definition ---
-# ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
 
-agent = ToolCallingAgent(tools=[search_tool], model=llm)
+# Create Alfred with all the tools
+agent = CodeAgent(
+    tools=[search_tool], 
+    model=llm,
+    add_base_tools=True,  # Add any additional base tools
+    planning_interval=3   # Enable planning every 3 steps
+)
 
+#agent = ToolCallingAgent(tools=[search_tool], model=llm)
+
+# --- Basic Agent Definition ---
 class BasicAgent:
     def __init__(self):
         print("BasicAgent initialized.")