Batch already means one-item-per-queued-run in this pack (AI2GoPromptBatch); these loaders emit everything in a single run, so Batch was misleading. Renamed the four loader nodes/files/classes to Multi (AI2GoMultiImageLoader, AI2GoMultiImageLoaderAdvanced, AI2GoMultiAudioLoader, AI2GoMultiAudioLoaderAdvanced), and reworded DOWNSCALE_MODES from fit/crop/stretch to the plain-word "keep aspect ratio"/"crop to square"/"stretch to square" (behavior unchanged). Nothing published yet, so no back-compat aliases were added. Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>
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.multi_image_loader import AI2GoMultiImageLoader, AI2GoMultiImageLoaderAdvanced
|
|
from .nodes.multi_audio_loader import AI2GoMultiAudioLoader, AI2GoMultiAudioLoaderAdvanced
|
|
|
|
# Key MUST match each node's schema node_id.
|
|
NODE_CLASS_MAPPINGS = {
|
|
"AI2GoIdeogram4PromptBuilder": AI2GoIdeogram4PromptBuilder,
|
|
"AI2GoIdeogram4StyleWizard": AI2GoIdeogram4StyleWizard,
|
|
"AI2GoResolutionSelector": AI2GoResolutionSelector,
|
|
"AI2GoPromptBatch": AI2GoPromptBatch,
|
|
"AI2GoSaveCivitaiMetadata": AI2GoSaveCivitaiMetadata,
|
|
"AI2GoSaveCivitaiMetadataAdvanced": AI2GoSaveCivitaiMetadataAdvanced,
|
|
"AI2GoMultiImageLoader": AI2GoMultiImageLoader,
|
|
"AI2GoMultiImageLoaderAdvanced": AI2GoMultiImageLoaderAdvanced,
|
|
"AI2GoMultiAudioLoader": AI2GoMultiAudioLoader,
|
|
"AI2GoMultiAudioLoaderAdvanced": AI2GoMultiAudioLoaderAdvanced,
|
|
}
|
|
|
|
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",
|
|
"AI2GoMultiImageLoader": "AI2Go Multi Image Loader",
|
|
"AI2GoMultiImageLoaderAdvanced": "AI2Go Multi Image Loader Advanced",
|
|
"AI2GoMultiAudioLoader": "AI2Go Multi Audio Loader",
|
|
"AI2GoMultiAudioLoaderAdvanced": "AI2Go Multi 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"]
|