Update pipe_method3.py

pipe_method3.py

@@ -80,10 +80,11 @@ PARTIAL_EMIT_EVERY_MS = 250
 
 
 LAST_STATE = {
-    "connected": False,
+     "connected": False,
+    "status": "idle",        # idle | connected | listening | thinking | speaking
+    "last_partial": "",
     "last_stt": "",
     "last_llm": "",
-    "last_tts": "",
     "updated_ms": 0,
 }
 
@@ -294,6 +295,9 @@ async def health():
 @app.websocket("/stream")
 async def stream(ws: WebSocket):
     await ws.accept()
+    LAST_STATE["connected"] = True
+    LAST_STATE["status"] = "connected"
+    LAST_STATE["updated_ms"] = now_ms()
     st = CallState(call_id=str(id(ws)))
     st.history = [{"role": "system", "content": SYSTEM_PROMPT}]
     P("SYS>", f"ws_open call_id={st.call_id}")
@@ -351,6 +355,10 @@ async def stream(ws: WebSocket):
             st.keepalive_task.cancel()
         if st.outbound_task:
             st.outbound_task.cancel()
+            LAST_STATE["connected"] = False
+            LAST_STATE["status"] = "idle"
+            LAST_STATE["last_partial"] = ""
+            LAST_STATE["updated_ms"] = now_ms()
         P("SYS>", "ws_closed")
 
 
@@ -389,6 +397,10 @@ async def vad_and_stt(ws: WebSocket, st: CallState, pcm16_16k: bytes, is_speech:
         if partial and partial != st.last_partial:
             st.last_partial = partial
             P("STT_PART>", partial)
+            LAST_STATE["status"] = "listening"
+            LAST_STATE["last_partial"] = partial
+            LAST_STATE["updated_ms"] = now_ms()
+
 
     if (t - st.utter_start_ms) > MAX_UTTERANCE_MS:
         await finalize_utterance(ws, st, "max_utterance")
@@ -422,6 +434,11 @@ async def finalize_utterance(ws: WebSocket, st: CallState, reason: str):
         return
 
     P("STT_FINAL>", f"{user_text}  ({reason})")
+    LAST_STATE["status"] = "thinking"
+    LAST_STATE["last_stt"] = user_text
+    LAST_STATE["last_partial"] = ""
+    LAST_STATE["updated_ms"] = now_ms()
+
 
     async def bot_job():
         async with st.bot_lock:
@@ -443,6 +460,10 @@ async def answer_and_speak(ws: WebSocket, st: CallState, user_text: str):
     ans = (ans or "").strip() or "Sorry, I didn’t catch that."
 
     P("LLM_ANS>", ans)
+    LAST_STATE["last_llm"] = ans
+    LAST_STATE["status"] = "speaking"
+    LAST_STATE["updated_ms"] = now_ms()
+
 
     st.history.append({"role": "assistant", "content": ans})
     st.history = st.history[:1] + st.history[-8:]
@@ -510,6 +531,8 @@ async def outbound_sender(ws: WebSocket, st: CallState):
                 await asyncio.sleep(0.02)
                 if st.outbound_q.empty():
                     st.bot_speaking = False
+                    LAST_STATE["status"] = "connected"
+                    LAST_STATE["updated_ms"] = now_ms()
                 st.outbound_q.task_done()
                 continue
 
@@ -531,3 +554,7 @@ async def outbound_sender(ws: WebSocket, st: CallState):
     except Exception as e:
         P("SYS>", f"outbound_sender_error={e}")
         log.exception("outbound_sender_error")
+
+@app.get("/debug/last")
+async def debug_last():
+    return LAST_STATE