from .nodes.cosyvoice_nodes import *
from .nodes.sensevoice_nodes import *

NODE_CONFIG = {
    "CosyVoiceZeroShotNode": {
        "class": CosyVoiceZeroShotNode,
        "name": "CosyVoice 3s极速克隆"
    },
    "CosyVoiceSaveSpeakerModelNode": {
        "class": CosyVoiceSaveSpeakerModelNode,
        "name": "CosyVoice 保存说话人模型"
    },
    "CosyVoiceLoadSpeakerModelNode": {
        "class": CosyVoiceLoadSpeakerModelNode,
        "name": "CosyVoice 加载说话人模型"
    },
    "CosyVoiceLoadSpeakerModelFromUrlNode": {
        "class": CosyVoiceLoadSpeakerModelFromUrlNode,
        "name": "CosyVoice 从URL加载说话人模型"
    },
}

def generate_node_mappings(node_config):
    node_class_mappings = {}
    node_display_name_mappings = {}

    for node_name, node_info in node_config.items():
        node_class_mappings[node_name] = node_info["class"]
        node_display_name_mappings[node_name] = node_info.get("name", node_info["class"].__name__)

    return node_class_mappings, node_display_name_mappings

NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS = generate_node_mappings(NODE_CONFIG)

WEB_DIRECTORY = "./web"

__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS", "WEB_DIRECTORY"]