From 3e87c02b1b9e767fafb6c5e0789e4e64c72be1fc Mon Sep 17 00:00:00 2001 From: DreamFly Date: Sun, 13 Apr 2025 01:31:19 +0800 Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E6=8F=90=E7=A4=BA=E8=AF=8D?= =?UTF-8?q?=E9=A2=84=E8=A7=88?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- __init__.py | 3 +++ nodes/prompt_preview.py | 29 +++++++++++++++++++++++++++++ 2 files changed, 32 insertions(+) create mode 100644 nodes/prompt_preview.py 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,)} + +