Update app.py

app.py

@@ -3,7 +3,7 @@ import requests
 import pandas as pd
 import gradio as gr
 
-from agent import BasicAgent  # <-- Change to match your agent.py
+from agent import BasicAgent
 
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
@@ -11,6 +11,7 @@ class GAIAAgent:
     def __init__(self):
         self.agent = BasicAgent()
     def __call__(self, question: str, task_id: str) -> str:
+        # Pass both question and task_id to the agent
         return self.agent(question, task_id)
 
 def run_and_submit_all(profile: gr.OAuthProfile | None):
@@ -63,7 +64,7 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
     except Exception as e:
         return f"Submission Failed: {e}", pd.DataFrame(results_log)
 
-# --- Gradio Interface ---
+# --- Gradio Interface matching original benchmark ---
 with gr.Blocks() as demo:
     gr.Markdown("# Basic Agent Evaluation Runner")
     gr.Markdown("""