import asyncio
from livekit.agents import (
    Agent,
    AgentSession,
    AutoSubscribe,
    RoomInputOptions,
    RoomOutputOptions,
    RunContext,
    JobContext,
    function_tool,
)
from livekit.plugins import aws, deepgram, openai, silero
from livekit.plugins.turn_detector.multilingual import MultilingualModel

# Define custom agent with tools
class Assistant(Agent):
    def __init__(self, tools):
        super().__init__(
            instructions="You are a helpful voice AI assistant.",
            tools=tools,
        )

# Function tools example
@function_tool
async def get_weather(location: str) -> dict:
    """Get weather for a location."""
    return {"weather": "sunny", "temperature": 70}

async def entrypoint(ctx: JobContext):
    await ctx.connect(auto_subscribe=AutoSubscribe.AUDIO_ONLY)
    
    # Wait for participant
    participant = await ctx.wait_for_participant()
    
    # Create AgentSession with comprehensive configuration
    session = AgentSession(
        # LLM Configuration - using AWS Realtime Model
        llm=aws.realtime.RealtimeModel(
            tool_choice="auto",
            max_tokens=10_000,
        ),
        # Alternative: Standard LLM configuration
        # llm=openai.LLM(model="gpt-4o-mini"),
        # stt=deepgram.STT(model="nova-3", language="multi"),
        # tts=openai.TTS(voice="ash"),
        
        # Voice Activity Detection
        vad=silero.VAD.load(),
        
        # Turn Detection
        turn_detection=MultilingualModel(),
    )
    
    # Start session with comprehensive options
    await session.start(
        room=ctx.room,
        agent=Assistant(tools=[get_weather]),
        room_input_options=RoomInputOptions(
            close_on_disconnect=False,
            # Optional: Enable noise cancellation
            # noise_cancellation=noise_cancellation.BVC(),
        ),
        room_output_options=RoomOutputOptions(
            audio_enabled=True,
            transcription_enabled=True,
        ),
    )

from livekit.plugins.elevenlabs import TTS, VoiceSettings

tts = TTS(
    model="eleven_turbo_v2_5",
    voice_id="EXAVITQu4vr4xnSDxMaL",
    voice_settings=VoiceSettings(
        stability=0.71,           # Controls voice consistency (0.0 - 1.0)
        similarity_boost=0.5,     # Enhances voice similarity to original (0.0 - 1.0)
        style=0.0,                # Adjusts speaking style (0.0 - 1.0)
        use_speaker_boost=True    # Improves voice clarity
    )
)

tts.update_options(
    voice_settings=VoiceSettings(
        stability=0.8,
        similarity_boost=0.6,
        style=0.2,
        use_speaker_boost=False
    )
)