Update app.py

app.py

@@ -191,19 +191,16 @@ def update_and_resubmit_one_answer(task_id_to_redo: str, profile: gr.OAuthProfil
 
     # 3. Run your Agent
     results_log = []
+    new_payload = []
+
     with open("saved_answers.json", "r") as f:
         saved_payload = json.load(f)
 
-    new_payload = []
-    for a in saved_payload:
-        if a["task_id"] != task_id_to_redo:
-            new_payload.append(a)
-
     print(f"Running agent on 1 question...")
     for item in questions_data:
         task_id = item.get("task_id")
+        question_text = item.get("question")
         if task_id == task_id_to_redo:
-            question_text = item.get("question")
             try:
                 messages = [HumanMessage(content=json.dumps(item))]
                 response = agent.invoke({"messages": messages})
@@ -213,6 +210,11 @@ def update_and_resubmit_one_answer(task_id_to_redo: str, profile: gr.OAuthProfil
             except Exception as e:
                  print(f"Error running agent on task {task_id}: {e}")
                  results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": f"AGENT ERROR: {e}"})
+        else:
+            for a in saved_payload:
+                if a["task_id"] == task_id:
+                    new_payload.append(a)
+                    results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": a["submitted_answer"]})
 
     
     # 4. Prepare Submission