Improve WebSocket error handling in app.py by adding exception management for connection issues and logging errors. This enhances the robustness of the chat functionality.

Author: robin-rpr

app.py

@@ -215,9 +215,15 @@ def ws_v1_chat(ws):
             chat.register(ws)
 
             while True:
-                # Receive message from client
-                message = ws.receive()
-                if not message:
+                # Receive.
+                try:
+                    message = ws.receive()
+                    if not message:
+                        break
+                except Exception as e:
+                    if 'ConnectionClosed' in str(type(e)):
+                        break
+                    app.logger.error(f"WebSocket error: {str(e)}", exc_info=True)
                     break
 
                 try: