Update app.py

app.py

@@ -3,7 +3,6 @@ import gradio as gr
 import requests
 import pandas as pd
 import json
-import io
 import re
 from langchain_core.messages import HumanMessage
 from langgraph_agent import build_graph
@@ -95,23 +94,16 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
             print(f"Skipping item with missing task_id or question: {item}")
             continue
         try:
-            model_answer = agent(question_text, task_id=task_id)
-            model_answer = model_answer.strip()
-            # Clean trailing quotes/brackets if any
-            if model_answer.endswith("]'") or model_answer.endswith(']"'):
-                model_answer = model_answer[:-2].strip()
+            model_answer = agent(question_text, task_id=task_id).strip()
+            # Clean trailing quotes if any
             if (model_answer.startswith('"') and model_answer.endswith('"')) or \
                (model_answer.startswith("'") and model_answer.endswith("'")):
                 model_answer = model_answer[1:-1].strip()
 
-            # Placeholder reasoning trace; replace with actual reasoning if available
-            reasoning_trace = f"Reasoning steps for task {task_id}."
-
             print(f"Answer for task {task_id}: '{model_answer}'")
             answers_payload.append({
                 "task_id": task_id,
-                "model_answer": model_answer,
-                "reasoning_trace": reasoning_trace
+                "submitted_answer": model_answer
             })
             results_log.append({
                 "Task ID": task_id,
@@ -122,8 +114,7 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
             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": ""
+                "submitted_answer": f"AGENT ERROR: {e}"
             })
             results_log.append({
                 "Task ID": task_id,
@@ -135,17 +126,10 @@ 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)
 
-    # Serialize to JSON Lines string
-    json_lines_str = "\n".join(json.dumps(ans, ensure_ascii=False) for ans in answers_payload)
-    file_like = io.BytesIO(json_lines_str.encode("utf-8"))
-    file_like.name = "submission.jsonl"
-
     data = {
         "username": username.strip(),
-        "agent_code": agent_code
-    }
-    files = {
-        "file": (file_like.name, file_like, "application/jsonl")
+        "agent_code": agent_code,
+        "answers": answers_payload
     }
 
     print(f"Agent finished. Submitting {len(answers_payload)} answers for user '{username}'...")
@@ -155,7 +139,7 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
 
     print(f"Submitting answers to: {submit_url}")
     try:
-        response = requests.post(submit_url, data=data, files=files, timeout=60)
+        response = requests.post(submit_url, json=data, timeout=60)
         response.raise_for_status()
         result_data = response.json()
         final_status = (