
                // Display error message
                displayMessage(message.content, 'error');
                enableInput();
            }
            else {
                // Display regular message
                displayMessage(message.content, message.source);
            }
        };

        ws.onerror = function(error) {
            displayMessage("WebSocket error occurred. Please refresh the page.", 'error');
            enableInput();
        };

        ws.onclose = function() {
           