answer checkfix

app.py

@@ -3,7 +3,7 @@ import gradio as gr
 import requests
 import pandas as pd
 import agent
-from utils import check_reverse, reverse
+from utils import check_reverse, reverse, check_fix_answer
 
 # (Keep Constants as is)
 # --- Constants ---
@@ -149,8 +149,8 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
             continue
         try:            
             with open('system_prompt.txt') as f:
-                submitted_answer = agent(f.read()+"\n\n"+question_text)
-                answers_payload.append({"task_id": task_id, "submitted_answer": submitted_answer})
+                submitted_answer = agent(f.read()+"\n\n"+f"TASK_ID:{task_id}\n"+question_text)
+                answers_payload.append({"task_id": task_id, "submitted_answer": check_fix_answer(submitted_answer)})
                 results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": submitted_answer})
         except Exception as e:
              print(f"Error running agent on task {task_id}: {e}")

utils.py

@@ -4,7 +4,5 @@ def check_reverse(question:str)->bool:
 def reverse(question:str)->str:
   return question[::-1]
 
-if __name__=='__main__':
-  question = '.epA'
-  if check_reverse(question):
-    print(reverse(question))
+def check_fix_answer(answer:str)->str:
+  return answer.split('FINAL ANSWER:')[-1].strip()