edit with re

app.py

@@ -837,12 +837,12 @@ agent = initialize_agent(
     llm=chat_llm,
     agent=AgentType.OPENAI_FUNCTIONS,#AgentType.ZERO_SHOT_REACT_DESCRIPTION,
     agent_kwargs={"system_message": system_prompt},
-    verbose=False,
+    verbose=True,
     max_iterations=15, # Increase as needed
     max_execution_time=4000, # Increase as needed
     early_stopping_method="generate",
     handle_parsing_errors=True,
-    return_intermediate_steps=False
+    # return_intermediate_steps=False
 )
 
 
@@ -915,9 +915,10 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
             # submitted_answer_raw = agent.run(question_text)
             submitted_answer = agent.run(question_text)
             if "YOUR FINAL ANSWER:" in submitted_answer:
-                scraped_answer = submitted_answer.split("YOUR FINAL ANSWER:")[-1].strip()
+                match = re.search(r"YOUR FINAL ANSWER:\s*(.+)", submitted_answer, re.IGNORECASE | re.DOTALL)
+                scraped_answer = match.group(1).strip()
             else:
-                scraped_answer = submitted_answer
+                scraped_answer = submitted_answer.strip()
             # submitted_answer = extract_final_answer(submitted_answer_raw)
             answers_payload.append({"task_id": task_id, "submitted_answer": scraped_answer})
             results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": submitted_answer})