32 lines
1.3 KiB
Python
32 lines
1.3 KiB
Python
from .image_generation import ChatGPTImageGenerationNode
|
|
from .image_generation import ChatGPTImageModelGenerationNode
|
|
from .image_generation import ChatGPTImageEditNode
|
|
from .text_generation import ChatGPTTextGenerationNode
|
|
|
|
from .banana_generation import BananaImageGenerationNode
|
|
from .banana_generation import BananaImageEditNode
|
|
|
|
from .utils import ImageToBase64
|
|
from . import workflow_chat as _workflow_chat
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"ChatGPTImageGenerationNode": ChatGPTImageGenerationNode,
|
|
"ChatGPTImageModelGenerationNode": ChatGPTImageModelGenerationNode,
|
|
"ChatGPTImageEditNode": ChatGPTImageEditNode,
|
|
"ChatGPTTextGenerationNode": ChatGPTTextGenerationNode,
|
|
"BananaImageGenerationNode": BananaImageGenerationNode,
|
|
"BananaImageEditNode": BananaImageEditNode,
|
|
"ImageToBase64": ImageToBase64
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"ChatGPTImageGenerationNode": "Chat GPT Image Generation Node",
|
|
"ChatGPTImageModelGenerationNode": "Chat GPT Image Model Generation Node",
|
|
"ChatGPTImageEditNode": "Chat GPT Image Edit Node",
|
|
"ChatGPTTextGenerationNode": "Chat GPT Text Generation Node",
|
|
"BananaImageGenerationNode": "Banana Image Generation Node",
|
|
"BananaImageEditNode": "Banana Image Edit Node",
|
|
"ImageToBase64": "Convert Image to Base64"
|
|
}
|
|
|
|
WEB_DIRECTORY = "./web" |