Fork of KJNodes' Ideogram 4 Prompt Builder with stores, multi-select, scene save/load, live background preview, and separate width/height outputs. Co-authored-by: Cursor <cursoragent@cursor.com>
31 lines
1006 B
Python
31 lines
1006 B
Python
from .nodes.ideogram4_nodes import (
|
|
Ideogram4PromptBuilderKJEnhancedByAE,
|
|
Ideogram4PromptBuilderKJPreviewUpdateByAE,
|
|
)
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"Ideogram4PromptBuilderKJEnhancedByAE": Ideogram4PromptBuilderKJEnhancedByAE,
|
|
"Ideogram4PromptBuilderKJPreviewUpdateByAE": Ideogram4PromptBuilderKJPreviewUpdateByAE,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"Ideogram4PromptBuilderKJEnhancedByAE": "Ideogram 4 Prompt Builder Pro (based on the KJ node)",
|
|
"Ideogram4PromptBuilderKJPreviewUpdateByAE": "Ideogram 4 Prompt Builder Pro - Preview BG Update",
|
|
}
|
|
|
|
WEB_DIRECTORY = "./web"
|
|
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS", "WEB_DIRECTORY"]
|
|
|
|
import logging
|
|
|
|
from server import PromptServer
|
|
|
|
from .routes import register_routes
|
|
|
|
if hasattr(PromptServer, "instance") and PromptServer.instance is not None:
|
|
try:
|
|
register_routes(PromptServer.instance.app)
|
|
except Exception:
|
|
logging.exception("ComfyUI-ByAE-IdeogramTools: failed to register store routes")
|