import asyncio
import websockets
import json

async def stream_transcription(audio_source):
    uri = "wss://api.hanzo.ai/v1/audio/stream"
    headers = {"Authorization": f"Bearer {HANZO_API_KEY}"}

    async with websockets.connect(uri, extra_headers=headers) as ws:
        await ws.send(json.dumps({
            "model": "zen3-asr-v1",
            "language": "en",
            "interim_results": True,
        }))

        async for audio_chunk in audio_source:
            await ws.send(audio_chunk)

        await ws.send(json.dumps({"type": "end"}))

        async for message in ws:
            result = json.loads(message)
            print(result.get("transcript", ""))

asyncio.run(stream_transcription(audio_source))