Update app.py

app.py

@@ -5,14 +5,12 @@ import inspect
 import pandas as pd
 import gradio as gr
 import random
-from langchain_core.messages import HumanMessage
 
+from langchain_core.messages import HumanMessage
+from smolagents import GradioUI, CodeAgent, HfApiModel, Tool
 
 # Import our custom tools from their modules
 from langchain_community.tools import DuckDuckGoSearchRun
-from smolagents import GradioUI, CodeAgent, HfApiModel
-
-
 
 # Initialize the Hugging Face model
 model = HfApiModel()
@@ -20,11 +18,21 @@ model = HfApiModel()
 # Initialize the web search tool
 search_tool = DuckDuckGoSearchRun()
 
+# Custom Tool Class
+class CustomSearchTool(Tool):
+    def __init__(self, search_tool):
+        super().__init__()
+        self.search_tool = search_tool
 
+    def run(self, query):
+        return self.search_tool.run(query)
 
+# Wrap the search tool in your custom tool class
+custom_search_tool = CustomSearchTool(search_tool)
 
+# Initialize the agent with the custom tool
 agent = CodeAgent(
-    tools=[search_tool], 
+    tools=[custom_search_tool],
     model=model,
     add_base_tools=True,  # Add any additional base tools
     planning_interval=3   # Enable planning every 3 steps
@@ -124,7 +132,7 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
         print("Agent did not produce any answers to submit.")
         return "Agent did not produce any answers to submit.", pd.DataFrame(results_log)
 
-    # 4. Prepare Submission 
+    # 4. Prepare Submission
     submission_data = {"username": username.strip(), "agent_code": agent_code, "answers": answers_payload}
     status_update = f"Agent finished. Submitting {len(answers_payload)} answers for user '{username}'..."
     print(status_update)
@@ -172,7 +180,6 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
         results_df = pd.DataFrame(results_log)
         return status_message, results_df
 
-
 # --- Build Gradio Interface using Blocks ---
 with gr.Blocks() as demo:
     gr.Markdown("# Basic Agent Evaluation Runner")
@@ -224,4 +231,4 @@ if __name__ == "__main__":
     print("-"*(60 + len(" App Starting ")) + "\n")
 
     print("Launching Gradio Interface for Basic Agent Evaluation...")
-    demo.launch(debug=True, share=False)
+    demo.launch(debug=True, share=False)