添加提示词预览

__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'] 

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,)}
+    
+