import os
import sys
from typing import Dict, Any

# Add the current directory to the Python path
sys.path.insert(0, os.path.dirname(os.path.realpath(__file__)))

# Import node mappings from each module
# 使用相对导入
from .model_loader import NODE_CLASS_MAPPINGS as MODEL_LOADER_NODES
from .image_understanding import NODE_CLASS_MAPPINGS as IMAGE_UNDERSTANDING_NODES
from .image_generation import NODE_CLASS_MAPPINGS as IMAGE_GENERATION_NODES

# Combine all node mappings
NODE_CLASS_MAPPINGS: Dict[str, Any] = {
    **MODEL_LOADER_NODES,
    **IMAGE_UNDERSTANDING_NODES,
    **IMAGE_GENERATION_NODES
}

# Define display names for nodes
NODE_DISPLAY_NAME_MAPPINGS = {
    "DeepSeekModelLoader": "Loader/Multimodal Loader",
    "DeepSeekImageUnderstanding": "DeepSeek Image Understanding",
    "DeepSeekImageGeneration": "DeepSeek Image Generation"
}

__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS']

# from .Loader import *
def init_custom_nodes():
    """Initialize any custom nodes setup here"""
    print("Initializing DeepSeek Janus Pro nodes...")