Update app.py

app.py

@@ -1,4 +1,3 @@
-# app.py
 import os
 import gradio as gr
 import requests
@@ -6,36 +5,35 @@ import pandas as pd
 
 from tools import AnswerTool
 from smolagents import CodeAgent, OpenAIServerModel
-from smolagents import DuckDuckGoSearchTool   # ← importiamo il search tool
+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, AnswerTool e DuckDuckGoSearchTool
-        model = OpenAIServerModel(model_id="gpt-4o")
+        # Initialize CodeAgent with GPT-4o, custom AnswerTool, DuckDuckGo and Wikipedia tools
+        model       = OpenAIServerModel(model_id="gpt-4o")
         answer_tool = AnswerTool()
         web_tool    = DuckDuckGoSearchTool()
+        wiki_tool   = WikipediaSearchTool()
         self.agent = CodeAgent(
             model=model,
-            tools=[answer_tool, web_tool],  # ← due tool
-            add_base_tools=False,           # niente altri tool
-            max_steps=2,                    # fino a due passaggi
+            tools=[answer_tool, web_tool, wiki_tool],
+            add_base_tools=True,
+            max_steps=2,
             verbosity_level=0
         )
 
     def __call__(self, question: str) -> str:
-        # Fai girare l'agente: prima prova AnswerTool, poi DuckDuckGo se serve
+        # Run the agent on the question
         return self.agent.run(question)
 
-
 def run_and_submit_all(username):
-    # Username fornito manualmente
     if not username:
         return "Please enter your Hugging Face username.", None
 
-    # Fetch questions
+    # 1. Fetch questions
     try:
         resp = requests.get(f"{DEFAULT_API_URL}/questions", timeout=15)
         if resp.status_code == 429:
@@ -45,30 +43,30 @@ def run_and_submit_all(username):
     except Exception as e:
         return f"Error fetching questions: {e}", None
 
-    # Run agent on all questions
-    agent = BasicAgent()
+    # 2. Run agent on all questions
+    agent  = BasicAgent()
     results = []
     payload = []
     for q in questions:
-        tid = q.get('task_id')
-        text = q.get('question')
+        tid  = q.get("task_id")
+        text = q.get("question")
         if not (tid and text):
             continue
         try:
             ans = agent(text)
         except Exception as e:
             ans = f"ERROR: {e}"
-        results.append({'Task ID': tid, 'Question': text, 'Answer': ans})
-        payload.append({'task_id': tid, 'submitted_answer': ans})
+        results.append({"Task ID": tid, "Question": text, "Answer": ans})
+        payload.append({"task_id": tid, "submitted_answer": ans})
 
     if not payload:
         return "Agent returned no answers.", pd.DataFrame(results)
 
-    # Submit answers
+    # 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)
@@ -85,20 +83,18 @@ 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.", ""
     try:
         q = requests.get(f"{DEFAULT_API_URL}/random-question", timeout=15).json()
-        question = q.get('question', '')
-        ans = BasicAgent()(question)
+        question = q.get("question", "")
+        ans      = BasicAgent()(question)
         return question, ans
     except Exception as e:
         return f"Error during test: {e}", ""
 
-
-# --- Gradio UI ---
+# Build Gradio interface
 with gr.Blocks() as demo:
     gr.Markdown("# Basic Agent Evaluation Runner")
     gr.Markdown(
@@ -111,8 +107,8 @@ with gr.Blocks() as demo:
     )
 
     username_input = gr.Textbox(label="Hugging Face Username", placeholder="your-username")
-    run_btn = gr.Button("Run Evaluation & Submit All Answers")
-    test_btn = gr.Button("Test Random Question")
+    run_btn        = gr.Button("Run Evaluation & Submit All Answers")
+    test_btn       = gr.Button("Test Random Question")
 
     status_out   = gr.Textbox(label="Status / Result", lines=5, interactive=False)
     table_out    = gr.DataFrame(label="Full Results Table", wrap=True)