import os
import logging


from dotenv import load_dotenv

from achatbot.common import interface
from achatbot.common.factory import EngineClass, EngineFactory
from achatbot.types.speech.turn_analyzer.smart_turn import (
    MODELS_DIR,
    SmartTurnArgs,
)

load_dotenv(override=True)


class TurnAnalyzerEnvInit:
    @staticmethod
    def get_smart_turn_analyzer_args() -> dict:
        args = dict(
            **SmartTurnArgs(
                model_path=os.getenv(
                    "TURN_MODEL_PATH", os.path.join(MODELS_DIR, "pipecat-ai/smart-turn-v2")
                ),
                warmup_steps=int(os.getenv("TURN_WARMUP_STEPS", "2")),
                sample_rate=int(os.getenv("TURN_SAMPLE_RATE", "16000")),
                stop_secs=float(os.getenv("TURN_STOP_SECS", "3")),
                pre_speech_ms=float(os.getenv("TURN_PRE_SPEECH_MS", "0")),
                max_duration_secs=float(os.getenv("TURN_MAX_DURATION_SECS", "8")),
            )
        )
        return args

    @staticmethod
    def getEngine(tag, **kwargs) -> interface.ITurnAnalyzer | EngineClass:
        if "v2_smart_turn_analyzer" in tag:
            from . import smart_turn

        engine = EngineFactory.get_engine_by_tag(EngineClass, tag, **kwargs)
        return engine

    @staticmethod
    def initTurnAnalyzerEngine(
        tag: str | None = None, kwargs: dict | None = None
    ) -> interface.ITurnAnalyzer | EngineClass:
        tag = tag or os.getenv("TURN_ANALYZER_TAG", "v2_smart_turn_analyzer")
        kwargs = kwargs or TurnAnalyzerEnvInit.map_config_func[tag]()
        logging.info(f"initTurnAnalyzerEngine: {tag}, {kwargs}")
        engine = TurnAnalyzerEnvInit.getEngine(tag, **kwargs)
        logging.info(f"initTurnAnalyzerEngine: {tag}, {engine}")
        return engine

    map_config_func = {
        "v2_smart_turn_analyzer": get_smart_turn_analyzer_args,
    }
