47 lines
2.3 KiB
Python
47 lines
2.3 KiB
Python
"""ComfyUI-AI2Go-Utils — a ComfyUI custom node pack by AIKnowledge2Go.
|
|
|
|
Node classes are registered in NODE_CLASS_MAPPINGS; their friendly labels in
|
|
NODE_DISPLAY_NAME_MAPPINGS. Front-end (JS) extensions live under ``web/`` and are
|
|
served to the ComfyUI frontend via WEB_DIRECTORY.
|
|
"""
|
|
|
|
from .nodes.ideogram4_nodes import AI2GoIdeogram4PromptBuilder
|
|
from .nodes.ideogram4_style_wizard import AI2GoIdeogram4StyleWizard
|
|
from .nodes.resolution_selector import AI2GoResolutionSelector
|
|
from .nodes.prompt_batch import AI2GoPromptBatch
|
|
from .nodes.save_civitai_metadata import AI2GoSaveCivitaiMetadata, AI2GoSaveCivitaiMetadataAdvanced
|
|
from .nodes.batch_image_loader import AI2GoBatchImageLoader, AI2GoBatchImageLoaderAdvanced
|
|
from .nodes.batch_audio_loader import AI2GoBatchAudioLoader, AI2GoBatchAudioLoaderAdvanced
|
|
|
|
# Key MUST match each node's schema node_id.
|
|
NODE_CLASS_MAPPINGS = {
|
|
"AI2GoIdeogram4PromptBuilder": AI2GoIdeogram4PromptBuilder,
|
|
"AI2GoIdeogram4StyleWizard": AI2GoIdeogram4StyleWizard,
|
|
"AI2GoResolutionSelector": AI2GoResolutionSelector,
|
|
"AI2GoPromptBatch": AI2GoPromptBatch,
|
|
"AI2GoSaveCivitaiMetadata": AI2GoSaveCivitaiMetadata,
|
|
"AI2GoSaveCivitaiMetadataAdvanced": AI2GoSaveCivitaiMetadataAdvanced,
|
|
"AI2GoBatchImageLoader": AI2GoBatchImageLoader,
|
|
"AI2GoBatchImageLoaderAdvanced": AI2GoBatchImageLoaderAdvanced,
|
|
"AI2GoBatchAudioLoader": AI2GoBatchAudioLoader,
|
|
"AI2GoBatchAudioLoaderAdvanced": AI2GoBatchAudioLoaderAdvanced,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"AI2GoIdeogram4PromptBuilder": "AI2Go Ideogram 4 Prompt Builder",
|
|
"AI2GoIdeogram4StyleWizard": "AI2Go Ideogram 4 Style Wizard",
|
|
"AI2GoResolutionSelector": "AI2Go Resolution Selector",
|
|
"AI2GoPromptBatch": "AI2Go Prompt Batch",
|
|
"AI2GoSaveCivitaiMetadata": "AI2Go Save Metadata (Civitai)",
|
|
"AI2GoSaveCivitaiMetadataAdvanced": "AI2Go Save Metadata (Civitai) Advanced",
|
|
"AI2GoBatchImageLoader": "AI2Go Batch Image Loader",
|
|
"AI2GoBatchImageLoaderAdvanced": "AI2Go Batch Image Loader Advanced",
|
|
"AI2GoBatchAudioLoader": "AI2Go Batch Audio Loader",
|
|
"AI2GoBatchAudioLoaderAdvanced": "AI2Go Batch Audio Loader Advanced",
|
|
}
|
|
|
|
# Folder of front-end JavaScript served to the ComfyUI client.
|
|
WEB_DIRECTORY = "./web"
|
|
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS", "WEB_DIRECTORY"]
|