51 lines
1.8 KiB
Python
51 lines
1.8 KiB
Python
import os
|
|
|
|
from .py.auto_exposure.nodes import MeldAutoExposure
|
|
from .py.load_image_configs import (
|
|
MeldImageLoader,
|
|
MeldImageLoaderBatch,
|
|
MeldSettingsUnpacker,
|
|
MeldUnifiedFluxLoader,
|
|
MeldUnifiedLoader,
|
|
)
|
|
from .py.meld_save_image.nodes import MeldSaveImage
|
|
from .py.pattern_heart.nodes import MeldPatternHeart
|
|
from .py.pixelate.nodes import MeldPixelate
|
|
from .py.prompt_constructor.nodes import MeldPromptConstructor
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"MeldAutoExposure": MeldAutoExposure,
|
|
"MeldPixelate": MeldPixelate,
|
|
"MeldImageLoader": MeldImageLoader,
|
|
"MeldImageLoaderBatch": MeldImageLoaderBatch,
|
|
"MeldSettingsUnpacker": MeldSettingsUnpacker,
|
|
"MeldUnifiedFluxLoader": MeldUnifiedFluxLoader,
|
|
"MeldUnifiedLoader": MeldUnifiedLoader,
|
|
"MeldPromptConstructor": MeldPromptConstructor,
|
|
"MeldPatternHeart": MeldPatternHeart,
|
|
"MeldSaveImage": MeldSaveImage,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"MeldAutoExposure": "Meld Auto Exposure",
|
|
"MeldPixelate": "Meld Instant Pixelate",
|
|
"MeldImageLoader": "Meld Image Loader",
|
|
"MeldImageLoaderBatch": "Meld Image Loader (Batch)",
|
|
"MeldSettingsUnpacker": "Meld Settings Unpacker",
|
|
"MeldUnifiedFluxLoader": "Meld Unified Flux Loader",
|
|
"MeldUnifiedLoader": "Meld Unified Loader",
|
|
"MeldPromptConstructor": "Meld Prompt Constructor",
|
|
"MeldPatternHeart": "Meld Infinite Heart Generator",
|
|
"MeldSaveImage": "Meld Save Image",
|
|
}
|
|
|
|
# Conditional registration of Meld
|
|
if os.environ.get("MELD_DEV") == "true":
|
|
NODE_CLASS_MAPPINGS["MeldSaveImage"] = MeldSaveImage
|
|
NODE_DISPLAY_NAME_MAPPINGS["MeldSaveImage"] = "Meld Save Image"
|
|
print("\033[34m[Meld] Meld enabled (Dev Mode)\033[0m")
|
|
|
|
WEB_DIRECTORY = "web"
|
|
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS", "WEB_DIRECTORY"]
|