Update app.py

app.py

@@ -711,13 +711,14 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
     # --- Determine HF Space Runtime URL and Repo URL ---
     space_id = os.getenv("SPACE_ID") # Get the SPACE_ID for sending link to the code
     agent_code = f"https://huggingface.co/spaces/{space_id}/tree/main" if space_id else ""
+    answers_payload = []
 
     if profile:
         username= f"{profile.username}"
         print(f"User logged in: {username}")
     else:
         print("User not logged in.")
-        return "Please Login to Hugging Face with the button.", None
+        return "Please Login to Hugging Face with the button.", pd.DataFrame([]), answers_payload
 
     api_url = DEFAULT_API_URL
     questions_url = f"{api_url}/questions"
@@ -746,7 +747,7 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
         print(f"Fetched {len(questions_data)} questions.")
     except requests.exceptions.RequestException as e:
         print(f"Error fetching questions: {e}")
-        return f"Error fetching questions: {e}", None
+        return f"Error fetching questions: {e}", pd.DataFrame([]), answers_payload
     except requests.exceptions.JSONDecodeError as e:
         print(f"Error decoding JSON response from questions endpoint: {e}")
         print(f"Response text: {response.text[:500]}")
@@ -757,7 +758,6 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
 
     # 3. Run your Agent
     results_log = []
-    answers_payload = []
     print(f"Running agent on {len(questions_data)} questions...")
     for item in questions_data:
         task_id = item.get("task_id")
@@ -775,7 +775,7 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
 
     if not answers_payload:
         print("Agent did not produce any answers to submit.")
-        return "Agent did not produce any answers to submit.", pd.DataFrame(results_log)
+        return "Agent did not produce any answers to submit.", pd.DataFrame(results_log), answers_payload
 
     # 4. Prepare Submission
     submission_data = {"username": username.strip(), "agent_code": agent_code, "answers": answers_payload}
@@ -797,7 +797,8 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
         )
         print("Submission successful.")
         results_df = pd.DataFrame(results_log)
-        return final_status, results_df
+
+        return final_status, results_df, answers_payload
     except requests.exceptions.HTTPError as e:
         error_detail = f"Server responded with status {e.response.status_code}."
         try:
@@ -823,20 +824,21 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
         status_message = f"An unexpected error occurred during submission: {e}"
         print(status_message)
         results_df = pd.DataFrame(results_log)
-        return status_message, results_df
-
-answers_state = gr.State([])
+        return status_message, results_df, answers_payload
 
 def build_jsonl(answers):
     import json, os
     path = "/tmp/submission.jsonl"
+    if not isinstance(answers, list):
+        answers = []
     with open(path, "w", encoding="utf-8") as f:
         for a in answers:
-            f.write(json.dumps(
-                {"task_id": a["task_id"], "model_answer": a["submitted_answer"]},
-                ensure_ascii=False
-            ) + "\n")
-    return path  # gr.File/DownloadButton will serve this file
+            tid = a.get("task_id")
+            ans = a.get("submitted_answer")
+            if tid is None or ans is None:
+                continue
+            f.write(json.dumps({"task_id": tid, "model_answer": ans}, ensure_ascii=False) + "\n")
+    return path
     
 # --- Build Gradio Interface using Blocks ---
 with gr.Blocks() as demo:
@@ -857,11 +859,12 @@ with gr.Blocks() as demo:
     )
 
     gr.LoginButton()
-
     run_button = gr.Button("Run Evaluation & Submit All Answers")
 
     status_output = gr.Textbox(label="Run Status / Submission Result", lines=5, interactive=False)
     results_table = gr.DataFrame(label="Questions and Agent Answers", wrap=True)
+
+    answers_state = gr.State([])  
     jsonl_file = gr.File(label="submission.jsonl", interactive=False)
     save_btn = gr.Button("Build JSONL for Leaderboard")