import asyncio

from equos import AvatarStreamer

API_KEY = "<MY_API_KEY>"

# Define your callback for receiving mp4 video chunks
def on_video_chunk(mp4_chunk: bytes):
    print(f"Received video chunk: {len(mp4_chunk)} bytes")

def on_eos():
   print("Received end of stream")

async def main():
    # Initialize the streamer with avatar name and callback
    streamer = AvatarStreamer(api_key=API_KEY, avatar="lucy", on_video=on_video_chunk, on_eos=on_eos)
    await streamer.connect()

    # Send an audio chunk (e.g. PCM or WAV bytes)
    chunks = load_my_raw_audio_chunks()

   for chunk in chunks:
        await streamer.stream(chunk)

    # Wait for server response
    await streamer.eos()

asyncio.run(main())