__init__.py

"""ComfyUI Batch Image Processing Nodes.

A set of custom nodes for batch processing images through ComfyUI pipelines.
"""

try:
    from .nodes.batch_loader import BatchImageLoader
    from .nodes.batch_saver import BatchImageSaver
    from .nodes.progress_formatter import BatchProgressFormatter

    NODE_CLASS_MAPPINGS = {
        "BatchImageLoader": BatchImageLoader,
        "BatchImageSaver": BatchImageSaver,
        "BatchProgressFormatter": BatchProgressFormatter,
    }

    NODE_DISPLAY_NAME_MAPPINGS = {
        "BatchImageLoader": "Batch Image Loader",
        "BatchImageSaver": "Batch Image Saver",
        "BatchProgressFormatter": "Batch Progress Formatter",
    }

    __all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"]

except ImportError:
    # Allow import to succeed even without ComfyUI dependencies (torch, numpy, PIL)
    # This enables running tests for modules that don't need these dependencies
    NODE_CLASS_MAPPINGS = {}
    NODE_DISPLAY_NAME_MAPPINGS = {}
    __all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"]