import os


def check_fw_local_models():
    """
    启动时检查本地是否有 Faster Whisper 模型.
    """
    model_size_list = [
        "medium",
        "medium.en",
        "distil-large-v2",
        "distil-large-v3",
        "large-v1",
        "large-v2",
        "large-v3",
    ]
    for i, size in enumerate(model_size_list):
        if os.path.exists(f"tools/asr/models/faster-whisper-{size}"):
            model_size_list[i] = size + "-local"
    return model_size_list


def get_models():
    model_size_list = [
        "medium",
        "medium.en",
        "distil-large-v2",
        "distil-large-v3",
        "large-v1",
        "large-v2",
        "large-v3",
    ]
    return model_size_list


asr_dict = {
    "达摩 ASR (中文)": {"lang": ["zh", "yue"], "size": ["large"], "path": "funasr_asr.py", "precision": ["float32"]},
    "Faster Whisper (多语种)": {
        "lang": ["auto", "zh", "en", "ja", "ko", "yue"],
        "size": get_models(),
        "path": "fasterwhisper_asr.py",
        "precision": ["float32", "float16", "int8"],
    },
}