Update app.py

app.py

@@ -126,7 +126,7 @@ def pvsnp(problem):
               output.append(message.pretty_print())
     print (output)'''
     output = agent.invoke({"messages": "Use tools to answer: "+problem})
-    answer = output['messages'][5]
+    answer = output['messages']
     critic_answer = llm_response(f'''Given the problem: {problem} and the agent response: {answer}, come up with a user friendly explanation that highlights the answer along
     with the tools leveraged.''')
     final_answer = f"""