77 lines
3.2 KiB
Python
77 lines
3.2 KiB
Python
import os
|
|
|
|
import folder_paths
|
|
|
|
FLAKES_DIR = os.path.join(folder_paths.models_dir, "flakes")
|
|
os.makedirs(FLAKES_DIR, exist_ok=True)
|
|
|
|
if "flakes" in folder_paths.folder_names_and_paths:
|
|
existing_paths, existing_exts = folder_paths.folder_names_and_paths["flakes"]
|
|
if FLAKES_DIR not in existing_paths:
|
|
existing_paths.append(FLAKES_DIR)
|
|
existing_exts.update({".yaml", ".yml"})
|
|
else:
|
|
folder_paths.folder_names_and_paths["flakes"] = ([FLAKES_DIR], {".yaml", ".yml"})
|
|
|
|
PRESETS_DIR = os.path.join(FLAKES_DIR, "model_presets")
|
|
os.makedirs(PRESETS_DIR, exist_ok=True)
|
|
|
|
if "model_presets" in folder_paths.folder_names_and_paths:
|
|
existing_paths, existing_exts = folder_paths.folder_names_and_paths["model_presets"]
|
|
if PRESETS_DIR not in existing_paths:
|
|
existing_paths.append(PRESETS_DIR)
|
|
existing_exts.update({".yaml", ".yml"})
|
|
else:
|
|
folder_paths.folder_names_and_paths["model_presets"] = ([PRESETS_DIR], {".yaml", ".yml"})
|
|
|
|
from .full_flake_node import FlakeModelPreset, FlakeStack, FlakeCombo, FlakeModelCombo # noqa: E402
|
|
from .flake_data_nodes import FlakeDataSplitAll, FlakeDataSplitSelect, IntoFlakeDataAll, IntoFlakeDataSelect, PreviewFlakeData, FlakeGenerate # noqa: E402
|
|
from .flake_extra_nodes import FlakeIPAdapter, FlakeInpaint # noqa: E402
|
|
from .flake_anim_nodes import FlakeAnimPreset, FlakeAnimPresetCombo, FlakeAnimStack, FlakeAnimCombo, FlakeAnimCaption, FlakeGenerateAnimation # noqa: E402
|
|
from . import flake_server # noqa: E402,F401 — registers aiohttp routes on import
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"FlakeModelPreset": FlakeModelPreset,
|
|
"FlakeStack": FlakeStack,
|
|
"FlakeCombo": FlakeCombo,
|
|
"FlakeModelCombo": FlakeModelCombo,
|
|
"FlakeDataSplitAll": FlakeDataSplitAll,
|
|
"FlakeDataSplitSelect": FlakeDataSplitSelect,
|
|
"IntoFlakeDataAll": IntoFlakeDataAll,
|
|
"IntoFlakeDataSelect": IntoFlakeDataSelect,
|
|
"PreviewFlakeData": PreviewFlakeData,
|
|
"FlakeGenerate": FlakeGenerate,
|
|
"FlakeIPAdapter": FlakeIPAdapter,
|
|
"FlakeInpaint": FlakeInpaint,
|
|
"FlakeAnimPreset": FlakeAnimPreset,
|
|
"FlakeAnimPresetCombo": FlakeAnimPresetCombo,
|
|
"FlakeAnimStack": FlakeAnimStack,
|
|
"FlakeAnimCombo": FlakeAnimCombo,
|
|
"FlakeAnimCaption": FlakeAnimCaption,
|
|
"FlakeGenerateAnimation": FlakeGenerateAnimation,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"FlakeModelPreset": "Flake Model Preset",
|
|
"FlakeStack": "Flake Stack",
|
|
"FlakeCombo": "Flake Combo",
|
|
"FlakeModelCombo": "Flake Model Combo",
|
|
"FlakeDataSplitAll": "Flake Data Split (All)",
|
|
"FlakeDataSplitSelect": "Flake Data Split (Select)",
|
|
"IntoFlakeDataAll": "Into Flake Data (All)",
|
|
"IntoFlakeDataSelect": "Into Flake Data (Select)",
|
|
"PreviewFlakeData": "Preview Flake Data",
|
|
"FlakeGenerate": "Flake Generate",
|
|
"FlakeIPAdapter": "Flake IPAdapter",
|
|
"FlakeInpaint": "Flake Inpaint",
|
|
"FlakeAnimPreset": "Flake Animate Preset",
|
|
"FlakeAnimPresetCombo": "Flake Animate Preset Combo",
|
|
"FlakeAnimStack": "Flake Animate Stack",
|
|
"FlakeAnimCombo": "Flake Animate Combo",
|
|
"FlakeAnimCaption": "Flake Animate Caption",
|
|
"FlakeGenerateAnimation": "Flake Generate Animation",
|
|
}
|
|
WEB_DIRECTORY = "./web"
|
|
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS", "WEB_DIRECTORY"]
|