Alexandr Radowskiy
6c2f56b1c3
- Implements the V3 (Modern Node Design) schema via io.ComfyNode. - Falls back to V1 (NODE_CLASS_MAPPINGS) for older ComfyUI installs. - Calls Ideogram's hosted magic-prompt API via urllib (no extra deps). - Strips bbox / aspect_ratio echoes so the caption is a clean drop-in for Ideogram4Pipeline.
28 lines
812 B
Python
28 lines
812 B
Python
"""Ideogram Magic Prompt -- ComfyUI custom node package.
|
|
|
|
Drop this entire folder into ComfyUI/custom_nodes/ and restart ComfyUI.
|
|
The node "Ideogram Magic Prompt" will appear under the Ideogram category.
|
|
"""
|
|
|
|
from .nodes import IdeogramExtension, IdeogramMagicPrompt
|
|
|
|
# V1-compatible fallback for older ComfyUI installs (V1 schema) so the
|
|
# node still loads even if comfy_api.latest isn't available. ComfyUI's
|
|
# loader expects NODE_CLASS_MAPPINGS / NODE_DISPLAY_NAME_MAPPINGS at
|
|
# module import time.
|
|
NODE_CLASS_MAPPINGS = {
|
|
"IdeogramMagicPrompt": IdeogramMagicPrompt,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"IdeogramMagicPrompt": "Ideogram Magic Prompt",
|
|
}
|
|
|
|
# Web-exposed metadata
|
|
__all__ = [
|
|
"IdeogramMagicPrompt",
|
|
"IdeogramExtension",
|
|
"NODE_CLASS_MAPPINGS",
|
|
"NODE_DISPLAY_NAME_MAPPINGS",
|
|
]
|