Update app.py

app.py

@@ -1,3 +1,4 @@
+# app.py
 import os
 import gradio as gr
 import requests
@@ -7,28 +8,29 @@ from tools import AnswerTool
 from smolagents import CodeAgent, OpenAIServerModel
 from smolagents import DuckDuckGoSearchTool, WikipediaSearchTool
 
-# Constants
+# --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 class BasicAgent:
     def __init__(self):
-        # Initialize CodeAgent with GPT-4o, custom AnswerTool, DuckDuckGo and Wikipedia tools
+        # Initialize CodeAgent with GPT-4o, Wikipedia, DuckDuckGo, and AnswerTool
         model       = OpenAIServerModel(model_id="gpt-4o")
-        answer_tool = AnswerTool()
-        web_tool    = DuckDuckGoSearchTool()
         wiki_tool   = WikipediaSearchTool()
+        web_tool    = DuckDuckGoSearchTool()
+        answer_tool = AnswerTool()
         self.agent = CodeAgent(
             model=model,
-            tools=[answer_tool, web_tool, wiki_tool],
-            add_base_tools=True,
-            max_steps=2,
+            # try wiki first, then web, then direct answer
+            tools=[wiki_tool, web_tool, answer_tool],
+            add_base_tools=True,    # include python_eval, image_ocr, etc.
+            max_steps=3,            # allow up to 3 planning steps
             verbosity_level=0
         )
 
     def __call__(self, question: str) -> str:
-        # Run the agent on the question
         return self.agent.run(question)
 
+
 def run_and_submit_all(username):
     if not username:
         return "Please enter your Hugging Face username.", None
@@ -44,7 +46,7 @@ def run_and_submit_all(username):
         return f"Error fetching questions: {e}", None
 
     # 2. Run agent on all questions
-    agent  = BasicAgent()
+    agent   = BasicAgent()
     results = []
     payload = []
     for q in questions:
@@ -64,9 +66,9 @@ def run_and_submit_all(username):
 
     # 3. Submit answers
     submission = {
-        "username": username,
-        "agent_code": f"https://huggingface.co/spaces/{os.getenv('SPACE_ID')}/tree/main",
-        "answers": payload
+        "username":    username,
+        "agent_code":  f"https://huggingface.co/spaces/{os.getenv('SPACE_ID')}/tree/main",
+        "answers":     payload
     }
     try:
         sub_resp = requests.post(f"{DEFAULT_API_URL}/submit", json=submission, timeout=60)
@@ -83,6 +85,7 @@ def run_and_submit_all(username):
 
     return status, pd.DataFrame(results)
 
+
 def test_random_question(username):
     if not username:
         return "Please enter your Hugging Face username.", ""
@@ -94,7 +97,8 @@ def test_random_question(username):
     except Exception as e:
         return f"Error during test: {e}", ""
 
-# Build Gradio interface
+
+# --- Gradio UI ---
 with gr.Blocks() as demo:
     gr.Markdown("# Basic Agent Evaluation Runner")
     gr.Markdown(
@@ -120,3 +124,4 @@ with gr.Blocks() as demo:
 
 if __name__ == "__main__":
     demo.launch(debug=True, share=False)
+