#
# Copyright (c) 2024, Daily
#
# SPDX-License-Identifier: BSD 2-Clause License
#

import asyncio
import sys
import os

from pipecat.pipeline.pipeline import Pipeline
from pipecat.pipeline.task import PipelineParams, PipelineTask
from pipecat.pipeline.runner import PipelineRunner
from pipecat.processors.frameworks.rtvi import (
    RTVIConfig,
    RTVILLMConfig,
    RTVIProcessor,
    RTVISetup,
    RTVITTSConfig)
from pipecat.transports.services.daily import DailyParams, DailyTransport
from pipecat.vad.silero import SileroVADAnalyzer

from runner import configure

from loguru import logger

from dotenv import load_dotenv
load_dotenv(override=True)

logger.remove(0)
logger.add(sys.stderr, level="DEBUG")


async def main(room_url, token):
    transport = DailyTransport(
        room_url,
        token,
        "Realtime AI",
        DailyParams(
            audio_out_enabled=True,
            transcription_enabled=True,
            vad_enabled=True,
            vad_analyzer=SileroVADAnalyzer()
        ))

    llm = RTVILLMConfig(
        model="llama3-70b-8192",
        messages=[{"role": "system", "content": "You are a helpful assistant named Gary. Briefly say hello!"}]
    )
    tts = RTVITTSConfig(voice="79a125e8-cd45-4c13-8a67-188112f4dd22")
    setup = RTVISetup(config=RTVIConfig(llm=llm, tts=tts))

    rtai = RTVIProcessor(
        transport=transport,
        setup=setup,
        llm_api_key=os.getenv("OPENAI_API_KEY"),
        tts_api_key=os.getenv("CARTESIA_API_KEY"))

    runner = PipelineRunner()

    pipeline = Pipeline([transport.input(), rtai])

    task = PipelineTask(
        pipeline,
        params=PipelineParams(
            allow_interruptions=True,
            enable_metrics=True,
            report_only_initial_ttfb=True))

    @transport.event_handler("on_participant_joined")
    async def on_participant_joined(transport, participant):
        transport.capture_participant_transcription(participant["id"])

    await runner.run(task)

if __name__ == "__main__":
    (url, token) = configure()
    asyncio.run(main(url, token))