import asyncio
import logging

from agents import set_default_openai_client
from agents.realtime import RealtimeAgent, RealtimeRunner
from openai import AsyncAzureOpenAI

logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)


async def main():
    try:
        azure_openai_client = AsyncAzureOpenAI(
            api_key="****",
            api_version="2025-03-01-preview",
            azure_endpoint="https://***.openai.azure.com",
            # azure_deployment=azure_deployment, # somehow we don't need this anymore
        )
        set_default_openai_client(azure_openai_client)
        logger.info("✅ Azure OpenAI client configured successfully")
    except Exception as e:
        logger.error(f"❌ Failed to configure Azure OpenAI client: {e}")
        raise

    # Create the agent
    agent = RealtimeAgent(
        name="Assistant",
        instructions="You are a helpful voice assistant. Keep responses brief and conversational.",
    )

    # Set up the runner with configuration
    runner = RealtimeRunner(
        starting_agent=agent,
        config={
            "model_settings": {
                "model_name": "gpt-4o-mini-realtime-preview",
                "voice": "alloy",
                "modalities": ["text", "audio"],
                "input_audio_transcription": {"model": "whisper-1"},
                "turn_detection": {
                    "type": "server_vad",
                    "threshold": 0.5,
                    "prefix_padding_ms": 300,
                    "silence_duration_ms": 200,
                },
            }
        },
    )

    # Start the session
    session = await runner.run(model_config={})

    async with session:
        print("Session started! The agent will stream audio responses in real-time.")

        # Process events
        async for event in session:
            if event.type == "response.audio_transcript.done":
                print(f"Assistant: {event.transcript}")
            elif event.type == "conversation.item.input_audio_transcription.completed":
                print(f"User: {event.transcript}")
            elif event.type == "error":
                print(f"Error: {event.error}")
                break


if __name__ == "__main__":
    asyncio.run(main())