Update routers/websockets_new.py

routers/websockets_new.py

@@ -12,8 +12,12 @@ router = APIRouter(prefix="/websockets")
 @router.websocket("/ws")
 async def get_data(websocket:WebSocket):
     await websocket.accept()
-    
-    json = await websocket.receive_json()
+    try:
+        json = await websocket.receive_json()
+    except Exception as e:
+        print(f"Error {e}")
+        return {"Error":"Something went wrong ! Please try again !"}
+        
     access_token = json['access_token']
     logging.info(f"access_token:{access_token}")
     user_query = json['brand_name'] if json.get('brand_name') is not None else None