// 创建 WebSocket 连接
const socket = new WebSocket('ws://your-websocket-server-url');

// 监听 WebSocket 连接状态
socket.addEventListener('open', () => {
    console.log('WebSocket 连接已建立');
});

socket.addEventListener('error', (error) => {
    console.error('WebSocket 错误:', error);
});

socket.addEventListener('close', () => {
    console.log('WebSocket 连接已关闭');
});

// 获取按钮元素
const alwaysShowBtn = document.getElementById('always-show');
const smartShowBtn = document.getElementById('smart-show');

// 为 "始终显示" 按钮添加点击事件
alwaysShowBtn.addEventListener('click', () => {
    // 向服务器发送消息
    const message = { action: 'set-always-show' };
    socket.send(JSON.stringify(message));

    // 显示提示信息
    alert('已设置为始终显示AI反诈助理');
});

// 为 "智能显示" 按钮添加点击事件
smartShowBtn.addEventListener('click', () => {
    // 向服务器发送消息
    const message = { action: 'set-smart-show' };
    socket.send(JSON.stringify(message));

    // 显示提示信息
    alert('已设置为智能显示AI反诈助理');
});

// 监听服务器返回的消息
socket.addEventListener('message', (event) => {
    const data = JSON.parse(event.data);

    // 根据服务器返回的消息执行相应操作
    if (data.status === 'success') {
        console.log('操作成功:', data.message);
    } else if (data.status === 'error') {
        console.error('操作失败:', data.message);
    }
});