Update app.py

app.py

@@ -11,6 +11,7 @@ import uvicorn
 import os
 from pathlib import Path
 import mimetypes
+from websocket_handler import websocket_endpoint
 
 from utils import log
 from chat_handler import router as chat_router  # ← start_session & chat
@@ -85,6 +86,11 @@ app.include_router(chat_router, prefix="/api")
 # ---------------- Admin API routes ----------------------------------
 app.include_router(admin_router)
 
+# ---------------- WebSocket route for real-time STT ------------------
+@app.websocket("/ws/conversation/{session_id}")
+async def conversation_websocket(websocket: WebSocket, session_id: str):
+    await websocket_endpoint(websocket, session_id)
+    
 # ---------------- Serve Angular UI if exists ------------------------
 static_dir = Path(__file__).parent / "static"
 if static_dir.exists():