Update submission.py

submission.py

@@ -1,79 +1,74 @@
-import requests
-import gradio as gr
-from agent import run_agent_on_question
-from utils import format_answer
-from retriever import retrieve_relevant_context
-
-API_BASE_URL = "https://agents-course-unit4-scoring.hf.space"
-
-def fetch_questions():
-    """
-    Retrieve all evaluation questions from the API.
-    """
-    response = requests.get(f"{API_BASE_URL}/questions")
-    return response.json() if response.status_code == 200 else []
-
-def submit_answers_to_leaderboard(username, agent_code_url):
-    """
-    Runs the agent on all evaluation questions and submits answers.
-    """
-    questions = fetch_questions()
-    print(f"Fetched {len(questions)} questions.")
-
-    answers = []
-
-    for q in questions:
-        task_id = q["task_id"]
-        question = q["question"]
-
-        # Add RAG step (if needed)
-        retrieved_context = retrieve_context(task_id)
-
-        print(f"\n🔍 Running agent on task: {task_id}")
-        response = run_agent_on_question(question + "\n" + retrieved_context)
-        formatted_answer = format_answer(response)
-
-        print(f"Answer: {formatted_answer}")
-        answers.append({
-            "task_id": task_id,
-            "submitted_answer": formatted_answer
-        })
-
-    # Prepare final submission payload
-    submission = {
-        "username": username,
-        "agent_code": agent_code_url,
-        "answers": answers
-    }
-
-    res = requests.post(f"{API_BASE_URL}/submit", json=submission)
-    if res.status_code == 200:
-        print("\n✅ Submission Complete!")
-        print("Result:", res.json())
-        return res.json()
-    else:
-        print("\n❌ Submission Failed!")
-        print("Status Code:", res.status_code)
-        print("Response:", res.text)
-        return None
-
-# Optional Gradio UI for easier submission
-with gr.Blocks() as demo:
-    gr.Markdown("## 🤖 GAIA Agent Submission")
-
-    with gr.Row():
-        username = gr.Textbox(label="Your Hugging Face Username")
-        agent_code_url = gr.Textbox(label="Public URL to Your Hugging Face Space Code")
-
-    submit_btn = gr.Button("Run Evaluation & Submit All Answers")
-
-    output = gr.Textbox(label="Submission Result")
-
-    submit_btn.click(
-        fn=submit_answers_to_leaderboard,
-        inputs=[username, agent_code_url],
-        outputs=[output]
-    )
-
-if __name__ == "__main__":
-    demo.launch()
+# submission.py
+
+import requests
+import gradio as gr
+from agent import run_agent_on_question
+from utils import format_answer
+
+API_BASE_URL = "https://agents-course-unit4-scoring.hf.space"
+
+def fetch_questions():
+    """
+    Retrieve all evaluation questions from the API.
+    """
+    response = requests.get(f"{API_BASE_URL}/questions")
+    return response.json() if response.status_code == 200 else []
+
+def submit_answers_to_leaderboard(username, agent_code_url):
+    """
+    Runs the agent on all evaluation questions and submits answers.
+    """
+    questions = fetch_questions()
+    print(f"Fetched {len(questions)} questions.")
+
+    answers = []
+
+    for q in questions:
+        print(f"\n🔍 Running agent on task: {q['task_id']}")
+        response = run_agent_on_question(q)  # ✅ Pass full task dictionary
+        formatted_answer = format_answer(response)
+
+        print(f"Answer: {formatted_answer}")
+        answers.append({
+            "task_id": q["task_id"],
+            "submitted_answer": formatted_answer
+        })
+
+    # Prepare final submission payload
+    submission = {
+        "username": username,
+        "agent_code": agent_code_url,
+        "answers": answers
+    }
+
+    res = requests.post(f"{API_BASE_URL}/submit", json=submission)
+    if res.status_code == 200:
+        print("\n✅ Submission Complete!")
+        print("Result:", res.json())
+        return res.json()
+    else:
+        print("\n❌ Submission Failed!")
+        print("Status Code:", res.status_code)
+        print("Response:", res.text)
+        return None
+
+# Optional Gradio UI for easier submission
+with gr.Blocks() as demo:
+    gr.Markdown("## 🤖 GAIA Agent Submission")
+
+    with gr.Row():
+        username = gr.Textbox(label="Your Hugging Face Username")
+        agent_code_url = gr.Textbox(label="Public URL to Your Hugging Face Space Code")
+
+    submit_btn = gr.Button("Run Evaluation & Submit All Answers")
+
+    output = gr.Textbox(label="Submission Result")
+
+    submit_btn.click(
+        fn=submit_answers_to_leaderboard,
+        inputs=[username, agent_code_url],
+        outputs=[output]
+    )
+
+if __name__ == "__main__":
+    demo.launch()