handle websocket disconnect

app.py

@@ -12,6 +12,7 @@ from fastapi import (
     FastAPI,
     Request,
     WebSocket,
+    WebSocketDisconnect,
     Depends,
     HTTPException,
     status,
@@ -95,6 +96,7 @@ async def websocket_endpoint(websocket: WebSocket):
         websocket: The WebSocket connection.
     """
     await websocket.accept()
+    try:
         while True:
             data = await websocket.receive_text()
 
@@ -102,3 +104,6 @@ async def websocket_endpoint(websocket: WebSocket):
                 await websocket.send_text(chunk)
 
             await websocket.send_text("<<END>>")
+    except WebSocketDisconnect:
+        logger.info("Client disconnected")
+        await websocket.close()