From 4ff370706234b6d627db8c766435f5fe28a633a2 Mon Sep 17 00:00:00 2001
From: CNlaojing <154053522+CNlaojing@users.noreply.github.com>
Date: Thu, 13 Mar 2025 07:15:59 +0800
Subject: [PATCH] Update generation_handlers.py

---
 ui/generation_handlers.py | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/ui/generation_handlers.py b/ui/generation_handlers.py
index 099a8d4..d1fbadb 100644
--- a/ui/generation_handlers.py
+++ b/ui/generation_handlers.py
@@ -25,13 +25,11 @@ def generate_novel_architecture_ui(self):
         return
 
     def task():
-
         confirm = messagebox.askyesno("确认", "确定要生成小说架构吗？")
         if not confirm:
             self.enable_button_safe(self.btn_generate_architecture)
             return
 
-
         self.disable_button_safe(self.btn_generate_architecture)
         try:
             interface_format = self.interface_format_var.get().strip()
@@ -46,6 +44,8 @@ def generate_novel_architecture_ui(self):
             genre = self.genre_var.get().strip()
             num_chapters = self.safe_get_int(self.num_chapters_var, 10)
             word_number = self.safe_get_int(self.word_number_var, 3000)
+            # 获取内容指导
+            user_guidance = self.user_guide_text.get("0.0", "end").strip()
 
             self.safe_log("开始生成小说架构...")
             Novel_architecture_generate(
@@ -60,7 +60,8 @@ def generate_novel_architecture_ui(self):
                 filepath=filepath,
                 temperature=temperature,
                 max_tokens=max_tokens,
-                timeout=timeout_val
+                timeout=timeout_val,
+                user_guidance=user_guidance  # 添加内容指导参数
             )
             self.safe_log("✅ 小说架构生成完成。请在 'Novel Architecture' 标签页查看或编辑。")
         except Exception: