Update app.py

app.py

@@ -140,11 +140,18 @@ async def websocket_endpoint(websocket: WebSocket):
     await websocket.accept()
     try:
         async for response in process_audio_stream(websocket):
-            await websocket.send_text(response)
+            try:
+                await websocket.send_text(response)
+            except Exception as send_error:
+                print(f"Error sending WebSocket message: {send_error}")
+                break
     except Exception as e:
         print(f"WebSocket error: {e}")
     finally:
-        await websocket.close()
+        try:
+            await websocket.close(code=1000)
+        except Exception as close_error:
+            print(f"Error closing WebSocket: {close_error}")
 
 if __name__ == "__main__":
     import uvicorn