From 8753343c4b7ce409a88709e2478e93f32c246ed1 Mon Sep 17 00:00:00 2001
From: DreamFly <dreamfly6789@gmail.com>
Date: Thu, 10 Apr 2025 01:03:29 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 nodes/prompt_generate.py |  6 +++---
 nodes/prompt_save.py     | 12 +++++-------
 2 files changed, 8 insertions(+), 10 deletions(-)

diff --git a/nodes/prompt_generate.py b/nodes/prompt_generate.py
index a2a00b1..9a9b93d 100644
--- a/nodes/prompt_generate.py
+++ b/nodes/prompt_generate.py
@@ -39,12 +39,12 @@ class GeneratePrompt:
     FUNCTION = "prompt_generate"
     CATEGORY = "PromptWrapper"
 
-    def prompt_generate(self, language, classify, seed, preset_prefix="", preset_suffix="", ainput_prompt=""):
+    def prompt_generate(self, language, classify, seed, preset_prefix="", preset_suffix="", input_prompt=""):
  
         prompt_words = []
 
-        if ainput_prompt != "":
-            prompt_words.append(ainput_prompt)
+        if input_prompt != "":
+            prompt_words.append(input_prompt)
 
         if classify != EMPTY_OPTION:
             # 获取Script的目录路径
diff --git a/nodes/prompt_save.py b/nodes/prompt_save.py
index 366c0d8..3f4cc48 100644
--- a/nodes/prompt_save.py
+++ b/nodes/prompt_save.py
@@ -9,9 +9,7 @@ class SavePrompt:
     def INPUT_TYPES(s):
         return {
             "required": {
-                "text": ("STRING", {
-                    "forceInput": True
-                }),
+                "prompt": ("STRING", {"forceInput": True}),
                 "filename": ("STRING", {
                     "default": "custom"
                 })
@@ -25,9 +23,10 @@ class SavePrompt:
     CATEGORY = "PromptWrapper"
 
 
-    def save(self, text="", filename=""):
+    def save(self, prompt="", filename=""):
  
-        text = text.strip()
+        text = prompt.strip()
+
         if text != "" and text != "\n" and text != "\r\n":
             # 获取Script的目录路径
             script_dir = os.path.dirname(os.path.abspath(__file__)) 
@@ -47,6 +46,5 @@ class SavePrompt:
                     fp.write(f"\n{text_data}")
 
         
-        return {"ui": {"text": text}}
-
+        return {"ui": {"text": text}, "result": ()}