Update app.py

app.py

@@ -2,6 +2,8 @@ import os
 import gradio as gr
 import requests
 import pandas as pd
+import json
+import io
 from langchain_core.messages import HumanMessage
 from langgraph_agent import build_graph
 
@@ -101,6 +103,7 @@ 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")
@@ -110,10 +113,11 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
             continue
         try:
             submitted_answer = agent(question_text, task_id=task_id)
-            print(f"Submitting answer for task {task_id}: '{submitted_answer}'")
+            print(f"Answer for task {task_id}: '{submitted_answer}'")
             answers_payload.append({
                 "task_id": task_id,
-                "submitted_answer": submitted_answer
+                "model_answer": submitted_answer,
+                # "reasoning_trace": None  # Optional: add reasoning trace if available
             })
             results_log.append({
                 "Task ID": task_id,
@@ -122,6 +126,11 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
             })
         except Exception as e:
             print(f"Error running agent on task {task_id}: {e}")
+            answers_payload.append({
+                "task_id": task_id,
+                "model_answer": f"AGENT ERROR: {e}",
+                # "reasoning_trace": None
+            })
             results_log.append({
                 "Task ID": task_id,
                 "Question": question_text,
@@ -132,18 +141,28 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
         print("Agent did not produce any answers to submit.")
         return "Agent did not produce any answers to submit.", pd.DataFrame(results_log)
 
-    # 4. Prepare Submission 
-    submission_data = {
+    # Serialize to JSON Lines string
+    json_lines_str = "\n".join(json.dumps(ans) for ans in answers_payload)
+
+    # Prepare file-like object for upload
+    file_like = io.BytesIO(json_lines_str.encode("utf-8"))
+    file_like.name = "submission.jsonl"  # Filename required for multipart upload
+
+    # Prepare multipart form data
+    files = {
+        "file": (file_like.name, file_like, "application/jsonl")
+    }
+    data = {
         "username": username.strip(),
-        "agent_code": agent_code,
-        "answers": answers_payload
+        "agent_code": agent_code
     }
+
     print(f"Agent finished. Submitting {len(answers_payload)} answers for user '{username}'...")
 
     # 5. Submit
-    print(f"Submitting {len(answers_payload)} answers to: {submit_url}")
+    print(f"Submitting answers to: {submit_url}")
     try:
-        response = requests.post(submit_url, json=submission_data, timeout=60)
+        response = requests.post(submit_url, data=data, files=files, timeout=60)
         response.raise_for_status()
         result_data = response.json()
         final_status = (