# ui.py # -*- coding: utf-8 -*- import logging import os import threading import customtkinter as ctk from tkinter import filedialog, messagebox import traceback from config_manager import load_config, save_config from utils import read_file, save_string_to_txt, clear_file_content from novel_generator import ( Novel_architecture_generate, Chapter_blueprint_generate, generate_chapter_draft, finalize_chapter, import_knowledge_file, clear_vector_store, get_last_n_chapters_text ) from consistency_checker import check_consistency # ---- Import the tooltip texts ---- from tooltips import tooltips def log_error(message: str): logging.error(f"{message}\n{traceback.format_exc()}") ctk.set_appearance_mode("System") ctk.set_default_color_theme("blue") class NovelGeneratorGUI: def __init__(self, master): self.master = master self.master.title("Novel Generator GUI") try: if os.path.exists("icon.ico"): self.master.iconbitmap("icon.ico") except Exception: pass self.master.geometry("1350x840") # 配置持久化 self.config_file = "config.json" self.loaded_config = load_config(self.config_file) # 主要属性变量 self.api_key_var = ctk.StringVar(value=self.loaded_config.get("api_key", "")) self.base_url_var = ctk.StringVar(value=self.loaded_config.get("base_url", "https://api.openai.com/v1")) self.interface_format_var = ctk.StringVar(value=self.loaded_config.get("interface_format", "OpenAI")) self.model_name_var = ctk.StringVar(value=self.loaded_config.get("model_name", "gpt-4o-mini")) self.temperature_var = ctk.DoubleVar(value=self.loaded_config.get("temperature", 0.7)) self.max_tokens_var = ctk.IntVar(value=self.loaded_config.get("max_tokens", 8192)) # Embedding相关 self.embedding_interface_format_var = ctk.StringVar(value=self.loaded_config.get("embedding_interface_format", "OpenAI")) self.embedding_api_key_var = ctk.StringVar(value=self.loaded_config.get("embedding_api_key", "")) self.embedding_url_var = ctk.StringVar(value=self.loaded_config.get("embedding_url", "https://api.openai.com/v1")) self.embedding_model_name_var = ctk.StringVar(value=self.loaded_config.get("embedding_model_name", "text-embedding-ada-002")) self.embedding_retrieval_k_var = ctk.StringVar(value=str(self.loaded_config.get("embedding_retrieval_k", 4))) self.topic_default = self.loaded_config.get("topic", "") self.genre_var = ctk.StringVar(value=self.loaded_config.get("genre", "玄幻")) self.num_chapters_var = ctk.StringVar(value=str(self.loaded_config.get("num_chapters", 10))) self.word_number_var = ctk.StringVar(value=str(self.loaded_config.get("word_number", 3000))) self.filepath_var = ctk.StringVar(value=self.loaded_config.get("filepath", "")) self.chapter_num_var = ctk.StringVar(value="1") # 四个可选要素 self.characters_involved_var = ctk.StringVar(value="") self.key_items_var = ctk.StringVar(value="") self.scene_location_var = ctk.StringVar(value="") self.time_constraint_var = ctk.StringVar(value="") # UI 布局 self.tabview = ctk.CTkTabview(self.master) self.tabview.pack(fill="both", expand=True) self.main_tab = self.tabview.add("Main Functions") self.setting_tab = self.tabview.add("Novel Architecture") self.directory_tab = self.tabview.add("Chapter Blueprint") self.character_tab = self.tabview.add("Character State") self.summary_tab = self.tabview.add("Global Summary") self.chapters_view_tab = self.tabview.add("Chapters Manage") self.build_main_tab() self.build_setting_tab() self.build_directory_tab() self.build_character_tab() self.build_summary_tab() self.build_chapters_tab() def show_tooltip(self, key: str): """Display a popup with tooltip text.""" info_text = tooltips.get(key, "暂无说明") messagebox.showinfo("参数说明", info_text) def safe_get_int(self, var, default=1): try: val_str = str(var.get()).strip() return int(val_str) except: var.set(str(default)) return default # ------------------ 主 Tab ------------------ def build_main_tab(self): self.main_tab.rowconfigure(0, weight=1) self.main_tab.columnconfigure(0, weight=1) self.main_tab.columnconfigure(1, weight=0) self.left_frame = ctk.CTkFrame(self.main_tab) self.left_frame.grid(row=0, column=0, sticky="nsew", padx=2, pady=2) self.right_frame = ctk.CTkFrame(self.main_tab) self.right_frame.grid(row=0, column=1, sticky="nsew", padx=2, pady=2) self.build_left_layout() self.build_right_layout() def build_left_layout(self): self.left_frame.grid_rowconfigure(0, weight=0) self.left_frame.grid_rowconfigure(1, weight=2) self.left_frame.grid_rowconfigure(2, weight=0) self.left_frame.grid_rowconfigure(3, weight=0) self.left_frame.grid_rowconfigure(4, weight=1) self.left_frame.columnconfigure(0, weight=1) chapter_label = ctk.CTkLabel(self.left_frame, text="本章内容 (可编辑)", font=("Microsoft YaHei", 12)) chapter_label.grid(row=0, column=0, padx=5, pady=(5, 0), sticky="w") self.chapter_result = ctk.CTkTextbox(self.left_frame, wrap="word", font=("Microsoft YaHei", 14)) self.chapter_result.grid(row=1, column=0, sticky="nsew", padx=5, pady=(0, 5)) # Step 按钮区域 self.step_buttons_frame = ctk.CTkFrame(self.left_frame) self.step_buttons_frame.grid(row=2, column=0, sticky="ew", padx=5, pady=5) self.step_buttons_frame.columnconfigure((0, 1, 2, 3), weight=1) self.btn_generate_architecture = ctk.CTkButton( self.step_buttons_frame, text="Step1. 生成架构", command=self.generate_novel_architecture_ui, font=("Microsoft YaHei", 12) ) self.btn_generate_architecture.grid(row=0, column=0, padx=5, pady=2, sticky="ew") self.btn_generate_directory = ctk.CTkButton( self.step_buttons_frame, text="Step2. 生成目录", command=self.generate_chapter_blueprint_ui, font=("Microsoft YaHei", 12) ) self.btn_generate_directory.grid(row=0, column=1, padx=5, pady=2, sticky="ew") self.btn_generate_chapter = ctk.CTkButton( self.step_buttons_frame, text="Step3. 生成草稿", command=self.generate_chapter_draft_ui, font=("Microsoft YaHei", 12) ) self.btn_generate_chapter.grid(row=0, column=2, padx=5, pady=2, sticky="ew") self.btn_finalize_chapter = ctk.CTkButton( self.step_buttons_frame, text="Step4. 定稿章节", command=self.finalize_chapter_ui, font=("Microsoft YaHei", 12) ) self.btn_finalize_chapter.grid(row=0, column=3, padx=5, pady=2, sticky="ew") # 日志 log_label = ctk.CTkLabel(self.left_frame, text="输出日志 (只读)", font=("Microsoft YaHei", 12)) log_label.grid(row=3, column=0, padx=5, pady=(5, 0), sticky="w") self.log_text = ctk.CTkTextbox(self.left_frame, wrap="word", font=("Microsoft YaHei", 12)) self.log_text.grid(row=4, column=0, sticky="nsew", padx=5, pady=(0, 5)) self.log_text.configure(state="disabled") def build_right_layout(self): self.right_frame.grid_rowconfigure(0, weight=0) self.right_frame.grid_rowconfigure(1, weight=1) self.right_frame.grid_rowconfigure(2, weight=0) self.right_frame.columnconfigure(0, weight=1) # 配置区 self.config_frame = ctk.CTkFrame(self.right_frame, corner_radius=10, border_width=2, border_color="gray") self.config_frame.grid(row=0, column=0, sticky="ew", padx=5, pady=5) self.config_frame.columnconfigure(0, weight=1) self.build_config_tabview() self.build_main_buttons_area() # 小说参数 self.build_novel_params_area(start_row=1) # 可选功能按钮 self.build_optional_buttons_area(start_row=2) def build_config_tabview(self): self.config_tabview = ctk.CTkTabview(self.config_frame) self.config_tabview.grid(row=0, column=0, sticky="we", padx=5, pady=5) self.ai_config_tab = self.config_tabview.add("LLM Model settings") self.embeddings_config_tab = self.config_tabview.add("Embedding settings") self.build_ai_config_tab() self.build_embeddings_config_tab() # 封装一个小工具函数，用来创建「标签 + 问号按钮」的组合 def create_label_with_help(self, parent, label_text, tooltip_key, row, column, font=None, sticky="e", padx=5, pady=5): # frame容器：同一格子里存放 label + "?"按钮 frame = ctk.CTkFrame(parent) frame.grid(row=row, column=column, padx=padx, pady=pady, sticky=sticky) frame.columnconfigure(0, weight=0) # 先放 label label = ctk.CTkLabel(frame, text=label_text, font=font) label.pack(side="left") # 再放问号按钮 btn = ctk.CTkButton( frame, text="?", width=22, height=22, font=("Microsoft YaHei", 10), command=lambda: self.show_tooltip(tooltip_key) ) btn.pack(side="left", padx=3) return frame def build_ai_config_tab(self): def on_interface_format_changed(new_value): if new_value == "Ollama": self.base_url_var.set("http://localhost:11434/v1") elif new_value == "ML Studio": self.base_url_var.set("http://localhost:1234/v1") elif new_value == "OpenAI": self.base_url_var.set("https://api.openai.com/v1") elif new_value == "DeepSeek": self.base_url_var.set("https://api.deepseek.com/v1") for i in range(6): self.ai_config_tab.grid_rowconfigure(i, weight=0) self.ai_config_tab.grid_columnconfigure(0, weight=0) self.ai_config_tab.grid_columnconfigure(1, weight=1) self.ai_config_tab.grid_columnconfigure(2, weight=0) # 1) API Key self.create_label_with_help( parent=self.ai_config_tab, label_text="LLM API Key:", tooltip_key="api_key", row=0, column=0, font=("Microsoft YaHei", 12) ) api_key_entry = ctk.CTkEntry(self.ai_config_tab, textvariable=self.api_key_var, font=("Microsoft YaHei", 12)) api_key_entry.grid(row=0, column=1, padx=5, pady=5, columnspan=2, sticky="nsew") # 2) Base URL self.create_label_with_help( parent=self.ai_config_tab, label_text="LLM Base URL:", tooltip_key="base_url", row=1, column=0, font=("Microsoft YaHei", 12) ) base_url_entry = ctk.CTkEntry(self.ai_config_tab, textvariable=self.base_url_var, font=("Microsoft YaHei", 12)) base_url_entry.grid(row=1, column=1, padx=5, pady=5, columnspan=2, sticky="nsew") # 3) 接口格式 label_frame = self.create_label_with_help( parent=self.ai_config_tab, label_text="LLM 接口格式:", tooltip_key="interface_format", row=2, column=0, font=("Microsoft YaHei", 12) ) interface_options = ["DeepSeek", "OpenAI", "Ollama", "ML Studio"] interface_dropdown = ctk.CTkOptionMenu( self.ai_config_tab, values=interface_options, variable=self.interface_format_var, command=on_interface_format_changed, font=("Microsoft YaHei", 12) ) interface_dropdown.grid(row=2, column=1, padx=5, pady=5, columnspan=2, sticky="nsew") # 4) Model Name self.create_label_with_help( parent=self.ai_config_tab, label_text="Model Name:", tooltip_key="model_name", row=3, column=0, font=("Microsoft YaHei", 12) ) model_name_entry = ctk.CTkEntry(self.ai_config_tab, textvariable=self.model_name_var, font=("Microsoft YaHei", 12)) model_name_entry.grid(row=3, column=1, padx=5, pady=5, columnspan=2, sticky="nsew") # 5) Temperature temp_frame = self.create_label_with_help( parent=self.ai_config_tab, label_text="Temperature:", tooltip_key="temperature", row=4, column=0, font=("Microsoft YaHei", 12) ) def update_temp_label(value): self.temp_value_label.configure(text=f"{float(value):.2f}") temp_scale = ctk.CTkSlider( self.ai_config_tab, from_=0.0, to=2.0, number_of_steps=200, command=update_temp_label, variable=self.temperature_var ) temp_scale.grid(row=4, column=1, padx=5, pady=5, sticky="we") self.temp_value_label = ctk.CTkLabel( self.ai_config_tab, text=f"{self.temperature_var.get():.2f}", font=("Microsoft YaHei", 12) ) self.temp_value_label.grid(row=4, column=2, padx=5, pady=5, sticky="w") # 6) Max Tokens self.create_label_with_help( parent=self.ai_config_tab, label_text="Max Tokens:", tooltip_key="max_tokens", row=5, column=0, font=("Microsoft YaHei", 12) ) def update_max_tokens_label(value): self.max_tokens_value_label.configure(text=str(int(float(value)))) max_tokens_slider = ctk.CTkSlider( self.ai_config_tab, from_=0, to=102400, number_of_steps=100, command=update_max_tokens_label, variable=self.max_tokens_var ) max_tokens_slider.grid(row=5, column=1, padx=5, pady=5, sticky="we") self.max_tokens_value_label = ctk.CTkLabel( self.ai_config_tab, text=str(self.max_tokens_var.get()), font=("Microsoft YaHei", 12) ) self.max_tokens_value_label.grid(row=5, column=2, padx=5, pady=5, sticky="w") def build_embeddings_config_tab(self): def on_embedding_interface_changed(new_value): if new_value == "Ollama": self.embedding_url_var.set("http://localhost:11434/api") elif new_value == "ML Studio": self.embedding_url_var.set("http://localhost:1234/v1") elif new_value == "OpenAI": self.embedding_url_var.set("https://api.openai.com/v1") elif new_value == "DeepSeek": self.embedding_url_var.set("https://api.deepseek.com/v1") for i in range(5): self.embeddings_config_tab.grid_rowconfigure(i, weight=0) self.embeddings_config_tab.grid_columnconfigure(0, weight=0) self.embeddings_config_tab.grid_columnconfigure(1, weight=1) self.embeddings_config_tab.grid_columnconfigure(2, weight=0) # 1) Embedding API Key self.create_label_with_help( parent=self.embeddings_config_tab, label_text="Embedding API Key:", tooltip_key="embedding_api_key", row=0, column=0, font=("Microsoft YaHei", 12) ) emb_api_key_entry = ctk.CTkEntry(self.embeddings_config_tab, textvariable=self.embedding_api_key_var, font=("Microsoft YaHei", 12)) emb_api_key_entry.grid(row=0, column=1, padx=5, pady=5, sticky="nsew") # 2) Embedding 接口格式 self.create_label_with_help( parent=self.embeddings_config_tab, label_text="Embedding 接口格式:", tooltip_key="embedding_interface_format", row=1, column=0, font=("Microsoft YaHei", 12) ) emb_interface_options = ["DeepSeek", "OpenAI", "Ollama", "ML Studio"] emb_interface_dropdown = ctk.CTkOptionMenu( self.embeddings_config_tab, values=emb_interface_options, variable=self.embedding_interface_format_var, command=on_embedding_interface_changed, font=("Microsoft YaHei", 12) ) emb_interface_dropdown.grid(row=1, column=1, padx=5, pady=5, sticky="nsew") # 3) Embedding Base URL self.create_label_with_help( parent=self.embeddings_config_tab, label_text="Embedding Base URL:", tooltip_key="embedding_url", row=2, column=0, font=("Microsoft YaHei", 12) ) emb_url_entry = ctk.CTkEntry(self.embeddings_config_tab, textvariable=self.embedding_url_var, font=("Microsoft YaHei", 12)) emb_url_entry.grid(row=2, column=1, padx=5, pady=5, sticky="nsew") # 4) Embedding Model Name self.create_label_with_help( parent=self.embeddings_config_tab, label_text="Embedding Model Name:", tooltip_key="embedding_model_name", row=3, column=0, font=("Microsoft YaHei", 12) ) emb_model_name_entry = ctk.CTkEntry(self.embeddings_config_tab, textvariable=self.embedding_model_name_var, font=("Microsoft YaHei", 12)) emb_model_name_entry.grid(row=3, column=1, padx=5, pady=5, sticky="nsew") # 5) Retrieval Top-K self.create_label_with_help( parent=self.embeddings_config_tab, label_text="Retrieval Top-K:", tooltip_key="embedding_retrieval_k", row=4, column=0, font=("Microsoft YaHei", 12) ) emb_retrieval_k_entry = ctk.CTkEntry(self.embeddings_config_tab, textvariable=self.embedding_retrieval_k_var, font=("Microsoft YaHei", 12)) emb_retrieval_k_entry.grid(row=4, column=1, padx=5, pady=5, sticky="nsew") def build_main_buttons_area(self): self.btn_frame_config = ctk.CTkFrame(self.config_frame) self.btn_frame_config.grid(row=1, column=0, padx=5, pady=5, sticky="ew") self.btn_frame_config.columnconfigure(0, weight=1) self.btn_frame_config.columnconfigure(1, weight=1) save_config_btn = ctk.CTkButton(self.btn_frame_config, text="保存配置", command=self.save_config_btn, font=("Microsoft YaHei", 12)) save_config_btn.grid(row=0, column=0, padx=5, pady=5, sticky="ew") load_config_btn = ctk.CTkButton(self.btn_frame_config, text="加载配置", command=self.load_config_btn, font=("Microsoft YaHei", 12)) load_config_btn.grid(row=0, column=1, padx=5, pady=5, sticky="ew") def build_novel_params_area(self, start_row=1): self.params_frame = ctk.CTkScrollableFrame(self.right_frame, orientation="vertical") self.params_frame.grid(row=start_row, column=0, sticky="nsew", padx=5, pady=5) self.params_frame.columnconfigure(1, weight=1) # 1) 主题(Topic) topic_label_frame = self.create_label_with_help( parent=self.params_frame, label_text="主题(Topic):", tooltip_key="topic", row=0, column=0, font=("Microsoft YaHei", 12), sticky="ne" ) self.topic_text = ctk.CTkTextbox(self.params_frame, height=80, wrap="word", font=("Microsoft YaHei", 12)) self.topic_text.grid(row=0, column=1, padx=5, pady=5, sticky="nsew") if self.topic_default: self.topic_text.insert("0.0", self.topic_default) # 2) 类型(Genre) self.create_label_with_help( parent=self.params_frame, label_text="类型(Genre):", tooltip_key="genre", row=1, column=0, font=("Microsoft YaHei", 12) ) genre_entry = ctk.CTkEntry(self.params_frame, textvariable=self.genre_var, font=("Microsoft YaHei", 12)) genre_entry.grid(row=1, column=1, padx=5, pady=5, sticky="ew") # 3) 章节数 & 每章字数 row_for_chapter_and_word = 2 chapter_word_frame = ctk.CTkFrame(self.params_frame) chapter_word_frame.grid(row=row_for_chapter_and_word, column=1, padx=5, pady=5, sticky="ew") chapter_word_frame.columnconfigure((0, 1, 2, 3), weight=0) # 左边标签 label_frame = self.create_label_with_help( parent=self.params_frame, label_text="章节数 & 每章字数:", tooltip_key="num_chapters", row=row_for_chapter_and_word, column=0, font=("Microsoft YaHei", 12) ) # 输入框 num_chapters_label = ctk.CTkLabel(chapter_word_frame, text="章节数:", font=("Microsoft YaHei", 12)) num_chapters_label.grid(row=0, column=0, padx=5, pady=5, sticky="e") num_chapters_entry = ctk.CTkEntry(chapter_word_frame, textvariable=self.num_chapters_var, width=60, font=("Microsoft YaHei", 12)) num_chapters_entry.grid(row=0, column=1, padx=5, pady=5, sticky="w") word_number_label = ctk.CTkLabel(chapter_word_frame, text="每章字数:", font=("Microsoft YaHei", 12)) word_number_label.grid(row=0, column=2, padx=(15, 5), pady=5, sticky="e") word_number_entry = ctk.CTkEntry(chapter_word_frame, textvariable=self.word_number_var, width=60, font=("Microsoft YaHei", 12)) word_number_entry.grid(row=0, column=3, padx=5, pady=5, sticky="w") # 4) 保存路径 row_fp = 3 self.create_label_with_help( parent=self.params_frame, label_text="保存路径:", tooltip_key="filepath", row=row_fp, column=0, font=("Microsoft YaHei", 12) ) self.filepath_frame = ctk.CTkFrame(self.params_frame) self.filepath_frame.grid(row=row_fp, column=1, padx=5, pady=5, sticky="nsew") self.filepath_frame.columnconfigure(0, weight=1) filepath_entry = ctk.CTkEntry(self.filepath_frame, textvariable=self.filepath_var, font=("Microsoft YaHei", 12)) filepath_entry.grid(row=0, column=0, padx=5, pady=5, sticky="ew") browse_btn = ctk.CTkButton(self.filepath_frame, text="浏览...", command=self.browse_folder, width=60, font=("Microsoft YaHei", 12)) browse_btn.grid(row=0, column=1, padx=5, pady=5, sticky="e") # 5) 章节号 row_chap_num = 4 self.create_label_with_help( parent=self.params_frame, label_text="章节号:", tooltip_key="chapter_num", row=row_chap_num, column=0, font=("Microsoft YaHei", 12) ) chapter_num_entry = ctk.CTkEntry(self.params_frame, textvariable=self.chapter_num_var, width=80, font=("Microsoft YaHei", 12)) chapter_num_entry.grid(row=row_chap_num, column=1, padx=5, pady=5, sticky="w") # 6) 本章指导 row_user_guide = 5 guide_label_frame = self.create_label_with_help( parent=self.params_frame, label_text="本章指导:", tooltip_key="user_guidance", row=row_user_guide, column=0, font=("Microsoft YaHei", 12), sticky="ne" ) self.user_guide_text = ctk.CTkTextbox(self.params_frame, height=80, wrap="word", font=("Microsoft YaHei", 12)) self.user_guide_text.grid(row=row_user_guide, column=1, padx=5, pady=5, sticky="nsew") # 7) 可选元素：核心人物/关键道具/空间坐标/时间压力 row_idx = 6 # 核心人物 self.create_label_with_help( parent=self.params_frame, label_text="核心人物:", tooltip_key="characters_involved", row=row_idx, column=0, font=("Microsoft YaHei", 12) ) char_inv_entry = ctk.CTkEntry(self.params_frame, textvariable=self.characters_involved_var, font=("Microsoft YaHei", 12)) char_inv_entry.grid(row=row_idx, column=1, padx=5, pady=5, sticky="ew") row_idx += 1 # 关键道具 self.create_label_with_help( parent=self.params_frame, label_text="关键道具:", tooltip_key="key_items", row=row_idx, column=0, font=("Microsoft YaHei", 12) ) key_items_entry = ctk.CTkEntry(self.params_frame, textvariable=self.key_items_var, font=("Microsoft YaHei", 12)) key_items_entry.grid(row=row_idx, column=1, padx=5, pady=5, sticky="ew") row_idx += 1 # 空间坐标 self.create_label_with_help( parent=self.params_frame, label_text="空间坐标:", tooltip_key="scene_location", row=row_idx, column=0, font=("Microsoft YaHei", 12) ) scene_loc_entry = ctk.CTkEntry(self.params_frame, textvariable=self.scene_location_var, font=("Microsoft YaHei", 12)) scene_loc_entry.grid(row=row_idx, column=1, padx=5, pady=5, sticky="ew") row_idx += 1 # 时间压力 self.create_label_with_help( parent=self.params_frame, label_text="时间压力:", tooltip_key="time_constraint", row=row_idx, column=0, font=("Microsoft YaHei", 12) ) time_const_entry = ctk.CTkEntry(self.params_frame, textvariable=self.time_constraint_var, font=("Microsoft YaHei", 12)) time_const_entry.grid(row=row_idx, column=1, padx=5, pady=5, sticky="ew") def build_optional_buttons_area(self, start_row=2): self.optional_btn_frame = ctk.CTkFrame(self.right_frame) self.optional_btn_frame.grid(row=start_row, column=0, sticky="ew", padx=5, pady=5) self.optional_btn_frame.columnconfigure((0, 1, 2, 3), weight=1) self.btn_check_consistency = ctk.CTkButton( self.optional_btn_frame, text="一致性审校", command=self.do_consistency_check, font=("Microsoft YaHei", 12) ) self.btn_check_consistency.grid(row=0, column=0, padx=5, pady=5, sticky="ew") self.btn_import_knowledge = ctk.CTkButton( self.optional_btn_frame, text="导入知识库", command=self.import_knowledge_handler, font=("Microsoft YaHei", 12) ) self.btn_import_knowledge.grid(row=0, column=1, padx=5, pady=5, sticky="ew") self.btn_clear_vectorstore = ctk.CTkButton( self.optional_btn_frame, text="清空向量库", fg_color="red", command=self.clear_vectorstore_handler, font=("Microsoft YaHei", 12) ) self.btn_clear_vectorstore.grid(row=0, column=2, padx=5, pady=5, sticky="ew") self.plot_arcs_btn = ctk.CTkButton( self.optional_btn_frame, text="查看剧情要点", command=self.show_plot_arcs_ui, font=("Microsoft YaHei", 12) ) self.plot_arcs_btn.grid(row=0, column=3, padx=5, pady=5, sticky="ew") def load_config_btn(self): cfg = load_config(self.config_file) if cfg: self.api_key_var.set(cfg.get("api_key", "")) self.base_url_var.set(cfg.get("base_url", "")) self.interface_format_var.set(cfg.get("interface_format", "OpenAI")) self.model_name_var.set(cfg.get("model_name", "")) self.temperature_var.set(cfg.get("temperature", 0.7)) self.max_tokens_var.set(cfg.get("max_tokens", 2048)) self.embedding_api_key_var.set(cfg.get("embedding_api_key", "")) self.embedding_interface_format_var.set(cfg.get("embedding_interface_format", "OpenAI")) self.embedding_url_var.set(cfg.get("embedding_url", "")) self.embedding_model_name_var.set(cfg.get("embedding_model_name", "")) self.embedding_retrieval_k_var.set(str(cfg.get("embedding_retrieval_k", 4))) self.genre_var.set(cfg.get("genre", "")) self.num_chapters_var.set(str(cfg.get("num_chapters", 10))) self.word_number_var.set(str(cfg.get("word_number", 3000))) self.filepath_var.set(cfg.get("filepath", "")) topic_value = cfg.get("topic", "") self.topic_text.delete("0.0", "end") self.topic_text.insert("0.0", topic_value) self.log("已加载配置。") else: messagebox.showwarning("提示", "未找到或无法读取配置文件。") def save_config_btn(self): config_data = { "api_key": self.api_key_var.get(), "base_url": self.base_url_var.get(), "interface_format": self.interface_format_var.get(), "model_name": self.model_name_var.get(), "temperature": self.temperature_var.get(), "max_tokens": self.max_tokens_var.get(), "embedding_api_key": self.embedding_api_key_var.get(), "embedding_interface_format": self.embedding_interface_format_var.get(), "embedding_url": self.embedding_url_var.get(), "embedding_model_name": self.embedding_model_name_var.get(), "embedding_retrieval_k": self.safe_get_int(self.embedding_retrieval_k_var, 4), "topic": self.topic_text.get("0.0", "end").strip(), "genre": self.genre_var.get(), "num_chapters": self.safe_get_int(self.num_chapters_var, 10), "word_number": self.safe_get_int(self.word_number_var, 3000), "filepath": self.filepath_var.get() } if save_config(config_data, self.config_file): messagebox.showinfo("提示", "配置已保存至 config.json") self.log("配置已保存。") else: messagebox.showerror("错误", "保存配置失败。") def browse_folder(self): selected_dir = filedialog.askdirectory() if selected_dir: self.filepath_var.set(selected_dir) def log(self, message: str): self.log_text.configure(state="normal") self.log_text.insert("end", message + "\n") self.log_text.see("end") self.log_text.configure(state="disabled") def safe_log(self, message: str): self.master.after(0, lambda: self.log(message)) def disable_button_safe(self, btn): self.master.after(0, lambda: btn.configure(state="disabled")) def enable_button_safe(self, btn): self.master.after(0, lambda: btn.configure(state="normal")) def handle_exception(self, context: str): full_message = f"{context}\n{traceback.format_exc()}" logging.error(full_message) self.safe_log(full_message) # ============ Step1: 生成小说架构 ============ def generate_novel_architecture_ui(self): filepath = self.filepath_var.get().strip() if not filepath: messagebox.showwarning("警告", "请先选择保存文件路径") return def task(): self.disable_button_safe(self.btn_generate_architecture) try: interface_format = self.interface_format_var.get().strip() api_key = self.api_key_var.get().strip() base_url = self.base_url_var.get().strip() model_name = self.model_name_var.get().strip() temperature = self.temperature_var.get() max_tokens = self.max_tokens_var.get() topic = self.topic_text.get("0.0", "end").strip() 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) self.safe_log("开始生成小说架构...") Novel_architecture_generate( interface_format=interface_format, api_key=api_key, base_url=base_url, llm_model=model_name, topic=topic, genre=genre, number_of_chapters=num_chapters, word_number=word_number, filepath=filepath, temperature=temperature, max_tokens=max_tokens ) self.safe_log("✅ 小说架构生成完成。请在 'Novel Architecture' 标签页查看或编辑。") except Exception: self.handle_exception("生成小说架构时出错") finally: self.enable_button_safe(self.btn_generate_architecture) threading.Thread(target=task, daemon=True).start() # ============ Step2: 生成章节蓝图 ============ def generate_chapter_blueprint_ui(self): filepath = self.filepath_var.get().strip() if not filepath: messagebox.showwarning("警告", "请先选择保存文件路径") return def task(): self.disable_button_safe(self.btn_generate_directory) try: interface_format = self.interface_format_var.get().strip() api_key = self.api_key_var.get().strip() base_url = self.base_url_var.get().strip() model_name = self.model_name_var.get().strip() number_of_chapters = self.safe_get_int(self.num_chapters_var, 10) temperature = self.temperature_var.get() max_tokens = self.max_tokens_var.get() self.safe_log("开始生成章节蓝图...") Chapter_blueprint_generate( interface_format=interface_format, api_key=api_key, base_url=base_url, llm_model=model_name, number_of_chapters=number_of_chapters, filepath=filepath, temperature=temperature, max_tokens=max_tokens ) self.safe_log("✅ 章节蓝图生成完成。请在 'Chapter Blueprint' 标签页查看或编辑。") except Exception: self.handle_exception("生成章节蓝图时出错") finally: self.enable_button_safe(self.btn_generate_directory) threading.Thread(target=task, daemon=True).start() # ============ Step3: 生成章节草稿 ============ def generate_chapter_draft_ui(self): filepath = self.filepath_var.get().strip() if not filepath: messagebox.showwarning("警告", "请先配置保存文件路径。") return def task(): self.disable_button_safe(self.btn_generate_chapter) try: interface_format = self.interface_format_var.get().strip() api_key = self.api_key_var.get().strip() base_url = self.base_url_var.get().strip() model_name = self.model_name_var.get().strip() temperature = self.temperature_var.get() max_tokens = self.max_tokens_var.get() chap_num = self.safe_get_int(self.chapter_num_var, 1) word_number = self.safe_get_int(self.word_number_var, 3000) user_guidance = self.user_guide_text.get("0.0", "end").strip() char_inv = self.characters_involved_var.get().strip() key_items = self.key_items_var.get().strip() scene_loc = self.scene_location_var.get().strip() time_constr = self.time_constraint_var.get().strip() embedding_api_key = self.embedding_api_key_var.get().strip() embedding_url = self.embedding_url_var.get().strip() embedding_interface_format = self.embedding_interface_format_var.get().strip() embedding_model_name = self.embedding_model_name_var.get().strip() embedding_k = self.safe_get_int(self.embedding_retrieval_k_var, 4) self.safe_log(f"开始生成第{chap_num}章草稿...") draft_text = generate_chapter_draft( api_key=api_key, base_url=base_url, model_name=model_name, filepath=filepath, novel_number=chap_num, word_number=word_number, temperature=temperature, user_guidance=user_guidance, characters_involved=char_inv, key_items=key_items, scene_location=scene_loc, time_constraint=time_constr, embedding_api_key=embedding_api_key, embedding_url=embedding_url, embedding_interface_format=embedding_interface_format, embedding_model_name=embedding_model_name, embedding_retrieval_k=embedding_k, interface_format=interface_format, max_tokens=max_tokens ) if draft_text: self.safe_log(f"✅ 第{chap_num}章草稿生成完成。请在左侧查看或编辑。") self.master.after(0, lambda: self.show_chapter_in_textbox(draft_text)) else: self.safe_log("⚠️ 本章草稿生成失败或无内容。") except Exception: self.handle_exception("生成章节草稿时出错") finally: self.enable_button_safe(self.btn_generate_chapter) threading.Thread(target=task, daemon=True).start() def show_chapter_in_textbox(self, text: str): self.chapter_result.delete("0.0", "end") self.chapter_result.insert("0.0", text) self.chapter_result.see("end") # ============ Step4: 定稿章节 ============ def finalize_chapter_ui(self): filepath = self.filepath_var.get().strip() if not filepath: messagebox.showwarning("警告", "请先配置保存文件路径。") return def task(): self.disable_button_safe(self.btn_finalize_chapter) try: interface_format = self.interface_format_var.get().strip() api_key = self.api_key_var.get().strip() base_url = self.base_url_var.get().strip() model_name = self.model_name_var.get().strip() temperature = self.temperature_var.get() max_tokens = self.max_tokens_var.get() embedding_api_key = self.embedding_api_key_var.get().strip() embedding_url = self.embedding_url_var.get().strip() embedding_interface_format = self.embedding_interface_format_var.get().strip() embedding_model_name = self.embedding_model_name_var.get().strip() chap_num = self.safe_get_int(self.chapter_num_var, 1) word_number = self.safe_get_int(self.word_number_var, 3000) self.safe_log(f"开始定稿第{chap_num}章...") chapters_dir = os.path.join(filepath, "chapters") os.makedirs(chapters_dir, exist_ok=True) chapter_file = os.path.join(chapters_dir, f"chapter_{chap_num}.txt") edited_text = self.chapter_result.get("0.0", "end").strip() clear_file_content(chapter_file) save_string_to_txt(edited_text, chapter_file) finalize_chapter( novel_number=chap_num, word_number=word_number, api_key=api_key, base_url=base_url, model_name=model_name, temperature=temperature, filepath=filepath, embedding_api_key=embedding_api_key, embedding_url=embedding_url, embedding_interface_format=embedding_interface_format, embedding_model_name=embedding_model_name, interface_format=interface_format, max_tokens=max_tokens ) self.safe_log(f"✅ 第{chap_num}章定稿完成（已更新全局摘要、角色状态、向量库）。") final_text = read_file(chapter_file) self.master.after(0, lambda: self.show_chapter_in_textbox(final_text)) except Exception: self.handle_exception("定稿章节时出错") finally: self.enable_button_safe(self.btn_finalize_chapter) threading.Thread(target=task, daemon=True).start() # ============ 一致性审校 (可选) ============ def do_consistency_check(self): filepath = self.filepath_var.get().strip() if not filepath: messagebox.showwarning("警告", "请先配置保存文件路径。") return def task(): self.disable_button_safe(self.btn_check_consistency) try: api_key = self.api_key_var.get().strip() base_url = self.base_url_var.get().strip() model_name = self.model_name_var.get().strip() temperature = self.temperature_var.get() chap_num = self.safe_get_int(self.chapter_num_var, 1) chap_file = os.path.join(filepath, "chapters", f"chapter_{chap_num}.txt") chapter_text = read_file(chap_file) if not chapter_text.strip(): self.safe_log("⚠️ 当前章节文件为空或不存在，无法审校。") return self.safe_log("开始一致性审校...") result = check_consistency( novel_setting="", character_state=read_file(os.path.join(filepath, "character_state.txt")), global_summary=read_file(os.path.join(filepath, "global_summary.txt")), chapter_text=chapter_text, api_key=api_key, base_url=base_url, model_name=model_name, temperature=temperature, plot_arcs="" ) self.safe_log("审校结果：") self.safe_log(result) except Exception: self.handle_exception("审校时出错") finally: self.enable_button_safe(self.btn_check_consistency) threading.Thread(target=task, daemon=True).start() # ============ 导入知识库 ============ def import_knowledge_handler(self): selected_file = filedialog.askopenfilename( title="选择要导入的知识库文件", filetypes=[("Text Files", "*.txt"), ("All Files", "*.*")] ) if selected_file: def task(): self.disable_button_safe(self.btn_import_knowledge) try: emb_api_key = self.embedding_api_key_var.get().strip() emb_url = self.embedding_url_var.get().strip() emb_format = self.embedding_interface_format_var.get().strip() emb_model = self.embedding_model_name_var.get().strip() self.safe_log(f"开始导入知识库文件: {selected_file}") import_knowledge_file( embedding_api_key=emb_api_key, embedding_url=emb_url, embedding_interface_format=emb_format, embedding_model_name=emb_model, file_path=selected_file, filepath=self.filepath_var.get().strip() ) self.safe_log("✅ 知识库文件导入完成。") except Exception: self.handle_exception("导入知识库时出错") finally: self.enable_button_safe(self.btn_import_knowledge) threading.Thread(target=task, daemon=True).start() def clear_vectorstore_handler(self): filepath = self.filepath_var.get().strip() if not filepath: messagebox.showwarning("警告", "请先配置保存文件路径。") return first_confirm = messagebox.askyesno("警告", "确定要清空本地向量库吗？此操作不可恢复！") if first_confirm: second_confirm = messagebox.askyesno("二次确认", "你确定真的要删除所有向量数据吗？此操作不可恢复！") if second_confirm: if clear_vector_store(filepath): self.log("已清空向量库。") else: self.log(f"未能清空向量库，请关闭程序后手动删除 {filepath} 下的 vectorstore 文件夹。") def show_plot_arcs_ui(self): filepath = self.filepath_var.get().strip() if not filepath: messagebox.showwarning("警告", "请先在主Tab中设置保存文件路径") return plot_arcs_file = os.path.join(filepath, "plot_arcs.txt") if not os.path.exists(plot_arcs_file): messagebox.showinfo("剧情要点", "当前还未生成任何剧情要点或冲突记录。") return arcs_text = read_file(plot_arcs_file).strip() if not arcs_text: arcs_text = "当前没有记录的剧情要点或冲突。" top = ctk.CTkToplevel(self.master) top.title("剧情要点/未解决冲突") top.geometry("600x400") text_area = ctk.CTkTextbox(top, wrap="word", font=("Microsoft YaHei", 12)) text_area.pack(fill="both", expand=True, padx=10, pady=10) text_area.insert("0.0", arcs_text) text_area.configure(state="disabled") # ============ 其余标签页: Novel Architecture, Chapter Blueprint, Character State, Summary ============ def build_setting_tab(self): self.setting_tab.rowconfigure(0, weight=0) self.setting_tab.rowconfigure(1, weight=1) self.setting_tab.columnconfigure(0, weight=1) load_btn = ctk.CTkButton( self.setting_tab, text="加载 Novel_architecture.txt", command=self.load_novel_architecture, font=("Microsoft YaHei", 12) ) load_btn.grid(row=0, column=0, padx=5, pady=5, sticky="w") save_btn = ctk.CTkButton( self.setting_tab, text="保存修改", command=self.save_novel_architecture, font=("Microsoft YaHei", 12) ) save_btn.grid(row=0, column=0, padx=5, pady=5, sticky="e") self.setting_text = ctk.CTkTextbox(self.setting_tab, wrap="word", font=("Microsoft YaHei", 12)) self.setting_text.grid(row=1, column=0, sticky="nsew", padx=5, pady=5) def load_novel_architecture(self): filepath = self.filepath_var.get().strip() if not filepath: messagebox.showwarning("警告", "请先设置保存文件路径") return filename = os.path.join(filepath, "Novel_architecture.txt") content = read_file(filename) self.setting_text.delete("0.0", "end") self.setting_text.insert("0.0", content) self.log("已加载 Novel_architecture.txt 内容到编辑区。") def save_novel_architecture(self): filepath = self.filepath_var.get().strip() if not filepath: messagebox.showwarning("警告", "请先设置保存文件路径") return content = self.setting_text.get("0.0", "end").strip() filename = os.path.join(filepath, "Novel_architecture.txt") clear_file_content(filename) save_string_to_txt(content, filename) self.log("已保存对 Novel_architecture.txt 的修改。") def build_directory_tab(self): self.directory_tab.rowconfigure(0, weight=0) self.directory_tab.rowconfigure(1, weight=1) self.directory_tab.columnconfigure(0, weight=1) load_btn = ctk.CTkButton( self.directory_tab, text="加载 Novel_directory.txt", command=self.load_chapter_blueprint, font=("Microsoft YaHei", 12) ) load_btn.grid(row=0, column=0, padx=5, pady=5, sticky="w") save_btn = ctk.CTkButton( self.directory_tab, text="保存修改", command=self.save_chapter_blueprint, font=("Microsoft YaHei", 12) ) save_btn.grid(row=0, column=0, padx=5, pady=5, sticky="e") self.directory_text = ctk.CTkTextbox(self.directory_tab, wrap="word", font=("Microsoft YaHei", 12)) self.directory_text.grid(row=1, column=0, sticky="nsew", padx=5, pady=5) def load_chapter_blueprint(self): filepath = self.filepath_var.get().strip() if not filepath: messagebox.showwarning("警告", "请先设置保存文件路径") return filename = os.path.join(filepath, "Novel_directory.txt") content = read_file(filename) self.directory_text.delete("0.0", "end") self.directory_text.insert("0.0", content) self.log("已加载 Novel_directory.txt 内容到编辑区。") def save_chapter_blueprint(self): filepath = self.filepath_var.get().strip() if not filepath: messagebox.showwarning("警告", "请先设置保存文件路径") return content = self.directory_text.get("0.0", "end").strip() filename = os.path.join(filepath, "Novel_directory.txt") clear_file_content(filename) save_string_to_txt(content, filename) self.log("已保存对 Novel_directory.txt 的修改。") def build_character_tab(self): self.character_tab.rowconfigure(0, weight=0) self.character_tab.rowconfigure(1, weight=1) self.character_tab.columnconfigure(0, weight=1) load_btn = ctk.CTkButton( self.character_tab, text="加载 character_state.txt", command=self.load_character_state, font=("Microsoft YaHei", 12) ) load_btn.grid(row=0, column=0, padx=5, pady=5, sticky="w") save_btn = ctk.CTkButton( self.character_tab, text="保存修改", command=self.save_character_state, font=("Microsoft YaHei", 12) ) save_btn.grid(row=0, column=0, padx=5, pady=5, sticky="e") self.character_text = ctk.CTkTextbox(self.character_tab, wrap="word", font=("Microsoft YaHei", 12)) self.character_text.grid(row=1, column=0, sticky="nsew", padx=5, pady=5) def load_character_state(self): filepath = self.filepath_var.get().strip() if not filepath: messagebox.showwarning("警告", "请先设置保存文件路径") return filename = os.path.join(filepath, "character_state.txt") content = read_file(filename) self.character_text.delete("0.0", "end") self.character_text.insert("0.0", content) self.log("已加载 character_state.txt 到编辑区。") def save_character_state(self): filepath = self.filepath_var.get().strip() if not filepath: messagebox.showwarning("警告", "请先设置保存文件路径") return content = self.character_text.get("0.0", "end").strip() filename = os.path.join(filepath, "character_state.txt") clear_file_content(filename) save_string_to_txt(content, filename) self.log("已保存对 character_state.txt 的修改。") def build_summary_tab(self): self.summary_tab.rowconfigure(0, weight=0) self.summary_tab.rowconfigure(1, weight=1) self.summary_tab.columnconfigure(0, weight=1) load_btn = ctk.CTkButton( self.summary_tab, text="加载 global_summary.txt", command=self.load_global_summary, font=("Microsoft YaHei", 12) ) load_btn.grid(row=0, column=0, padx=5, pady=5, sticky="w") save_btn = ctk.CTkButton( self.summary_tab, text="保存修改", command=self.save_global_summary, font=("Microsoft YaHei", 12) ) save_btn.grid(row=0, column=0, padx=5, pady=5, sticky="e") self.summary_text = ctk.CTkTextbox(self.summary_tab, wrap="word", font=("Microsoft YaHei", 12)) self.summary_text.grid(row=1, column=0, sticky="nsew", padx=5, pady=5) def load_global_summary(self): filepath = self.filepath_var.get().strip() if not filepath: messagebox.showwarning("警告", "请先设置保存文件路径") return filename = os.path.join(filepath, "global_summary.txt") content = read_file(filename) self.summary_text.delete("0.0", "end") self.summary_text.insert("0.0", content) self.log("已加载 global_summary.txt 到编辑区。") def save_global_summary(self): filepath = self.filepath_var.get().strip() if not filepath: messagebox.showwarning("警告", "请先设置保存文件路径") return content = self.summary_text.get("0.0", "end").strip() filename = os.path.join(filepath, "global_summary.txt") clear_file_content(filename) save_string_to_txt(content, filename) self.log("已保存对 global_summary.txt 的修改。") # ============ 章节管理标签页 ============ def build_chapters_tab(self): self.chapters_view_tab.rowconfigure(0, weight=0) self.chapters_view_tab.rowconfigure(1, weight=1) self.chapters_view_tab.columnconfigure(0, weight=1) top_frame = ctk.CTkFrame(self.chapters_view_tab) top_frame.grid(row=0, column=0, sticky="ew", padx=5, pady=5) top_frame.columnconfigure(0, weight=0) top_frame.columnconfigure(1, weight=0) top_frame.columnconfigure(2, weight=0) top_frame.columnconfigure(3, weight=0) top_frame.columnconfigure(4, weight=1) prev_btn = ctk.CTkButton(top_frame, text="<< 上一章", command=self.prev_chapter, font=("Microsoft YaHei", 12)) prev_btn.grid(row=0, column=0, padx=5, pady=5, sticky="w") next_btn = ctk.CTkButton(top_frame, text="下一章 >>", command=self.next_chapter, font=("Microsoft YaHei", 12)) next_btn.grid(row=0, column=1, padx=5, pady=5, sticky="w") self.chapter_select_var = ctk.StringVar(value="") self.chapter_select_menu = ctk.CTkOptionMenu( top_frame, values=[], variable=self.chapter_select_var, command=self.on_chapter_selected, font=("Microsoft YaHei", 12) ) self.chapter_select_menu.grid(row=0, column=2, padx=5, pady=5, sticky="w") save_btn = ctk.CTkButton(top_frame, text="保存修改", command=self.save_current_chapter, font=("Microsoft YaHei", 12)) save_btn.grid(row=0, column=3, padx=5, pady=5, sticky="w") refresh_btn = ctk.CTkButton(top_frame, text="刷新章节列表", command=self.refresh_chapters_list, font=("Microsoft YaHei", 12)) refresh_btn.grid(row=0, column=4, padx=5, pady=5, sticky="e") self.chapter_view_text = ctk.CTkTextbox(self.chapters_view_tab, wrap="word", font=("Microsoft YaHei", 12)) self.chapter_view_text.grid(row=1, column=0, sticky="nsew", padx=5, pady=5) self.chapters_list = [] self.refresh_chapters_list() def refresh_chapters_list(self): filepath = self.filepath_var.get().strip() chapters_dir = os.path.join(filepath, "chapters") if not os.path.exists(chapters_dir): self.safe_log("尚未找到 chapters 文件夹，请先生成章节或检查保存路径。") self.chapter_select_menu.configure(values=[]) return all_files = os.listdir(chapters_dir) chapter_nums = [] for f in all_files: if f.startswith("chapter_") and f.endswith(".txt"): number_part = f.replace("chapter_", "").replace(".txt", "") if number_part.isdigit(): chapter_nums.append(number_part) chapter_nums.sort(key=lambda x: int(x)) self.chapters_list = chapter_nums self.chapter_select_menu.configure(values=self.chapters_list) current_selected = self.chapter_select_var.get() if current_selected not in self.chapters_list: if self.chapters_list: self.chapter_select_var.set(self.chapters_list[0]) self.load_chapter_content(self.chapters_list[0]) else: self.chapter_select_var.set("") self.chapter_view_text.delete("0.0", "end") def on_chapter_selected(self, value): self.load_chapter_content(value) def load_chapter_content(self, chapter_number_str): if not chapter_number_str: return filepath = self.filepath_var.get().strip() chapter_file = os.path.join(filepath, "chapters", f"chapter_{chapter_number_str}.txt") if not os.path.exists(chapter_file): self.safe_log(f"章节文件 {chapter_file} 不存在！") return content = read_file(chapter_file) self.chapter_view_text.delete("0.0", "end") self.chapter_view_text.insert("0.0", content) def save_current_chapter(self): chapter_number_str = self.chapter_select_var.get() if not chapter_number_str: messagebox.showwarning("警告", "尚未选择章节，无法保存。") return filepath = self.filepath_var.get().strip() if not filepath: messagebox.showwarning("警告", "请先配置保存文件路径") return chapter_file = os.path.join(filepath, "chapters", f"chapter_{chapter_number_str}.txt") content = self.chapter_view_text.get("0.0", "end").strip() clear_file_content(chapter_file) save_string_to_txt(content, chapter_file) self.safe_log(f"已保存对第 {chapter_number_str} 章的修改。") def prev_chapter(self): if not self.chapters_list: return current = self.chapter_select_var.get() if current not in self.chapters_list: return idx = self.chapters_list.index(current) if idx > 0: new_idx = idx - 1 self.chapter_select_var.set(self.chapters_list[new_idx]) self.load_chapter_content(self.chapters_list[new_idx]) else: messagebox.showinfo("提示", "已经是第一章了。") def next_chapter(self): if not self.chapters_list: return current = self.chapter_select_var.get() if current not in self.chapters_list: return idx = self.chapters_list.index(current) if idx < len(self.chapters_list) - 1: new_idx = idx + 1 self.chapter_select_var.set(self.chapters_list[new_idx]) self.load_chapter_content(self.chapters_list[new_idx]) else: messagebox.showinfo("提示", "已经是最后一章了。") if __name__ == "__main__": app = ctk.CTk() gui = NovelGeneratorGUI(app) app.mainloop()