Update app.py

app.py

@@ -112,22 +112,7 @@ def pvsnp(problem):
               output.append(message.pretty_print())
     print (output)'''
     output = agent.invoke({"messages": "Use tools to answer: "+problem})
-
-# Assuming `output` is a JSON-formatted string
-    messages = output.get("messages", [])
-    answer = None
-    for msg in reversed(messages):
-        if msg.get("content"):
-            answer = msg["content"]
-            break
-
-# Extract tool names used in tool_calls
-    tools_used = []
-    for msg in messages:
-        if "tool_calls" in msg:
-            for tool_call in msg["tool_calls"]:
-                tools_used.append(tool_call["name"])
-    #response = "Answer: "+answer+'\n'+"Tools used: "+tools_used
+    answer = output['messages'][5]
     return answer