Update app.py

app.py

@@ -42,11 +42,11 @@ def safe_json(text: str) -> Optional[Dict[str, Any]]:
         return None
 
 
-def brief(d: Dict[str, Any]) -> str:
-    for k in ("wiki_result", "ocr_result", "transcript"):
-        if k in d:
-            return f"{k}: {str(d[k])[:160].replace('\n', ' ')}…"
-    return "(no output)"
+# def brief(d: Dict[str, Any]) -> str:
+#     for k in ("wiki_result", "ocr_result", "transcript"):
+#         if k in d:
+#             return f"{k}: {str(d[k])[:160].replace('\n', ' ')}…"
+#     return "(no output)"
 
 # ─────────────────────────── Agent state ⬇ ───────────────────────────────
 
@@ -87,7 +87,7 @@ def tool_selector(state: AgentState) -> AgentState:
 def wiki_tool(state: AgentState) -> AgentState:
     out = wikipedia_search_tool({"wiki_query": state.query or ""})
     state.tool_calls += 1
-    state.add(SystemMessage(content=f"WIKI_TOOL_OUT: {brief(out)}"))
+    state.add(SystemMessage(content=f"WIKI_TOOL_OUT: {out}"))
     state.next_action = None
     return state
 
@@ -95,7 +95,7 @@ def wiki_tool(state: AgentState) -> AgentState:
 def ocr_tool(state: AgentState) -> AgentState:
     out = ocr_image_tool({"task_id": state.task_id, "ocr_path": ""})
     state.tool_calls += 1
-    state.add(SystemMessage(content=f"OCR_TOOL_OUT: {brief(out)}"))
+    state.add(SystemMessage(content=f"OCR_TOOL_OUT: {out}"))
     state.next_action = None
     return state
 
@@ -103,7 +103,7 @@ def ocr_tool(state: AgentState) -> AgentState:
 def audio_tool(state: AgentState) -> AgentState:
     out = audio_transcriber_tool({"task_id": state.task_id, "audio_path": ""})
     state.tool_calls += 1
-    state.add(SystemMessage(content=f"AUDIO_TOOL_OUT: {brief(out)}"))
+    state.add(SystemMessage(content=f"AUDIO_TOOL_OUT: {out}"))
     state.next_action = None
     return state
 
@@ -114,7 +114,7 @@ def excel_tool(state: AgentState) -> AgentState:
     })
     out = {"excel_result": result}
     state.tool_calls += 1
-    state.add(SystemMessage(content=f"EXCEL_TOOL_OUT: {brief(out)}"))
+    state.add(SystemMessage(content=f"EXCEL_TOOL_OUT: {out}"))
     state.next_action = None
     return state