Update app.py

app.py

@@ -45,8 +45,9 @@ async def run_and_submit_all(profile: gr.OAuthProfile | None):
         question_text = item.get("question")
         if not task_id or question_text is None:
             continue
+
     try:
-        # Construct full prompt with required system instructions
+    # Full prompt with required system instructions
         system_prompt = (
             "You are a general AI assistant. I will ask you a question. "
             "Report your thoughts, and finish your answer with the following template: "
@@ -61,26 +62,22 @@ async def run_and_submit_all(profile: gr.OAuthProfile | None):
         loop = asyncio.get_event_loop()
         agent_response = await loop.run_in_executor(None, agent, full_prompt)
     
-        # Extract FINAL ANSWER and reasoning
+        # Try to extract FINAL ANSWER
         if "FINAL ANSWER:" in agent_response:
-            reasoning_trace, final_answer = agent_response.rsplit("FINAL ANSWER:", 1)
+            _, final_answer = agent_response.rsplit("FINAL ANSWER:", 1)
             final_answer = final_answer.strip()
-            reasoning_trace = reasoning_trace.strip()
         else:
             final_answer = agent_response.strip()
-            reasoning_trace = "Model did not follow format; full response returned."
-
+    
         answers_payload.append({
             "task_id": task_id,
-            "model_answer": final_answer,
-            "reasoning_trace": reasoning_trace
+            "model_answer": final_answer
         })
-    
+
         results_log.append({
             "Task ID": task_id,
             "Question": question_text,
-            "Submitted Answer": final_answer,
-            "Reasoning Trace": reasoning_trace
+            "Submitted Answer": final_answer
         })
 
     except Exception as e:
@@ -88,10 +85,10 @@ async def run_and_submit_all(profile: gr.OAuthProfile | None):
         results_log.append({
             "Task ID": task_id,
             "Question": question_text,
-            "Submitted Answer": f"AGENT ERROR: {e}",
-            "Reasoning Trace": "N/A"
+            "Submitted Answer": f"AGENT ERROR: {e}"
         })
 
+
     if not answers_payload:
         return "Agent did not produce any answers to submit.", pd.DataFrame(results_log)