WebSocket streaming

import json
import websockets
import asyncio

async def main():
    url = (
        "wss://api.chariot.in/v1/tts/ws"
        "?voice_id=VOICE_ID"
        "&model_type=chariot-multilingual-v1&speed=1.0"
        "&optimize_streaming_latency=0"
    )
    headers = {"chariotai-api-key": "YOUR_API_KEY"}

    async with websockets.connect(url, additional_headers=headers) as ws:
        # Buffer text as it becomes available.
        await ws.send(json.dumps({"type": "input.text", "text": "Hello and welcome. "}))
        await ws.send(json.dumps({"type": "input.text", "text": "This is Chariot streaming over a WebSocket."}))
        # Flush to generate audio for everything buffered so far.
        await ws.send(json.dumps({"type": "input.flush"}))
        # Finish the session.
        await ws.send(json.dumps({"type": "input.done"}))

        async for message in ws:
            if isinstance(message, bytes):
                play_pcm(message)  # 44.1 kHz, mono, s16le
            else:
                event = json.loads(message)
                if event["type"] == "audio.done":
                    print("charged", event["credits_utilized"], "credits")
                elif event["type"] == "session.done":
                    break
                elif event["type"] == "error":
                    raise RuntimeError(event["message"])

asyncio.run(main())