37 lines
867 B
Python
37 lines
867 B
Python
from sys import path
|
|
from os.path import dirname, exists, join
|
|
from subprocess import run
|
|
from os import mkdir
|
|
from importlib.util import find_spec
|
|
from platform import system
|
|
|
|
os_name = system()
|
|
|
|
path.append(dirname(__file__))
|
|
|
|
from prompt_generator import PromptGenerator
|
|
|
|
print("/_\ Loading Prompt Generator")
|
|
|
|
# Check prompt_generators folder under models folder
|
|
|
|
root = join("models", "prompt_generators")
|
|
if exists(root) is False:
|
|
print(f"/_\ {root} is created. Please add your prompt generators to {root} folder")
|
|
mkdir(root)
|
|
|
|
if exists("generated_prompts") is False:
|
|
mkdir("generated_prompts")
|
|
|
|
# Import PromptGenerator node to ComfyUI
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"Prompt Generator": PromptGenerator,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"Prompt Generator": "Prompt Generator",
|
|
}
|
|
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"]
|