lenML/ChatTTS-Forge

1import os2 3from modules import config4 5 6def get_env_val(key, val_type):7    env_val = os.getenv(key.upper())8    if env_val is not None and env_val != "":9        if val_type == bool:10            return env_val.lower() in ("true", "1", "t")11        return val_type(env_val)12 13    if env_val == "":14        return None15 16    return env_val17 18 19def get_env_or_arg(args, arg_name, default, arg_type):20    arg_val = getattr(args, arg_name)21    env_val = get_env_val(arg_name, arg_type)22 23    if arg_type == bool and env_val is not None:24        return env_val25 26    if arg_val is not None:27        return arg_val28    elif env_val is not None:29        return env_val30 31    return default32 33 34def get_and_update_env(*args):35    val = get_env_or_arg(*args)36    key = args[1]37    config.runtime_env_vars[key] = val38    return val39