Function run_mic

client_python.py:89–152

Stream from microphone.

(args)

async def run_mic(args):
    """Stream from microphone."""
    try:
        import sounddevice as sd
    except ImportError:
        print("Please install sounddevice: pip install sounddevice")
        sys.exit(1)

    print(f"Connecting to {args.server}...")
    async with websockets.connect(args.server, ping_interval=None) as ws:
        await ws.send("START")
        resp = await ws.recv()
        event = json.loads(resp)
        if event.get("event") != "started":
            print(f"Unexpected response: {resp}")
            return

        if args.hotwords:
            await ws.send(f"HOTWORDS:{args.hotwords}")
            await ws.recv()

        print("Recording... Press Ctrl+C to stop\n")

        audio_queue = asyncio.Queue()

        def audio_callback(indata, frames, time_info, status):
            audio_queue.put_nowait(indata.copy())

        stream = sd.InputStream(
            samplerate=SAMPLE_RATE, channels=1, dtype='int16',
            blocksize=CHUNK_SAMPLES, callback=audio_callback,
        )

        async def send_audio():
            with stream:
                while True:
                    chunk = await audio_queue.get()
                    await ws.send(chunk.tobytes())

        async def recv_results():
            async for msg in ws:
                data = json.loads(msg)
                if "sentences" in data:
                    print_result(data, show_spk=args.spk)
                if data.get("is_final") or data.get("event") == "stopped":
                    break

        send_task = asyncio.create_task(send_audio())
        recv_task = asyncio.create_task(recv_results())

        try:
            await asyncio.gather(send_task, recv_task)
        except (KeyboardInterrupt, asyncio.CancelledError):
            pass
        finally:
            send_task.cancel()
            if ws.open:
                await ws.send("STOP")