diff --git a/PromptEmbedMango.py b/PromptEmbedMango.py new file mode 100644 index 0000000..4de87b2 --- /dev/null +++ b/PromptEmbedMango.py @@ -0,0 +1,57 @@ +class PromptEmbedMango: + + @classmethod + def INPUT_TYPES(cls): + return { + "required": { + "QualityTags": ( + "STRING", + { + "default": "", + "multiline": True, + "rows": 3 + } + ), + "SceneDescription": ( + "STRING", + { + "default": "", + "multiline": True, + "rows": 3 + } + ), + "Tags": ( + "STRING", + { + "default": "", + "multiline": True, + "rows": 3 + } + ), + "Embeddings": ( + "STRING", + { + "default": "", + "multiline": True, + "rows": 3 + } + ), + }, + "optional": { + "string": ( + "STRING", + { + "default": "" + } + ), + }, + } + + RETURN_TYPES = ("STRING",) + FUNCTION = "execute" + CATEGORY = "Mango Node Pack/Metadata" + + def execute(self, QualityTags, SceneDescription, Tags, Embeddings, string=""): + parts = [s.strip() for s in (QualityTags, SceneDescription, Tags, Embeddings, string) if s.strip()] + concatenated = ", ".join(parts) + return (concatenated,) diff --git a/__init__.py b/__init__.py index bd43204..349a8a0 100644 --- a/__init__.py +++ b/__init__.py @@ -4,6 +4,7 @@ from .MangoLoader import MangoLoader from .ImageSaverMango import ImageSaverMango from .LatentImageMango import LatentImageMango from .PromptMango import PromptMango +from .PromptEmbedMango import PromptEmbedMango NODE_CLASS_MAPPINGS = { "MangoTriggerExporter": MangoTriggerExporter, @@ -11,7 +12,8 @@ NODE_CLASS_MAPPINGS = { "MangoLoader": MangoLoader, "ImageSaverMango": ImageSaverMango, "LatentImageMango": LatentImageMango, - "PromptMango": PromptMango + "PromptMango": PromptMango, + "PromptEmbedMango": PromptEmbedMango, } NODE_DISPLAY_NAME_MAPPINGS = { @@ -20,5 +22,6 @@ NODE_DISPLAY_NAME_MAPPINGS = { "MangoLoader": "Loader (Mango)", "ImageSaverMango": "Image Saver (Mango)", "LatentImageMango": "Latent Image (Mango)", - "PromptMango": "Prompt (Mango)" + "PromptMango": "Prompt (Mango)", + "PromptEmbedMango": "Prompt /w Embedding (Mango)", }