Update app.py

app.py

@@ -5,7 +5,7 @@ import requests
 import inspect
 import pandas as pd
 from smolagents import (
-    ToolCallingAgent,
+    CodeAgent,
     LiteLLMModel,
     DuckDuckGoSearchTool,
     WikipediaSearchTool,
@@ -42,12 +42,13 @@ class BasicAgent:
             PythonInterpreterTool(),
             get_current_date_time,
         ]
-        self.agent = ToolCallingAgent(
+        self.agent = CodeAgent(
             tools=self.tools,
             model=self.model,
-            max_steps=6,
+            max_steps=8,
+            additional_authorized_imports=["datetime", "re", "json", "math", "collections"],
         )
-        print("BasicAgent ready with Groq Llama-3.3-70B.")
+        print("BasicAgent ready with Groq Llama-3.3-70B (CodeAgent).")
 
     def __call__(self, question: str) -> str:
         print(f"Agent received question: {question[:80]}...")
@@ -105,7 +106,7 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
         except Exception as e:
             print(f"Error on task {task_id}: {e}")
             results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": f"ERROR: {e}"})
-        time.sleep(1)
+        time.sleep(2)
     if not answers_payload:
         return "No answers.", pd.DataFrame(results_log)
     submission_data = {"username": username.strip(), "agent_code": agent_code, "answers": answers_payload}