Method handle_chat

web_demo/server/server.py:237–1118

(self, request)

            log("info", "Global TTS process stopped")

    async def handle_chat(self, request):
        # Set heartbeat interval to 30 seconds and timeout to 60 seconds to prevent connections from being disconnected by intermediate proxies/firewalls
        ws = web.WebSocketResponse(heartbeat=30.0, receive_timeout=None)
        await ws.prepare(request)

        client_id = f"Client-{id(ws)}"
        turn_counter = 0
        is_recording = True 
        
        # system_prompt for current session
        session_system_prompt = self.system_prompt
        
        # history
        messages = []
        audio_list = []

        async def recv_loop():
            nonlocal close, is_recording, turn_counter, opus_reader
            try:
                async for message in ws:
                    if message.type == aiohttp.WSMsgType.ERROR:
                        log("error", f"{ws.exception()}")
                        break
                    elif message.type == aiohttp.WSMsgType.CLOSED:
                        break
                    elif message.type != aiohttp.WSMsgType.BINARY:
                        log("error", f"unexpected message type {message.type}")
                        continue
                    message = message.data
                    if not isinstance(message, bytes):
                        log("error", f"unsupported message type {type(message)}")
                        continue
                    if len(message) == 0:
                        log("warning", "empty message")
                        continue
                    
                    try:
                        decoded = decode_message(message)
                        msg_type = decoded['type']
                        
                        if msg_type == 'audio':
                            if is_recording:
                                payload = decoded['data']
                                log("info", f"Received audio data: {len(payload)} bytes")
                                pcm = opus_reader.append_bytes(payload)
                                if pcm is not None and len(pcm) > 0:
                                    await pcm_queue.put(pcm)
                        
                        elif msg_type == 'control':
                            action = decoded['action']
                            if action == 'pause':
                                log("info", f"Received PAUSE signal")
                                is_recording = False
                                await save_audio_queue.put(('pause', None))
                            elif action == 'start':
                                log("info", f"Received START signal")
                                is_recording = True
                                turn_counter += 1