"""
EndlessSeaofStars Custom Nodes for ComfyUI
Batch processing nodes with specialized support for FLUX and SDXL models
"""

from .endless_batchers import (
    EndlessNode_SimpleBatchPrompts,
    EndlessNode_FluxBatchPrompts,
    EndlessNode_SDXLBatchPrompts,
    EndlessNode_BatchNegativePrompts,
    EndlessNode_PromptCounter,
    EndlessNode_FluxKontextBatchPrompts,
    EndlessNode_ReplicateLatents,
# IGNORE ME, I AM NOT READY!!
# from .endless_fluxlatent import (
    # EndlessNode_FluxLatentReplicator,
    # EndlessNode_FluxLatentReplicatorFromPrompts,
)

# Node class mappings for ComfyUI
NODE_CLASS_MAPPINGS = {
    "SimpleBatchPrompts": EndlessNode_SimpleBatchPrompts,
    "FluxBatchPrompts": EndlessNode_FluxBatchPrompts,
    "SDXLBatchPrompts": EndlessNode_SDXLBatchPrompts,
    "BatchNegativePrompts": EndlessNode_BatchNegativePrompts,
    "PromptCounter": EndlessNode_PromptCounter,
    "FluxKontextBatchPrompts": EndlessNode_FluxKontextBatchPrompts,
    "EndlessReplicateLatents": EndlessNode_ReplicateLatents,
# IGNORE ME, I AM NOT READY!!
    # "LatentReplicator": EndlessNode_FluxLatentReplicator,
    # "LatentReplicatorPrompts": EndlessNode_FluxLatentReplicatorFromPrompts,
}

# Display names for ComfyUI interface
NODE_DISPLAY_NAME_MAPPINGS = {
    "SimpleBatchPrompts": "Simple Batch Prompts",
    "FluxBatchPrompts": "FLUX Batch Prompts",
    "SDXLBatchPrompts": "SDXL Batch Prompts", 
    "BatchNegativePrompts": "Batch Negative Prompts",
    "PromptCounter": "Prompt Counter",
    "FluxKontextBatchPrompts": "FLUX Kontext Batch Prompts",
    "EndlessReplicateLatents": "Replicate Latents",
    # IGNORE ME, I AM NOT READY!!
    # "LatentReplicator": "Latent Replicator",
    # "LatentReplicatorPrompts": "Latent Replicator from Prompts",
}
