Update app.py

app.py

@@ -2,6 +2,7 @@
 
 import os
 import time
+import traceback
 import requests
 import pandas as pd
 import gradio as gr
@@ -27,10 +28,7 @@ class GAIAAgent:
     def answer(self, prompt: str) -> str:
         payload = {
             "inputs": prompt,
-            "parameters": {
-                "max_new_tokens": 512,
-                "temperature": 0.2
-            }
+            "parameters": {"max_new_tokens": 512, "temperature": 0.2}
         }
         url = f"https://api-inference.huggingface.co/models/{self.model_id}"
         resp = requests.post(url, headers=self.headers, json=payload, timeout=60)
@@ -42,59 +40,62 @@ class GAIAAgent:
 
 # --- Gradio callback ---
 def run_and_submit_all(profile: gr.OAuthProfile | None):
-    if profile is None:
-        return "⚠️ Please log in with your Hugging Face account.", pd.DataFrame()
-    username = profile.username
-    hf_token = HF_TOKEN_ENV or getattr(profile, "access_token", None)
-    if not hf_token:
-        return (
-            "❌ No Hugging Face token found.\n"
-            "Set HUGGINGFACEHUB_API_TOKEN in Secrets or log in via the button.",
-            pd.DataFrame()
+    try:
+        if profile is None:
+            return ("⚠️ Please log in with your Hugging Face account.", pd.DataFrame())
+        username = profile.username
+        hf_token = HF_TOKEN_ENV or getattr(profile, "access_token", None)
+        if not hf_token:
+            return (
+                "❌ No Hugging Face token found.\n"
+                "Set HUGGINGFACEHUB_API_TOKEN in Secrets or log in via the button.",
+                pd.DataFrame()
+            )
+
+        # 1) Fetch GAIA questions
+        q_resp = requests.get(f"{API_URL}/questions", timeout=15)
+        q_resp.raise_for_status()
+        questions = q_resp.json() or []
+        if not questions:
+            return ("❌ No questions found. Check your API_URL.", pd.DataFrame())
+
+        # 2) Init agent
+        agent = GAIAAgent(MODEL_ID, hf_token)
+
+        # 3) Answer each
+        results = []
+        payload = []
+        for item in questions:
+            tid  = item.get("task_id")
+            qtxt = item.get("question", "")
+            try:
+                ans = agent.answer(qtxt)
+            except Exception as e:
+                ans = f"ERROR: {e}"
+            results.append({"Task ID": tid, "Question": qtxt, "Answer": ans})
+            payload.append({"task_id": tid, "submitted_answer": ans})
+            time.sleep(0.5)
+
+        # 4) Submit
+        submission = {"username": username, "answers": payload}
+        s_resp = requests.post(f"{API_URL}/submit", json=submission, timeout=60)
+        s_resp.raise_for_status()
+        data = s_resp.json()
+
+        # 5) Build status text
+        status = (
+            f"✅ **Submission Successful!**\n\n"
+            f"**User:** {data.get('username')}\n"
+            f"**Score:** {data.get('score')}% "
+            f"({data.get('correct_count')}/{data.get('total_attempted')} correct)\n"
+            f"**Message:** {data.get('message')}"
         )
+        return status, pd.DataFrame(results)
 
-    # 1) Fetch GAIA questions
-    q_resp = requests.get(f"{API_URL}/questions", timeout=15)
-    q_resp.raise_for_status()
-    questions = q_resp.json() or []
-    if not questions:
-        return "❌ No questions found. Check your API_URL.", pd.DataFrame()
-
-    # 2) Init agent
-    agent = GAIAAgent(MODEL_ID, hf_token)
-
-    # 3) Answer each
-    results = []
-    payload = []
-    for item in questions:
-        tid  = item.get("task_id")
-        qtxt = item.get("question", "")
-        try:
-            ans = agent.answer(qtxt)
-        except Exception as e:
-            ans = f"ERROR: {e}"
-        results.append({"Task ID": tid, "Question": qtxt, "Answer": ans})
-        payload.append({"task_id": tid, "submitted_answer": ans})
-        time.sleep(0.5)
-
-    # 4) Submit (no agent_code)
-    submission = {
-        "username": username,
-        "answers":  payload
-    }
-    s_resp = requests.post(f"{API_URL}/submit", json=submission, timeout=60)
-    s_resp.raise_for_status()
-    data = s_resp.json()
-
-    # 5) Build status text
-    status = (
-        f"✅ **Submission Successful!**\n\n"
-        f"**User:** {data.get('username')}\n"
-        f"**Score:** {data.get('score')}% "
-        f"({data.get('correct_count')}/{data.get('total_attempted')} correct)\n"
-        f"**Message:** {data.get('message')}"
-    )
-    return status, pd.DataFrame(results)
+    except Exception as e:
+        tb = traceback.format_exc()
+        print("[ERROR]", tb)
+        return (f"❌ Unexpected error:\n{e}", pd.DataFrame())
 
 # --- Gradio UI ---
 with gr.Blocks() as demo: