Update app.py

app.py

@@ -4,50 +4,77 @@ import requests
 import pandas as pd
 from agent import BasicAgent
 
+# --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
+
 def run_and_submit_all(profile: gr.OAuthProfile | None):
+    """
+    Fetches GAIA questions, queries the smolagents-based BasicAgent, and submits answers.
+    """
     if not profile:
         return "Please log in with Hugging Face", None
 
+    username = profile.username
     agent = BasicAgent()
-    # 1) fetch questions
-    questions = requests.get(f"{DEFAULT_API_URL}/questions", timeout=15).json()
 
-    # 2) run agent
+    # 1) Fetch questions
+    questions_url = f"{DEFAULT_API_URL}/questions"
+    try:
+        resp = requests.get(questions_url, timeout=15)
+        resp.raise_for_status()
+        questions = resp.json()
+    except Exception as e:
+        return f"Error fetching questions: {e}", None
+
+    # 2) Run agent on each question
     answers = []
     log = []
     for item in questions:
-        task_id = item["task_id"]
-        q = item["question"]
-        ans = agent(q)
-        answers.append({"task_id": task_id, "submitted_answer": ans})
-        log.append({"Task ID": task_id, "Question": q, "Answer": ans})
+        task_id = item.get("task_id")
+        question = item.get("question")
+        if not task_id or question is None:
+            continue
+
+        answer = agent(question)
+        answers.append({"task_id": task_id, "submitted_answer": answer})
+        log.append({"Task ID": task_id, "Question": question, "Answer": answer})
+
+    if not answers:
+        return "Agent did not produce any answers to submit.", pd.DataFrame(log)
 
-    # 3) submit
+    # 3) Submit answers
     payload = {
-        "username": profile.username,
+        "username": username,
         "agent_code": f"https://huggingface.co/spaces/{os.getenv('SPACE_ID')}/tree/main",
         "answers": answers,
     }
-    resp = requests.post(f"{DEFAULT_API_URL}/submit", json=payload, timeout=60)
-    data = resp.json()
-
-    status = (
-        f"Score: {data['score']}% "
-        f"({data['correct_count']}/{data['total_attempted']})\n"
-        f"{data.get('message','')}"
-    )
+    submit_url = f"{DEFAULT_API_URL}/submit"
+    try:
+        resp_s = requests.post(submit_url, json=payload, timeout=60)
+        resp_s.raise_for_status()
+        data = resp_s.json()
+        status = (
+            f"Score: {data.get('score', 'N/A')}% "
+            f"({data.get('correct_count', '?')}/{data.get('total_attempted', '?')})\n"
+            f"{data.get('message', '')}"
+        )
+    except Exception as e:
+        status = f"Submission Failed: {e}"
+
     return status, pd.DataFrame(log)
 
+
+# ----- Gradio UI -----
 with gr.Blocks() as demo:
     gr.Markdown("# GAIA Q&A with smolagents")
     login = gr.LoginButton()
     run_btn = gr.Button("Run & Submit All", variant="primary")
-    status_out = gr.Textbox(lines=4, label="Submission Status")
-    results_tbl = gr.DataFrame(label="Per-Question Log")
 
-    run_btn.click(fn=run_and_submit_all, inputs=[login], outputs=[status_out, results_tbl])
+    status_out = gr.Textbox(label="Submission Status", lines=4)
+    results_table = gr.DataFrame(label="Per-Question Log")
+
+    run_btn.click(fn=run_and_submit_all, inputs=[login], outputs=[status_out, results_table])
 
 if __name__ == "__main__":
     demo.launch()