from .smart_image_stitch import NODE_CLASS_MAPPINGS as STITCH_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS as STITCH_DISPLAY_MAPPINGS
from .smart_grid_stitch import NODE_CLASS_MAPPINGS as GRID_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS as GRID_DISPLAY_MAPPINGS
from .smart_panorama_stitch import NODE_CLASS_MAPPINGS as PANORAMA_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS as PANORAMA_DISPLAY_MAPPINGS
from .smart_collage_maker import NODE_CLASS_MAPPINGS as COLLAGE_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS as COLLAGE_DISPLAY_MAPPINGS
from .smart_image_splitter import NODE_CLASS_MAPPINGS as SPLITTER_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS as SPLITTER_DISPLAY_MAPPINGS
from .smart_border_frame import NODE_CLASS_MAPPINGS as BORDER_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS as BORDER_DISPLAY_MAPPINGS
from .smart_batch_processor import NODE_CLASS_MAPPINGS as BATCH_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS as BATCH_DISPLAY_MAPPINGS

# Combine all node mappings
NODE_CLASS_MAPPINGS = {}
NODE_CLASS_MAPPINGS.update(STITCH_MAPPINGS)
NODE_CLASS_MAPPINGS.update(GRID_MAPPINGS)
NODE_CLASS_MAPPINGS.update(PANORAMA_MAPPINGS)
NODE_CLASS_MAPPINGS.update(COLLAGE_MAPPINGS)
NODE_CLASS_MAPPINGS.update(SPLITTER_MAPPINGS)
NODE_CLASS_MAPPINGS.update(BORDER_MAPPINGS)
NODE_CLASS_MAPPINGS.update(BATCH_MAPPINGS)

# Combine all display name mappings
NODE_DISPLAY_NAME_MAPPINGS = {}
NODE_DISPLAY_NAME_MAPPINGS.update(STITCH_DISPLAY_MAPPINGS)
NODE_DISPLAY_NAME_MAPPINGS.update(GRID_DISPLAY_MAPPINGS)
NODE_DISPLAY_NAME_MAPPINGS.update(PANORAMA_DISPLAY_MAPPINGS)
NODE_DISPLAY_NAME_MAPPINGS.update(COLLAGE_DISPLAY_MAPPINGS)
NODE_DISPLAY_NAME_MAPPINGS.update(SPLITTER_DISPLAY_MAPPINGS)
NODE_DISPLAY_NAME_MAPPINGS.update(BORDER_DISPLAY_MAPPINGS)
NODE_DISPLAY_NAME_MAPPINGS.update(BATCH_DISPLAY_MAPPINGS)

__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS']