"""
Pivoine Custom ComfyUI Nodes
Custom node extensions and wrappers for RunPod deployment

Author: valknar@pivoine.art
"""

from .pivoine_diffrhythm import NODE_CLASS_MAPPINGS as DIFFRHYTHM_MAPPINGS
from .pivoine_diffrhythm import NODE_DISPLAY_NAME_MAPPINGS as DIFFRHYTHM_DISPLAY

# Combine all node mappings
NODE_CLASS_MAPPINGS = {
    **DIFFRHYTHM_MAPPINGS,
}

NODE_DISPLAY_NAME_MAPPINGS = {
    **DIFFRHYTHM_DISPLAY,
}

__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS']