Update app.py

app.py

@@ -75,7 +75,9 @@ def tool_selector(state: AgentState) -> AgentState:
     raw = LLM(state.messages + [prompt]).content.strip()
     print(f"Tool selector response: {raw}")
     state.add(AIMessage(content=raw))
-    parsed = safe_json(raw)
+    # parsed = safe_json(raw)
+    parsed = json.loads(raw)
+    print("parsed : ", parsed)
     print(f"Parsed: {parsed}")
     if not parsed or "action" not in parsed:
         state.next_action = "final"