Update app.py

app.py

@@ -102,12 +102,12 @@ def tool_selector(state: AgentState) -> AgentState:
         )
     )
     raw = LLM(state.messages + [prompt]).content.strip()
-    print(f"Tool selector response: {raw}")
+    # print(f"Tool selector response: {raw}")
     state.add(AIMessage(content=raw))
     parsed = safe_json(raw)
     # parsed = json.loads(raw)
     # print("parsed : ", parsed)
-    print(f"Parsed: {parsed},  type: {type(parsed)}")
+    # print(f"Parsed: {parsed},  type: {type(parsed)}")
     if not parsed or "action" not in parsed:
         state.next_action = "final"
         return state
@@ -163,12 +163,12 @@ def final_node(state: AgentState) -> AgentState:
         "{\"final_answer\":\"…\"} (no markdown, no commentary)."
     )
     raw = LLM(state.messages + [wrap]).content.strip()
-    print("raw : ", raw)
+    # print("raw : ", raw)
     state.add(AIMessage(content=raw))
     parsed = safe_json(raw)
-    print("parsed : ", parsed, "type : ", type(parsed))
+    # print("parsed : ", parsed, "type : ", type(parsed))
     state.final_answer = parsed.get("final_answer") if parsed else "Unable to parse final answer."
-    print("state.final_answer : ", state.final_answer)
+    # print("state.final_answer : ", state.final_answer)
     return state
 
 # ─────────────────────────── Graph wiring ───────────────────────────────
@@ -225,9 +225,9 @@ def answer(question: str, task_id: Optional[str] = None) -> str:
     state = AgentState(user_question=question, task_id=task_id)
     state.add(SystemMessage(content="You are a helpful assistant."))
     state.add(HumanMessage(content=question))
-    compiled_graph.invoke(state)
-    print(f"Final answer: {state.final_answer}")
-    return state.final_answer or "No answer."
+    ans = compiled_graph.invoke(state)
+    print(f"Final answer: {ans.final_answer}")
+    return ans.final_answer or "No answer."