diff --git a/__init__.py b/__init__.py index 40ea191..33b53bf 100644 --- a/__init__.py +++ b/__init__.py @@ -13,6 +13,7 @@ from .nodes.prompt_combine import CombinePrompt, MultiCombinePrompt from .nodes.prompt_replace import ReplacePrompt, MultiReplacePrompt from .nodes.prompt_custom import CustomPrompt from .nodes.prompt_save import SavePrompt +from .nodes.prompt_preview import PreviewPrompt @@ -34,6 +35,7 @@ NODE_CLASS_MAPPINGS = { "MultiReplacePrompt": MultiReplacePrompt, "CustomPrompt": CustomPrompt, "SavePrompt": SavePrompt, + "PreviewPrompt": PreviewPrompt, } NODE_DISPLAY_NAME_MAPPINGS = { @@ -54,6 +56,7 @@ NODE_DISPLAY_NAME_MAPPINGS = { "MultiReplacePrompt": "Multi Replace Prompt", "CustomPrompt": "Custom Prompt", "SavePrompt": "Save Prompt", + "PreviewPrompt", "Preview Prompt", } __all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS'] \ No newline at end of file diff --git a/nodes/prompt_preview.py b/nodes/prompt_preview.py new file mode 100644 index 0000000..a873c2e --- /dev/null +++ b/nodes/prompt_preview.py @@ -0,0 +1,29 @@ +import os + + +# Preview Prompt +class PreviewPrompt: + + @classmethod + def INPUT_TYPES(s): + return { + "required": { + "text": ("STRING", {"forceInput": True}), + }, + "hidden": { + "unique_id": "UNIQUE_ID" + }, + } + + + RETURN_TYPES = ("STRING",) + RETURN_NAMES = ("text",) + FUNCTION = "preview" + OUTPUT_NODE = True + CATEGORY = "PromptWrapper" + + def preview(self, text, unique_id=None): + + return {"ui": {"text": text}, "result": (text,)} + +