Update app.py

app.py

@@ -72,9 +72,9 @@ class BasicAgent:
 # GAIA RUN + SUBMIT
 # -----------------------------
 def run_and_submit_all(profile):
-    # Check if logged in
-    if profile is None or "username" not in profile:
-        return "Please login to Hugging Face", pd.DataFrame()
+    # Check login
+    if not profile or "username" not in profile:
+        return "⚠️ Please login to Hugging Face", pd.DataFrame()
 
     username = profile["username"]
     space_id = os.getenv("SPACE_ID", "Annessha18/gaia-agent")
@@ -86,12 +86,13 @@ def run_and_submit_all(profile):
 
     agent = BasicAgent()
 
+    # Fetch questions
     try:
         response = requests.get(questions_url, timeout=15)
         response.raise_for_status()
         questions = response.json()
     except Exception as e:
-        return f"Error fetching questions: {e}", pd.DataFrame()
+        return f"❌ Error fetching questions: {e}", pd.DataFrame()
 
     answers = []
     log = []
@@ -100,15 +101,8 @@ def run_and_submit_all(profile):
         task_id = q.get("task_id", "unknown")
         question_text = q.get("question", "")
         answer = agent(question_text)
-        answers.append({
-            "task_id": task_id,
-            "submitted_answer": answer
-        })
-        log.append({
-            "Task ID": task_id,
-            "Question": question_text,
-            "Answer": answer
-        })
+        answers.append({"task_id": task_id, "submitted_answer": answer})
+        log.append({"Task ID": task_id, "Question": question_text, "Answer": answer})
 
     payload = {
         "username": username,
@@ -116,19 +110,20 @@ def run_and_submit_all(profile):
         "answers": answers
     }
 
+    # Submit answers
     try:
         response = requests.post(submit_url, json=payload, timeout=30)
         response.raise_for_status()
         result = response.json()
         status = (
-            f"Submission Successful!\n"
+            f"✅ Submission Successful!\n"
             f"User: {result.get('username')}\n"
             f"Score: {result.get('score')}%\n"
             f"Correct: {result.get('correct_count')}/{result.get('total_attempted')}\n"
             f"Message: {result.get('message')}"
         )
     except Exception as e:
-        status = f"Submission Failed: {e}"
+        status = f"❌ Submission Failed: {e}"
 
     return status, pd.DataFrame(log)
 
@@ -138,13 +133,16 @@ def run_and_submit_all(profile):
 with gr.Blocks() as demo:
     gr.Markdown("# 🤖 GAIA Level 1 Agent")
     
+    # Login button
     login_btn = gr.LoginButton(label="Login to Hugging Face")
     run_btn = gr.Button("Run Evaluation & Submit All Answers")
-
     status_out = gr.Textbox(label="Submission Result", lines=5)
     table_out = gr.Dataframe(label="Questions and Answers")
 
-    # login_btn returns a dictionary, pass it as input
-    run_btn.click(run_and_submit_all, inputs=[login_btn], outputs=[status_out, table_out])
+    # Wrap login profile to avoid runtime error
+    def wrapped_submit(profile):
+        return run_and_submit_all(profile)
+
+    run_btn.click(wrapped_submit, inputs=[login_btn], outputs=[status_out, table_out])
 
 demo.launch()