Update app.py

app.py

@@ -114,10 +114,7 @@ def pvsnp(problem):
     output = agent.invoke({"messages": "Use tools to answer: "+problem})
 
 # Assuming `output` is a JSON-formatted string
-    data = json.loads(output)
-
-# Extract the answer from the final AIMessage
-    messages = data.get("messages", [])
+    messages = output.get("messages", [])
     answer = None
     for msg in reversed(messages):
         if msg.get("content"):
@@ -130,11 +127,8 @@ def pvsnp(problem):
         if "tool_calls" in msg:
             for tool_call in msg["tool_calls"]:
                 tools_used.append(tool_call["name"])
-
-    print("Answer:", answer)
-    print("Tools used:", tools_used)
-    response = "Answer: "+answer+'\n'+"Tools used: "+tools_used
-    return response
+    #response = "Answer: "+answer+'\n'+"Tools used: "+tools_used
+    return answer
                                       
 
 iface = gr.Interface(