Initial commit
This commit is contained in:
@@ -0,0 +1,2 @@
|
||||
# ui/__init__.py
|
||||
from .main_window import NovelGeneratorGUI
|
||||
@@ -0,0 +1,140 @@
|
||||
# ui/chapters_tab.py
|
||||
# -*- coding: utf-8 -*-
|
||||
import os
|
||||
import customtkinter as ctk
|
||||
from tkinter import messagebox
|
||||
from ui.context_menu import TextWidgetContextMenu
|
||||
from utils import read_file, save_string_to_txt, clear_file_content
|
||||
|
||||
def build_chapters_tab(self):
|
||||
self.chapters_view_tab = self.tabview.add("Chapters Manage")
|
||||
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=5, padx=5, pady=5, sticky="e")
|
||||
|
||||
self.chapters_word_count_label = ctk.CTkLabel(top_frame, text="字数:0", font=("Microsoft YaHei", 12))
|
||||
self.chapters_word_count_label.grid(row=0, column=4, padx=(0,10), sticky="e")
|
||||
|
||||
self.chapter_view_text = ctk.CTkTextbox(self.chapters_view_tab, wrap="word", font=("Microsoft YaHei", 12))
|
||||
|
||||
def update_word_count(event=None):
|
||||
text = self.chapter_view_text.get("0.0", "end-1c")
|
||||
text_length = len(text)
|
||||
self.chapters_word_count_label.configure(text=f"字数:{text_length}")
|
||||
|
||||
self.chapter_view_text.bind("<KeyRelease>", update_word_count)
|
||||
self.chapter_view_text.bind("<ButtonRelease>", update_word_count)
|
||||
TextWidgetContextMenu(self.chapter_view_text)
|
||||
self.chapter_view_text.grid(row=1, column=0, sticky="nsew", padx=5, pady=5, columnspan=6)
|
||||
|
||||
self.chapters_list = []
|
||||
refresh_chapters_list(self)
|
||||
|
||||
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])
|
||||
load_chapter_content(self, self.chapters_list[0])
|
||||
else:
|
||||
self.chapter_select_var.set("")
|
||||
self.chapter_view_text.delete("0.0", "end")
|
||||
|
||||
def on_chapter_selected(self, value):
|
||||
load_chapter_content(self, 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])
|
||||
load_chapter_content(self, 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])
|
||||
load_chapter_content(self, self.chapters_list[new_idx])
|
||||
else:
|
||||
messagebox.showinfo("提示", "已经是最后一章了。")
|
||||
@@ -0,0 +1,56 @@
|
||||
# ui/character_tab.py
|
||||
# -*- coding: utf-8 -*-
|
||||
import os
|
||||
import customtkinter as ctk
|
||||
from tkinter import messagebox
|
||||
from utils import read_file, save_string_to_txt, clear_file_content
|
||||
from ui.context_menu import TextWidgetContextMenu
|
||||
|
||||
def build_character_tab(self):
|
||||
self.character_tab = self.tabview.add("Character State")
|
||||
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")
|
||||
|
||||
self.character_wordcount_label = ctk.CTkLabel(self.character_tab, text="字数:0", font=("Microsoft YaHei", 12))
|
||||
self.character_wordcount_label.grid(row=0, column=1, 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=2, padx=5, pady=5, sticky="e")
|
||||
|
||||
self.character_text = ctk.CTkTextbox(self.character_tab, wrap="word", font=("Microsoft YaHei", 12))
|
||||
|
||||
def update_word_count(event=None):
|
||||
text = self.character_text.get("0.0", "end-1c")
|
||||
text_length = len(text)
|
||||
self.character_wordcount_label.configure(text=f"字数:{text_length}")
|
||||
|
||||
self.character_text.bind("<KeyRelease>", update_word_count)
|
||||
self.character_text.bind("<ButtonRelease>", update_word_count)
|
||||
TextWidgetContextMenu(self.character_text)
|
||||
self.character_text.grid(row=1, column=0, sticky="nsew", padx=5, pady=5, columnspan=3)
|
||||
|
||||
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 的修改。")
|
||||
@@ -0,0 +1,319 @@
|
||||
# ui/config_tab.py
|
||||
# -*- coding: utf-8 -*-
|
||||
from tkinter import messagebox
|
||||
|
||||
import customtkinter as ctk
|
||||
|
||||
from config_manager import load_config, save_config
|
||||
from tooltips import tooltips
|
||||
|
||||
|
||||
def create_label_with_help(self, parent, label_text, tooltip_key, row, column,
|
||||
font=None, sticky="e", padx=5, pady=5):
|
||||
"""
|
||||
封装一个带"?"按钮的Label,用于展示提示信息。
|
||||
"""
|
||||
frame = ctk.CTkFrame(parent)
|
||||
frame.grid(row=row, column=column, padx=padx, pady=pady, sticky=sticky)
|
||||
frame.columnconfigure(0, weight=0)
|
||||
|
||||
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: messagebox.showinfo("参数说明", tooltips.get(tooltip_key, "暂无说明"))
|
||||
)
|
||||
btn.pack(side="left", padx=3)
|
||||
|
||||
return frame
|
||||
|
||||
def build_config_tabview(self):
|
||||
"""
|
||||
创建包含 LLM Model settings 和 Embedding settings 的选项卡。
|
||||
"""
|
||||
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")
|
||||
|
||||
build_ai_config_tab(self)
|
||||
build_embeddings_config_tab(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_ai_config_tab(self):
|
||||
def on_interface_format_changed(new_value):
|
||||
self.interface_format_var.set(new_value)
|
||||
config_data = load_config(self.config_file)
|
||||
if config_data:
|
||||
config_data["last_interface_format"] = new_value
|
||||
save_config(config_data, self.config_file)
|
||||
if self.loaded_config and "llm_configs" in self.loaded_config and new_value in self.loaded_config["llm_configs"]:
|
||||
llm_conf = self.loaded_config["llm_configs"][new_value]
|
||||
self.api_key_var.set(llm_conf.get("api_key", ""))
|
||||
self.base_url_var.set(llm_conf.get("base_url", self.base_url_var.get()))
|
||||
self.model_name_var.set(llm_conf.get("model_name", ""))
|
||||
self.temperature_var.set(llm_conf.get("temperature", 0.7))
|
||||
self.max_tokens_var.set(llm_conf.get("max_tokens", 8192))
|
||||
self.timeout_var.set(llm_conf.get("timeout", 600))
|
||||
else:
|
||||
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")
|
||||
self.model_name_var.set("gpt-4o-mini")
|
||||
elif new_value == "Azure OpenAI":
|
||||
self.base_url_var.set("https://[az].openai.azure.com/openai/deployments/[model]/chat/completions?api-version=2024-08-01-preview")
|
||||
elif new_value == "DeepSeek":
|
||||
self.base_url_var.set("https://api.deepseek.com/v1")
|
||||
self.model_name_var.set("deepseek-chat")
|
||||
elif new_value == "Gemini":
|
||||
self.base_url_var.set("")
|
||||
elif new_value == "Azure AI":
|
||||
self.base_url_var.set("https://<your-endpoint>.services.ai.azure.com/models/chat/completions?api-version=2024-05-01-preview")
|
||||
elif new_value == "阿里云百炼":
|
||||
self.base_url_var.set("https://dashscope.aliyuncs.com/compatible-mode/v1")
|
||||
self.model_name_var.set("qwen-plus")
|
||||
elif new_value == "硅基流动":
|
||||
self.base_url_var.set("https://api.siliconflow.cn/v1")
|
||||
self.model_name_var.set("deepseek-ai/DeepSeek-V3")
|
||||
|
||||
for i in range(7):
|
||||
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
|
||||
create_label_with_help(self, 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),show="*")
|
||||
api_key_entry.grid(row=0, column=1, padx=5, pady=5, columnspan=2, sticky="nsew")
|
||||
|
||||
# 2) Base URL
|
||||
create_label_with_help(self, 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) 接口格式
|
||||
create_label_with_help(self, parent=self.ai_config_tab, label_text="LLM 接口格式:", tooltip_key="interface_format", row=2, column=0, font=("Microsoft YaHei", 12))
|
||||
# 在这里的接口选项列表中添加 "硅基流动"
|
||||
interface_options = ["DeepSeek", "阿里云百炼", "OpenAI", "Azure OpenAI", "Azure AI", "Ollama", "ML Studio", "Gemini", "火山引擎", "硅基流动"]
|
||||
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
|
||||
create_label_with_help(self, 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
|
||||
create_label_with_help(self, 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
|
||||
create_label_with_help(self, 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")
|
||||
|
||||
# 7) Timeout (sec)
|
||||
create_label_with_help(self, parent=self.ai_config_tab, label_text="Timeout (sec):", tooltip_key="timeout", row=6, column=0, font=("Microsoft YaHei", 12))
|
||||
def update_timeout_label(value):
|
||||
integer_val = int(float(value))
|
||||
self.timeout_value_label.configure(text=str(integer_val))
|
||||
timeout_slider = ctk.CTkSlider(self.ai_config_tab, from_=0, to=3600, number_of_steps=3600, command=update_timeout_label, variable=self.timeout_var)
|
||||
timeout_slider.grid(row=6, column=1, padx=5, pady=5, sticky="we")
|
||||
self.timeout_value_label = ctk.CTkLabel(self.ai_config_tab, text=str(self.timeout_var.get()), font=("Microsoft YaHei", 12))
|
||||
self.timeout_value_label.grid(row=6, column=2, padx=5, pady=5, sticky="w")
|
||||
|
||||
# 添加测试按钮
|
||||
test_btn = ctk.CTkButton(self.ai_config_tab, text="测试配置", command=self.test_llm_config, font=("Microsoft YaHei", 12))
|
||||
test_btn.grid(row=7, column=0, columnspan=3, padx=5, pady=5, sticky="ew")
|
||||
|
||||
def build_embeddings_config_tab(self):
|
||||
def on_embedding_interface_changed(new_value):
|
||||
self.embedding_interface_format_var.set(new_value)
|
||||
config_data = load_config(self.config_file)
|
||||
if config_data:
|
||||
config_data["last_embedding_interface_format"] = new_value
|
||||
save_config(config_data, self.config_file)
|
||||
if self.loaded_config and "embedding_configs" in self.loaded_config and new_value in self.loaded_config["embedding_configs"]:
|
||||
emb_conf = self.loaded_config["embedding_configs"][new_value]
|
||||
self.embedding_api_key_var.set(emb_conf.get("api_key", ""))
|
||||
self.embedding_url_var.set(emb_conf.get("base_url", self.embedding_url_var.get()))
|
||||
self.embedding_model_name_var.set(emb_conf.get("model_name", ""))
|
||||
self.embedding_retrieval_k_var.set(str(emb_conf.get("retrieval_k", 4)))
|
||||
else:
|
||||
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")
|
||||
self.embedding_model_name_var.set("text-embedding-ada-002")
|
||||
elif new_value == "Azure OpenAI":
|
||||
self.embedding_url_var.set("https://[az].openai.azure.com/openai/deployments/[model]/embeddings?api-version=2023-05-15")
|
||||
elif new_value == "DeepSeek":
|
||||
self.embedding_url_var.set("https://api.deepseek.com/v1")
|
||||
elif new_value == "Gemini":
|
||||
self.embedding_url_var.set("https://generativelanguage.googleapis.com/v1beta/")
|
||||
self.embedding_model_name_var.set("models/text-embedding-004")
|
||||
elif new_value == "SiliconFlow":
|
||||
self.embedding_url_var.set("https://api.siliconflow.cn/v1/embeddings")
|
||||
self.embedding_model_name_var.set("BAAI/bge-m3")
|
||||
|
||||
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
|
||||
create_label_with_help(self, 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 接口格式
|
||||
create_label_with_help(self, 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", "Azure OpenAI", "Gemini", "Ollama", "ML Studio","SiliconFlow"]
|
||||
|
||||
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
|
||||
create_label_with_help(self, 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
|
||||
create_label_with_help(self, 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
|
||||
create_label_with_help(self, 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")
|
||||
|
||||
# 添加测试按钮
|
||||
test_btn = ctk.CTkButton(self.embeddings_config_tab, text="测试配置", command=self.test_embedding_config, font=("Microsoft YaHei", 12))
|
||||
test_btn.grid(row=5, column=0, columnspan=2, padx=5, pady=5, sticky="ew")
|
||||
|
||||
def load_config_btn(self):
|
||||
cfg = load_config(self.config_file)
|
||||
if cfg:
|
||||
last_llm = cfg.get("last_interface_format", "OpenAI")
|
||||
last_embedding = cfg.get("last_embedding_interface_format", "OpenAI")
|
||||
self.interface_format_var.set(last_llm)
|
||||
self.embedding_interface_format_var.set(last_embedding)
|
||||
llm_configs = cfg.get("llm_configs", {})
|
||||
if last_llm in llm_configs:
|
||||
llm_conf = llm_configs[last_llm]
|
||||
self.api_key_var.set(llm_conf.get("api_key", ""))
|
||||
self.base_url_var.set(llm_conf.get("base_url", "https://api.openai.com/v1"))
|
||||
self.model_name_var.set(llm_conf.get("model_name", "gpt-4o-mini"))
|
||||
self.temperature_var.set(llm_conf.get("temperature", 0.7))
|
||||
self.max_tokens_var.set(llm_conf.get("max_tokens", 8192))
|
||||
self.timeout_var.set(llm_conf.get("timeout", 600))
|
||||
embedding_configs = cfg.get("embedding_configs", {})
|
||||
if last_embedding in embedding_configs:
|
||||
emb_conf = embedding_configs[last_embedding]
|
||||
self.embedding_api_key_var.set(emb_conf.get("api_key", ""))
|
||||
self.embedding_url_var.set(emb_conf.get("base_url", "https://api.openai.com/v1"))
|
||||
self.embedding_model_name_var.set(emb_conf.get("model_name", "text-embedding-ada-002"))
|
||||
self.embedding_retrieval_k_var.set(str(emb_conf.get("retrieval_k", 4)))
|
||||
other_params = cfg.get("other_params", {})
|
||||
self.topic_text.delete("0.0", "end")
|
||||
self.topic_text.insert("0.0", other_params.get("topic", ""))
|
||||
self.genre_var.set(other_params.get("genre", "玄幻"))
|
||||
self.num_chapters_var.set(str(other_params.get("num_chapters", 10)))
|
||||
self.word_number_var.set(str(other_params.get("word_number", 3000)))
|
||||
self.filepath_var.set(other_params.get("filepath", ""))
|
||||
self.chapter_num_var.set(str(other_params.get("chapter_num", "1")))
|
||||
self.user_guide_text.delete("0.0", "end")
|
||||
self.user_guide_text.insert("0.0", other_params.get("user_guidance", ""))
|
||||
self.characters_involved_var.set(other_params.get("characters_involved", ""))
|
||||
self.key_items_var.set(other_params.get("key_items", ""))
|
||||
self.scene_location_var.set(other_params.get("scene_location", ""))
|
||||
self.time_constraint_var.set(other_params.get("time_constraint", ""))
|
||||
self.log("已加载配置。")
|
||||
else:
|
||||
messagebox.showwarning("提示", "未找到或无法读取配置文件。")
|
||||
|
||||
def save_config_btn(self):
|
||||
current_llm_interface = self.interface_format_var.get().strip()
|
||||
current_embedding_interface = self.embedding_interface_format_var.get().strip()
|
||||
llm_config = {
|
||||
"api_key": self.api_key_var.get(),
|
||||
"base_url": self.base_url_var.get(),
|
||||
"model_name": self.model_name_var.get(),
|
||||
"temperature": self.temperature_var.get(),
|
||||
"max_tokens": self.max_tokens_var.get(),
|
||||
"timeout": self.safe_get_int(self.timeout_var, 600)
|
||||
}
|
||||
embedding_config = {
|
||||
"api_key": self.embedding_api_key_var.get(),
|
||||
"base_url": self.embedding_url_var.get(),
|
||||
"model_name": self.embedding_model_name_var.get(),
|
||||
"retrieval_k": self.safe_get_int(self.embedding_retrieval_k_var, 4)
|
||||
}
|
||||
other_params = {
|
||||
"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(),
|
||||
"chapter_num": self.chapter_num_var.get(),
|
||||
"user_guidance": self.user_guide_text.get("0.0", "end").strip(),
|
||||
"characters_involved": self.characters_involved_var.get(),
|
||||
"key_items": self.key_items_var.get(),
|
||||
"scene_location": self.scene_location_var.get(),
|
||||
"time_constraint": self.time_constraint_var.get()
|
||||
}
|
||||
existing_config = load_config(self.config_file)
|
||||
if not existing_config:
|
||||
existing_config = {}
|
||||
existing_config["last_interface_format"] = current_llm_interface
|
||||
existing_config["last_embedding_interface_format"] = current_embedding_interface
|
||||
if "llm_configs" not in existing_config:
|
||||
existing_config["llm_configs"] = {}
|
||||
existing_config["llm_configs"][current_llm_interface] = llm_config
|
||||
|
||||
if "embedding_configs" not in existing_config:
|
||||
existing_config["embedding_configs"] = {}
|
||||
existing_config["embedding_configs"][current_embedding_interface] = embedding_config
|
||||
|
||||
existing_config["other_params"] = other_params
|
||||
|
||||
if save_config(existing_config, self.config_file):
|
||||
messagebox.showinfo("提示", "配置已保存至 config.json")
|
||||
self.log("配置已保存。")
|
||||
else:
|
||||
messagebox.showerror("错误", "保存配置失败。")
|
||||
@@ -0,0 +1,54 @@
|
||||
# ui/context_menu.py
|
||||
# -*- coding: utf-8 -*-
|
||||
import tkinter as tk
|
||||
import customtkinter as ctk
|
||||
|
||||
class TextWidgetContextMenu:
|
||||
"""
|
||||
为 customtkinter.TextBox 或 tkinter.Text 提供右键复制/剪切/粘贴/全选的功能。
|
||||
"""
|
||||
def __init__(self, widget):
|
||||
self.widget = widget
|
||||
self.menu = tk.Menu(widget, tearoff=0)
|
||||
self.menu.add_command(label="复制", command=self.copy)
|
||||
self.menu.add_command(label="粘贴", command=self.paste)
|
||||
self.menu.add_command(label="剪切", command=self.cut)
|
||||
self.menu.add_separator()
|
||||
self.menu.add_command(label="全选", command=self.select_all)
|
||||
|
||||
# 绑定右键事件
|
||||
self.widget.bind("<Button-3>", self.show_menu)
|
||||
|
||||
def show_menu(self, event):
|
||||
if isinstance(self.widget, ctk.CTkTextbox):
|
||||
try:
|
||||
self.menu.tk_popup(event.x_root, event.y_root)
|
||||
finally:
|
||||
self.menu.grab_release()
|
||||
|
||||
def copy(self):
|
||||
try:
|
||||
text = self.widget.get("sel.first", "sel.last")
|
||||
self.widget.clipboard_clear()
|
||||
self.widget.clipboard_append(text)
|
||||
except tk.TclError:
|
||||
pass # 没有选中文本时忽略错误
|
||||
|
||||
def paste(self):
|
||||
try:
|
||||
text = self.widget.clipboard_get()
|
||||
self.widget.insert("insert", text)
|
||||
except tk.TclError:
|
||||
pass # 剪贴板为空时忽略错误
|
||||
|
||||
def cut(self):
|
||||
try:
|
||||
text = self.widget.get("sel.first", "sel.last")
|
||||
self.widget.delete("sel.first", "sel.last")
|
||||
self.widget.clipboard_clear()
|
||||
self.widget.clipboard_append(text)
|
||||
except tk.TclError:
|
||||
pass # 没有选中文本时忽略错误
|
||||
|
||||
def select_all(self):
|
||||
self.widget.tag_add("sel", "1.0", "end")
|
||||
@@ -0,0 +1,56 @@
|
||||
# ui/directory_tab.py
|
||||
# -*- coding: utf-8 -*-
|
||||
import os
|
||||
import customtkinter as ctk
|
||||
from tkinter import messagebox
|
||||
from utils import read_file, save_string_to_txt, clear_file_content
|
||||
from ui.context_menu import TextWidgetContextMenu
|
||||
|
||||
def build_directory_tab(self):
|
||||
self.directory_tab = self.tabview.add("Chapter Blueprint")
|
||||
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")
|
||||
|
||||
self.directory_word_count_label = ctk.CTkLabel(self.directory_tab, text="字数:0", font=("Microsoft YaHei", 12))
|
||||
self.directory_word_count_label.grid(row=0, column=1, 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=2, padx=5, pady=5, sticky="e")
|
||||
|
||||
self.directory_text = ctk.CTkTextbox(self.directory_tab, wrap="word", font=("Microsoft YaHei", 12))
|
||||
|
||||
def update_word_count(event=None):
|
||||
text = self.directory_text.get("0.0", "end")
|
||||
count = len(text) - 1
|
||||
self.directory_word_count_label.configure(text=f"字数:{count}")
|
||||
|
||||
self.directory_text.bind("<KeyRelease>", update_word_count)
|
||||
self.directory_text.bind("<ButtonRelease>", update_word_count)
|
||||
TextWidgetContextMenu(self.directory_text)
|
||||
self.directory_text.grid(row=1, column=0, sticky="nsew", padx=5, pady=5, columnspan=3)
|
||||
|
||||
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 的修改。")
|
||||
@@ -0,0 +1,538 @@
|
||||
# ui/generation_handlers.py
|
||||
# -*- coding: utf-8 -*-
|
||||
import os
|
||||
import threading
|
||||
import tkinter as tk
|
||||
from tkinter import messagebox
|
||||
import customtkinter as ctk
|
||||
import traceback
|
||||
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,
|
||||
enrich_chapter_text
|
||||
)
|
||||
from consistency_checker import check_consistency
|
||||
|
||||
def generate_novel_architecture_ui(self):
|
||||
filepath = self.filepath_var.get().strip()
|
||||
if not filepath:
|
||||
messagebox.showwarning("警告", "请先选择保存文件路径")
|
||||
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()
|
||||
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()
|
||||
timeout_val = self.safe_get_int(self.timeout_var, 600)
|
||||
|
||||
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)
|
||||
# 获取内容指导
|
||||
user_guidance = self.user_guide_text.get("0.0", "end").strip()
|
||||
|
||||
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,
|
||||
timeout=timeout_val,
|
||||
user_guidance=user_guidance # 添加内容指导参数
|
||||
)
|
||||
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()
|
||||
|
||||
def generate_chapter_blueprint_ui(self):
|
||||
filepath = self.filepath_var.get().strip()
|
||||
if not filepath:
|
||||
messagebox.showwarning("警告", "请先选择保存文件路径")
|
||||
return
|
||||
|
||||
def task():
|
||||
if not messagebox.askyesno("确认", "确定要生成章节目录吗?"):
|
||||
self.enable_button_safe(self.btn_generate_chapter)
|
||||
return
|
||||
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()
|
||||
timeout_val = self.safe_get_int(self.timeout_var, 600)
|
||||
user_guidance = self.user_guide_text.get("0.0", "end").strip() # 新增获取用户指导
|
||||
|
||||
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,
|
||||
timeout=timeout_val,
|
||||
user_guidance=user_guidance # 新增参数
|
||||
)
|
||||
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()
|
||||
|
||||
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()
|
||||
timeout_val = self.safe_get_int(self.timeout_var, 600)
|
||||
|
||||
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}章草稿:准备生成请求提示词...")
|
||||
|
||||
# 调用新添加的 build_chapter_prompt 函数构造初始提示词
|
||||
from novel_generator.chapter import build_chapter_prompt
|
||||
prompt_text = build_chapter_prompt(
|
||||
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,
|
||||
timeout=timeout_val
|
||||
)
|
||||
|
||||
# 弹出可编辑提示词对话框,等待用户确认或取消
|
||||
result = {"prompt": None}
|
||||
event = threading.Event()
|
||||
|
||||
def create_dialog():
|
||||
dialog = ctk.CTkToplevel(self.master)
|
||||
dialog.title("当前章节请求提示词(可编辑)")
|
||||
dialog.geometry("600x400")
|
||||
text_box = ctk.CTkTextbox(dialog, wrap="word", font=("Microsoft YaHei", 12))
|
||||
text_box.pack(fill="both", expand=True, padx=10, pady=10)
|
||||
|
||||
# 字数统计标签
|
||||
wordcount_label = ctk.CTkLabel(dialog, text="字数:0", font=("Microsoft YaHei", 12))
|
||||
wordcount_label.pack(side="left", padx=(10,0), pady=5)
|
||||
|
||||
# 插入角色内容
|
||||
final_prompt = prompt_text
|
||||
role_names = [name.strip() for name in self.char_inv_text.get("0.0", "end").strip().split(',') if name.strip()]
|
||||
role_lib_path = os.path.join(filepath, "角色库")
|
||||
role_contents = []
|
||||
|
||||
if os.path.exists(role_lib_path):
|
||||
for root, dirs, files in os.walk(role_lib_path):
|
||||
for file in files:
|
||||
if file.endswith(".txt") and os.path.splitext(file)[0] in role_names:
|
||||
file_path = os.path.join(root, file)
|
||||
try:
|
||||
with open(file_path, 'r', encoding='utf-8') as f:
|
||||
role_contents.append(f.read().strip()) # 直接使用文件内容,不添加重复名字
|
||||
except Exception as e:
|
||||
self.safe_log(f"读取角色文件 {file} 失败: {str(e)}")
|
||||
|
||||
if role_contents:
|
||||
role_content_str = "\n".join(role_contents)
|
||||
# 更精确的替换逻辑,处理不同情况下的占位符
|
||||
placeholder_variations = [
|
||||
"核心人物(可能未指定):{characters_involved}",
|
||||
"核心人物:{characters_involved}",
|
||||
"核心人物(可能未指定):{characters_involved}",
|
||||
"核心人物:{characters_involved}"
|
||||
]
|
||||
|
||||
for placeholder in placeholder_variations:
|
||||
if placeholder in final_prompt:
|
||||
final_prompt = final_prompt.replace(
|
||||
placeholder,
|
||||
f"核心人物:\n{role_content_str}"
|
||||
)
|
||||
break
|
||||
else: # 如果没有找到任何已知占位符变体
|
||||
lines = final_prompt.split('\n')
|
||||
for i, line in enumerate(lines):
|
||||
if "核心人物" in line and ":" in line:
|
||||
lines[i] = f"核心人物:\n{role_content_str}"
|
||||
break
|
||||
final_prompt = '\n'.join(lines)
|
||||
|
||||
text_box.insert("0.0", final_prompt)
|
||||
# 更新字数函数
|
||||
def update_word_count(event=None):
|
||||
text = text_box.get("0.0", "end-1c")
|
||||
text_length = len(text)
|
||||
wordcount_label.configure(text=f"字数:{text_length}")
|
||||
|
||||
text_box.bind("<KeyRelease>", update_word_count)
|
||||
text_box.bind("<ButtonRelease>", update_word_count)
|
||||
update_word_count() # 初始化统计
|
||||
|
||||
button_frame = ctk.CTkFrame(dialog)
|
||||
button_frame.pack(pady=10)
|
||||
def on_confirm():
|
||||
result["prompt"] = text_box.get("1.0", "end").strip()
|
||||
dialog.destroy()
|
||||
event.set()
|
||||
def on_cancel():
|
||||
result["prompt"] = None
|
||||
dialog.destroy()
|
||||
event.set()
|
||||
btn_confirm = ctk.CTkButton(button_frame, text="确认使用", font=("Microsoft YaHei", 12), command=on_confirm)
|
||||
btn_confirm.pack(side="left", padx=10)
|
||||
btn_cancel = ctk.CTkButton(button_frame, text="取消请求", font=("Microsoft YaHei", 12), command=on_cancel)
|
||||
btn_cancel.pack(side="left", padx=10)
|
||||
# 若用户直接关闭弹窗,则调用 on_cancel 处理
|
||||
dialog.protocol("WM_DELETE_WINDOW", on_cancel)
|
||||
dialog.grab_set()
|
||||
self.master.after(0, create_dialog)
|
||||
event.wait() # 等待用户操作完成
|
||||
edited_prompt = result["prompt"]
|
||||
if edited_prompt is None:
|
||||
self.safe_log("❌ 用户取消了草稿生成请求。")
|
||||
return
|
||||
|
||||
self.safe_log("开始生成章节草稿...")
|
||||
from novel_generator.chapter import generate_chapter_draft
|
||||
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,
|
||||
timeout=timeout_val,
|
||||
custom_prompt_text=edited_prompt # 使用用户编辑后的提示词
|
||||
)
|
||||
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 finalize_chapter_ui(self):
|
||||
filepath = self.filepath_var.get().strip()
|
||||
if not filepath:
|
||||
messagebox.showwarning("警告", "请先配置保存文件路径。")
|
||||
return
|
||||
|
||||
def task():
|
||||
if not messagebox.askyesno("确认", "确定要定稿当前章节吗?"):
|
||||
self.enable_button_safe(self.btn_finalize_chapter)
|
||||
return
|
||||
|
||||
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()
|
||||
timeout_val = self.safe_get_int(self.timeout_var, 600)
|
||||
|
||||
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()
|
||||
|
||||
if len(edited_text) < 0.7 * word_number:
|
||||
ask = messagebox.askyesno("字数不足", f"当前章节字数 ({len(edited_text)}) 低于目标字数({word_number})的70%,是否要尝试扩写?")
|
||||
if ask:
|
||||
self.safe_log("正在扩写章节内容...")
|
||||
enriched = enrich_chapter_text(
|
||||
chapter_text=edited_text,
|
||||
word_number=word_number,
|
||||
api_key=api_key,
|
||||
base_url=base_url,
|
||||
model_name=model_name,
|
||||
temperature=temperature,
|
||||
interface_format=interface_format,
|
||||
max_tokens=max_tokens,
|
||||
timeout=timeout_val
|
||||
)
|
||||
edited_text = enriched
|
||||
self.master.after(0, lambda: self.chapter_result.delete("0.0", "end"))
|
||||
self.master.after(0, lambda: self.chapter_result.insert("0.0", edited_text))
|
||||
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,
|
||||
timeout=timeout_val
|
||||
)
|
||||
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()
|
||||
interface_format = self.interface_format_var.get()
|
||||
max_tokens = self.max_tokens_var.get()
|
||||
timeout = self.timeout_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,
|
||||
interface_format=interface_format,
|
||||
max_tokens=max_tokens,
|
||||
timeout=timeout,
|
||||
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 = tk.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()
|
||||
|
||||
# 尝试不同编码读取文件
|
||||
content = None
|
||||
encodings = ['utf-8', 'gbk', 'gb2312', 'ansi']
|
||||
for encoding in encodings:
|
||||
try:
|
||||
with open(selected_file, 'r', encoding=encoding) as f:
|
||||
content = f.read()
|
||||
break
|
||||
except UnicodeDecodeError:
|
||||
continue
|
||||
except Exception as e:
|
||||
self.safe_log(f"读取文件时发生错误: {str(e)}")
|
||||
raise
|
||||
|
||||
if content is None:
|
||||
raise Exception("无法以任何已知编码格式读取文件")
|
||||
|
||||
# 创建临时UTF-8文件
|
||||
import tempfile
|
||||
import os
|
||||
with tempfile.NamedTemporaryFile(mode='w', encoding='utf-8', delete=False, suffix='.txt') as temp:
|
||||
temp.write(content)
|
||||
temp_path = temp.name
|
||||
|
||||
try:
|
||||
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=temp_path,
|
||||
filepath=self.filepath_var.get().strip()
|
||||
)
|
||||
self.safe_log("✅ 知识库文件导入完成。")
|
||||
finally:
|
||||
# 清理临时文件
|
||||
try:
|
||||
os.unlink(temp_path)
|
||||
except:
|
||||
pass
|
||||
|
||||
except Exception:
|
||||
self.handle_exception("导入知识库时出错")
|
||||
finally:
|
||||
self.enable_button_safe(self.btn_import_knowledge)
|
||||
|
||||
try:
|
||||
thread = threading.Thread(target=task, daemon=True)
|
||||
thread.start()
|
||||
except Exception as e:
|
||||
self.enable_button_safe(self.btn_import_knowledge)
|
||||
messagebox.showerror("错误", f"线程启动失败: {str(e)}")
|
||||
|
||||
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")
|
||||
@@ -0,0 +1,7 @@
|
||||
# ui/helpers.py
|
||||
# -*- coding: utf-8 -*-
|
||||
import logging
|
||||
import traceback
|
||||
|
||||
def log_error(message: str):
|
||||
logging.error(f"{message}\n{traceback.format_exc()}")
|
||||
+113
@@ -0,0 +1,113 @@
|
||||
# ui/main_tab.py
|
||||
# -*- coding: utf-8 -*-
|
||||
import customtkinter as ctk
|
||||
from tkinter import messagebox
|
||||
from ui.context_menu import TextWidgetContextMenu
|
||||
|
||||
def build_main_tab(self):
|
||||
"""
|
||||
主Tab包含左侧的"本章内容"编辑框和输出日志,以及右侧的主要操作和参数设置区
|
||||
"""
|
||||
self.main_tab = self.tabview.add("Main Functions")
|
||||
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)
|
||||
|
||||
build_left_layout(self)
|
||||
build_right_layout(self)
|
||||
|
||||
def build_left_layout(self):
|
||||
"""
|
||||
左侧区域:本章内容(可编辑) + Step流程按钮 + 输出日志(只读)
|
||||
"""
|
||||
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)
|
||||
|
||||
self.chapter_label = ctk.CTkLabel(self.left_frame, text="本章内容(可编辑) 字数:0", font=("Microsoft YaHei", 12))
|
||||
self.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))
|
||||
TextWidgetContextMenu(self.chapter_result)
|
||||
self.chapter_result.grid(row=1, column=0, sticky="nsew", padx=5, pady=(0, 5))
|
||||
|
||||
|
||||
|
||||
def update_word_count(event=None):
|
||||
text = self.chapter_result.get("0.0", "end")
|
||||
count = len(text) - 1 # 减去最后一个换行符
|
||||
self.chapter_label.configure(text=f"本章内容(可编辑) 字数:{count}")
|
||||
|
||||
self.chapter_result.bind("<KeyRelease>", update_word_count)
|
||||
self.chapter_result.bind("<ButtonRelease>", update_word_count)
|
||||
|
||||
# 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))
|
||||
TextWidgetContextMenu(self.log_text)
|
||||
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):
|
||||
"""
|
||||
右侧区域:配置区(tabview) + 小说主参数 + 可选功能按钮
|
||||
"""
|
||||
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)
|
||||
|
||||
# 配置区(AI/Embedding)
|
||||
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)
|
||||
# 其余部分将在 config_tab.py 与 novel_params_tab.py 中构建
|
||||
@@ -0,0 +1,367 @@
|
||||
# ui/main_window.py
|
||||
# -*- coding: utf-8 -*-
|
||||
import os
|
||||
import threading
|
||||
import logging
|
||||
import traceback
|
||||
import customtkinter as ctk
|
||||
import tkinter as tk
|
||||
from tkinter import filedialog, messagebox
|
||||
from .role_library import RoleLibrary
|
||||
from llm_adapters import create_llm_adapter
|
||||
|
||||
from config_manager import load_config, save_config, test_llm_config, test_embedding_config
|
||||
from utils import read_file, save_string_to_txt, clear_file_content
|
||||
from tooltips import tooltips
|
||||
|
||||
from ui.context_menu import TextWidgetContextMenu
|
||||
from ui.main_tab import build_main_tab, build_left_layout, build_right_layout
|
||||
from ui.config_tab import build_config_tabview, load_config_btn, save_config_btn
|
||||
from ui.novel_params_tab import build_novel_params_area, build_optional_buttons_area
|
||||
from ui.generation_handlers import (
|
||||
generate_novel_architecture_ui,
|
||||
generate_chapter_blueprint_ui,
|
||||
generate_chapter_draft_ui,
|
||||
finalize_chapter_ui,
|
||||
do_consistency_check,
|
||||
import_knowledge_handler,
|
||||
clear_vectorstore_handler,
|
||||
show_plot_arcs_ui
|
||||
)
|
||||
from ui.setting_tab import build_setting_tab, load_novel_architecture, save_novel_architecture
|
||||
from ui.directory_tab import build_directory_tab, load_chapter_blueprint, save_chapter_blueprint
|
||||
from ui.character_tab import build_character_tab, load_character_state, save_character_state
|
||||
from ui.summary_tab import build_summary_tab, load_global_summary, save_global_summary
|
||||
from ui.chapters_tab import build_chapters_tab, refresh_chapters_list, on_chapter_selected, load_chapter_content, save_current_chapter, prev_chapter, next_chapter
|
||||
|
||||
class NovelGeneratorGUI:
|
||||
"""
|
||||
小说生成器的主GUI类,包含所有的界面布局、事件处理、与后端逻辑的交互等。
|
||||
"""
|
||||
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)
|
||||
|
||||
if self.loaded_config:
|
||||
last_llm = self.loaded_config.get("last_interface_format", "OpenAI")
|
||||
last_embedding = self.loaded_config.get("last_embedding_interface_format", "OpenAI")
|
||||
else:
|
||||
last_llm = "OpenAI"
|
||||
last_embedding = "OpenAI"
|
||||
|
||||
if self.loaded_config and "llm_configs" in self.loaded_config and last_llm in self.loaded_config["llm_configs"]:
|
||||
llm_conf = self.loaded_config["llm_configs"][last_llm]
|
||||
else:
|
||||
llm_conf = {
|
||||
"api_key": "",
|
||||
"base_url": "https://api.openai.com/v1",
|
||||
"model_name": "gpt-4o-mini",
|
||||
"temperature": 0.7,
|
||||
"max_tokens": 8192,
|
||||
"timeout": 600
|
||||
}
|
||||
|
||||
if self.loaded_config and "embedding_configs" in self.loaded_config and last_embedding in self.loaded_config["embedding_configs"]:
|
||||
emb_conf = self.loaded_config["embedding_configs"][last_embedding]
|
||||
else:
|
||||
emb_conf = {
|
||||
"api_key": "",
|
||||
"base_url": "https://api.openai.com/v1",
|
||||
"model_name": "text-embedding-ada-002",
|
||||
"retrieval_k": 4
|
||||
}
|
||||
|
||||
# -- LLM通用参数 --
|
||||
self.api_key_var = ctk.StringVar(value=llm_conf.get("api_key", ""))
|
||||
self.base_url_var = ctk.StringVar(value=llm_conf.get("base_url", "https://api.openai.com/v1"))
|
||||
self.interface_format_var = ctk.StringVar(value=last_llm)
|
||||
self.model_name_var = ctk.StringVar(value=llm_conf.get("model_name", "gpt-4o-mini"))
|
||||
self.temperature_var = ctk.DoubleVar(value=llm_conf.get("temperature", 0.7))
|
||||
self.max_tokens_var = ctk.IntVar(value=llm_conf.get("max_tokens", 8192))
|
||||
self.timeout_var = ctk.IntVar(value=llm_conf.get("timeout", 600))
|
||||
|
||||
# -- Embedding相关 --
|
||||
self.embedding_interface_format_var = ctk.StringVar(value=last_embedding)
|
||||
self.embedding_api_key_var = ctk.StringVar(value=emb_conf.get("api_key", ""))
|
||||
self.embedding_url_var = ctk.StringVar(value=emb_conf.get("base_url", "https://api.openai.com/v1"))
|
||||
self.embedding_model_name_var = ctk.StringVar(value=emb_conf.get("model_name", "text-embedding-ada-002"))
|
||||
self.embedding_retrieval_k_var = ctk.StringVar(value=str(emb_conf.get("retrieval_k", 4)))
|
||||
|
||||
# -- 小说参数相关 --
|
||||
if self.loaded_config and "other_params" in self.loaded_config:
|
||||
op = self.loaded_config["other_params"]
|
||||
self.topic_default = op.get("topic", "")
|
||||
self.genre_var = ctk.StringVar(value=op.get("genre", "玄幻"))
|
||||
self.num_chapters_var = ctk.StringVar(value=str(op.get("num_chapters", 10)))
|
||||
self.word_number_var = ctk.StringVar(value=str(op.get("word_number", 3000)))
|
||||
self.filepath_var = ctk.StringVar(value=op.get("filepath", ""))
|
||||
self.chapter_num_var = ctk.StringVar(value=str(op.get("chapter_num", "1")))
|
||||
self.characters_involved_var = ctk.StringVar(value=op.get("characters_involved", ""))
|
||||
self.key_items_var = ctk.StringVar(value=op.get("key_items", ""))
|
||||
self.scene_location_var = ctk.StringVar(value=op.get("scene_location", ""))
|
||||
self.time_constraint_var = ctk.StringVar(value=op.get("time_constraint", ""))
|
||||
self.user_guidance_default = op.get("user_guidance", "")
|
||||
else:
|
||||
self.topic_default = ""
|
||||
self.genre_var = ctk.StringVar(value="玄幻")
|
||||
self.num_chapters_var = ctk.StringVar(value="10")
|
||||
self.word_number_var = ctk.StringVar(value="3000")
|
||||
self.filepath_var = ctk.StringVar(value="")
|
||||
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="")
|
||||
self.user_guidance_default = ""
|
||||
|
||||
# --------------- 整体Tab布局 ---------------
|
||||
self.tabview = ctk.CTkTabview(self.master)
|
||||
self.tabview.pack(fill="both", expand=True)
|
||||
|
||||
# 创建各个标签页
|
||||
build_main_tab(self)
|
||||
build_config_tabview(self)
|
||||
build_novel_params_area(self, start_row=1)
|
||||
build_optional_buttons_area(self, start_row=2)
|
||||
build_setting_tab(self)
|
||||
build_directory_tab(self)
|
||||
build_character_tab(self)
|
||||
build_summary_tab(self)
|
||||
build_chapters_tab(self)
|
||||
|
||||
# ----------------- 通用辅助函数 -----------------
|
||||
def show_tooltip(self, key: str):
|
||||
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
|
||||
|
||||
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)
|
||||
|
||||
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")
|
||||
|
||||
def test_llm_config(self):
|
||||
"""
|
||||
测试当前的LLM配置是否可用
|
||||
"""
|
||||
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()
|
||||
timeout = self.timeout_var.get()
|
||||
|
||||
test_llm_config(
|
||||
interface_format=interface_format,
|
||||
api_key=api_key,
|
||||
base_url=base_url,
|
||||
model_name=model_name,
|
||||
temperature=temperature,
|
||||
max_tokens=max_tokens,
|
||||
timeout=timeout,
|
||||
log_func=self.safe_log,
|
||||
handle_exception_func=self.handle_exception
|
||||
)
|
||||
|
||||
def test_embedding_config(self):
|
||||
"""
|
||||
测试当前的Embedding配置是否可用
|
||||
"""
|
||||
api_key = self.embedding_api_key_var.get().strip()
|
||||
base_url = self.embedding_url_var.get().strip()
|
||||
interface_format = self.embedding_interface_format_var.get().strip()
|
||||
model_name = self.embedding_model_name_var.get().strip()
|
||||
|
||||
test_embedding_config(
|
||||
api_key=api_key,
|
||||
base_url=base_url,
|
||||
interface_format=interface_format,
|
||||
model_name=model_name,
|
||||
log_func=self.safe_log,
|
||||
handle_exception_func=self.handle_exception
|
||||
)
|
||||
|
||||
def browse_folder(self):
|
||||
selected_dir = filedialog.askdirectory()
|
||||
if selected_dir:
|
||||
self.filepath_var.set(selected_dir)
|
||||
|
||||
def show_character_import_window(self):
|
||||
"""显示角色导入窗口"""
|
||||
import_window = ctk.CTkToplevel(self.master)
|
||||
import_window.title("导入角色信息")
|
||||
import_window.geometry("600x500")
|
||||
import_window.transient(self.master) # 设置为父窗口的临时窗口
|
||||
import_window.grab_set() # 保持窗口在顶层
|
||||
|
||||
# 主容器
|
||||
main_frame = ctk.CTkFrame(import_window)
|
||||
main_frame.pack(fill="both", expand=True, padx=10, pady=10)
|
||||
|
||||
# 滚动容器
|
||||
scroll_frame = ctk.CTkScrollableFrame(main_frame)
|
||||
scroll_frame.pack(fill="both", expand=True, padx=5, pady=5)
|
||||
|
||||
# 获取角色库路径
|
||||
role_lib_path = os.path.join(self.filepath_var.get().strip(), "角色库")
|
||||
self.selected_roles = [] # 存储选中的角色名称
|
||||
|
||||
# 动态加载角色分类
|
||||
if os.path.exists(role_lib_path):
|
||||
# 配置网格布局参数
|
||||
scroll_frame.columnconfigure(0, weight=1)
|
||||
max_roles_per_row = 4
|
||||
current_row = 0
|
||||
|
||||
for category in os.listdir(role_lib_path):
|
||||
category_path = os.path.join(role_lib_path, category)
|
||||
if os.path.isdir(category_path):
|
||||
# 创建分类容器
|
||||
category_frame = ctk.CTkFrame(scroll_frame)
|
||||
category_frame.grid(row=current_row, column=0, sticky="w", pady=(10,5), padx=5)
|
||||
|
||||
# 添加分类标签
|
||||
category_label = ctk.CTkLabel(category_frame, text=f"【{category}】",
|
||||
font=("Microsoft YaHei", 12, "bold"))
|
||||
category_label.grid(row=0, column=0, padx=(0,10), sticky="w")
|
||||
|
||||
# 初始化角色排列参数
|
||||
role_count = 0
|
||||
row_num = 0
|
||||
col_num = 1 # 从第1列开始(第0列是分类标签)
|
||||
|
||||
# 添加角色复选框
|
||||
for role_file in os.listdir(category_path):
|
||||
if role_file.endswith(".txt"):
|
||||
role_name = os.path.splitext(role_file)[0]
|
||||
if not any(name == role_name for _, name in self.selected_roles):
|
||||
chk = ctk.CTkCheckBox(category_frame, text=role_name)
|
||||
chk.grid(row=row_num, column=col_num, padx=5, pady=2, sticky="w")
|
||||
self.selected_roles.append((chk, role_name))
|
||||
|
||||
# 更新行列位置
|
||||
role_count += 1
|
||||
col_num += 1
|
||||
if col_num > max_roles_per_row:
|
||||
col_num = 1
|
||||
row_num += 1
|
||||
|
||||
# 如果没有角色,调整分类标签占满整行
|
||||
if role_count == 0:
|
||||
category_label.grid(columnspan=max_roles_per_row+1, sticky="w")
|
||||
|
||||
# 更新主布局的行号
|
||||
current_row += 1
|
||||
|
||||
# 添加分隔线
|
||||
separator = ctk.CTkFrame(scroll_frame, height=1, fg_color="gray")
|
||||
separator.grid(row=current_row, column=0, sticky="ew", pady=5)
|
||||
current_row += 1
|
||||
|
||||
# 底部按钮框架
|
||||
btn_frame = ctk.CTkFrame(main_frame)
|
||||
btn_frame.pack(fill="x", pady=10)
|
||||
|
||||
# 选择按钮
|
||||
def confirm_selection():
|
||||
selected = [name for chk, name in self.selected_roles if chk.get() == 1]
|
||||
self.char_inv_text.delete("0.0", "end")
|
||||
self.char_inv_text.insert("0.0", ", ".join(selected))
|
||||
import_window.destroy()
|
||||
|
||||
btn_confirm = ctk.CTkButton(btn_frame, text="选择", command=confirm_selection)
|
||||
btn_confirm.pack(side="left", padx=20)
|
||||
|
||||
# 取消按钮
|
||||
btn_cancel = ctk.CTkButton(btn_frame, text="取消", command=import_window.destroy)
|
||||
btn_cancel.pack(side="right", padx=20)
|
||||
|
||||
def show_role_library(self):
|
||||
save_path = self.filepath_var.get().strip()
|
||||
if not save_path:
|
||||
messagebox.showwarning("警告", "请先设置保存路径")
|
||||
return
|
||||
|
||||
# 初始化LLM适配器
|
||||
llm_adapter = create_llm_adapter(
|
||||
interface_format=self.interface_format_var.get(),
|
||||
base_url=self.base_url_var.get(),
|
||||
model_name=self.model_name_var.get(),
|
||||
api_key=self.api_key_var.get(),
|
||||
temperature=self.temperature_var.get(),
|
||||
max_tokens=self.max_tokens_var.get(),
|
||||
timeout=self.timeout_var.get()
|
||||
)
|
||||
|
||||
# 传递LLM适配器实例到角色库
|
||||
if hasattr(self, '_role_lib'):
|
||||
if self._role_lib.window and self._role_lib.window.winfo_exists():
|
||||
self._role_lib.window.destroy()
|
||||
|
||||
self._role_lib = RoleLibrary(self.master, save_path, llm_adapter) # 新增参数
|
||||
|
||||
# ----------------- 将导入的各模块函数直接赋给类方法 -----------------
|
||||
generate_novel_architecture_ui = generate_novel_architecture_ui
|
||||
generate_chapter_blueprint_ui = generate_chapter_blueprint_ui
|
||||
generate_chapter_draft_ui = generate_chapter_draft_ui
|
||||
finalize_chapter_ui = finalize_chapter_ui
|
||||
do_consistency_check = do_consistency_check
|
||||
import_knowledge_handler = import_knowledge_handler
|
||||
clear_vectorstore_handler = clear_vectorstore_handler
|
||||
show_plot_arcs_ui = show_plot_arcs_ui
|
||||
load_config_btn = load_config_btn
|
||||
save_config_btn = save_config_btn
|
||||
load_novel_architecture = load_novel_architecture
|
||||
save_novel_architecture = save_novel_architecture
|
||||
load_chapter_blueprint = load_chapter_blueprint
|
||||
save_chapter_blueprint = save_chapter_blueprint
|
||||
load_character_state = load_character_state
|
||||
save_character_state = save_character_state
|
||||
load_global_summary = load_global_summary
|
||||
save_global_summary = save_global_summary
|
||||
refresh_chapters_list = refresh_chapters_list
|
||||
on_chapter_selected = on_chapter_selected
|
||||
save_current_chapter = save_current_chapter
|
||||
prev_chapter = prev_chapter
|
||||
next_chapter = next_chapter
|
||||
test_llm_config = test_llm_config
|
||||
test_embedding_config = test_embedding_config
|
||||
browse_folder = browse_folder
|
||||
@@ -0,0 +1,146 @@
|
||||
# ui/novel_params_tab.py
|
||||
# -*- coding: utf-8 -*-
|
||||
import customtkinter as ctk
|
||||
from tkinter import filedialog, messagebox
|
||||
from ui.context_menu import TextWidgetContextMenu
|
||||
from tooltips import tooltips
|
||||
|
||||
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)
|
||||
create_label_with_help_for_novel_params(self, 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))
|
||||
TextWidgetContextMenu(self.topic_text)
|
||||
self.topic_text.grid(row=0, column=1, padx=5, pady=5, sticky="nsew")
|
||||
if hasattr(self, 'topic_default') and self.topic_default:
|
||||
self.topic_text.insert("0.0", self.topic_default)
|
||||
|
||||
# 2) 类型(Genre)
|
||||
create_label_with_help_for_novel_params(self, 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
|
||||
create_label_with_help_for_novel_params(self, parent=self.params_frame, label_text="章节数 & 每章字数:", tooltip_key="num_chapters", row=row_for_chapter_and_word, column=0, font=("Microsoft YaHei", 12))
|
||||
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)
|
||||
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
|
||||
create_label_with_help_for_novel_params(self, 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
|
||||
create_label_with_help_for_novel_params(self, 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
|
||||
create_label_with_help_for_novel_params(self, 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))
|
||||
TextWidgetContextMenu(self.user_guide_text)
|
||||
self.user_guide_text.grid(row=row_user_guide, column=1, padx=5, pady=5, sticky="nsew")
|
||||
if hasattr(self, 'user_guidance_default') and self.user_guidance_default:
|
||||
self.user_guide_text.insert("0.0", self.user_guidance_default)
|
||||
|
||||
# 7) 可选元素:核心人物/关键道具/空间坐标/时间压力
|
||||
row_idx = 6
|
||||
create_label_with_help_for_novel_params(self, parent=self.params_frame, label_text="核心人物:", tooltip_key="characters_involved", row=row_idx, column=0, font=("Microsoft YaHei", 12))
|
||||
|
||||
# 核心人物输入框+按钮容器
|
||||
char_inv_frame = ctk.CTkFrame(self.params_frame)
|
||||
char_inv_frame.grid(row=row_idx, column=1, padx=5, pady=5, sticky="nsew")
|
||||
char_inv_frame.columnconfigure(0, weight=1)
|
||||
char_inv_frame.rowconfigure(0, weight=1)
|
||||
|
||||
# 三行文本输入框
|
||||
self.char_inv_text = ctk.CTkTextbox(char_inv_frame, height=60, wrap="word", font=("Microsoft YaHei", 12))
|
||||
self.char_inv_text.grid(row=0, column=0, padx=(0,5), pady=5, sticky="nsew")
|
||||
if hasattr(self, 'characters_involved_var'):
|
||||
self.char_inv_text.insert("0.0", self.characters_involved_var.get())
|
||||
|
||||
# 导入按钮
|
||||
import_btn = ctk.CTkButton(char_inv_frame, text="导入", width=60,
|
||||
command=self.show_character_import_window,
|
||||
font=("Microsoft YaHei", 12))
|
||||
import_btn.grid(row=0, column=1, padx=(0,5), pady=5, sticky="e")
|
||||
row_idx += 1
|
||||
create_label_with_help_for_novel_params(self, 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
|
||||
create_label_with_help_for_novel_params(self, 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
|
||||
create_label_with_help_for_novel_params(self, 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, 4), weight=1)
|
||||
|
||||
self.btn_check_consistency = ctk.CTkButton(
|
||||
self.optional_btn_frame, text="一致性审校", command=self.do_consistency_check,
|
||||
font=("Microsoft YaHei", 12), width=100 # 固定宽度
|
||||
)
|
||||
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), width=100
|
||||
)
|
||||
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), width=100
|
||||
)
|
||||
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), width=100
|
||||
)
|
||||
self.plot_arcs_btn.grid(row=0, column=3, padx=5, pady=5, sticky="ew")
|
||||
|
||||
# 新增角色库按钮
|
||||
self.role_library_btn = ctk.CTkButton(
|
||||
self.optional_btn_frame, text="角色库", command=self.show_role_library,
|
||||
font=("Microsoft YaHei", 12), width=100
|
||||
)
|
||||
self.role_library_btn.grid(row=0, column=4, padx=5, pady=5, sticky="ew")
|
||||
|
||||
def create_label_with_help_for_novel_params(self, parent, label_text, tooltip_key, row, column, font=None, sticky="e", padx=5, pady=5):
|
||||
frame = ctk.CTkFrame(parent)
|
||||
frame.grid(row=row, column=column, padx=padx, pady=pady, sticky=sticky)
|
||||
frame.columnconfigure(0, weight=0)
|
||||
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: messagebox.showinfo("参数说明", tooltips.get(tooltip_key, "暂无说明")))
|
||||
btn.pack(side="left", padx=3)
|
||||
return frame
|
||||
+1528
File diff suppressed because it is too large
Load Diff
@@ -0,0 +1,56 @@
|
||||
# ui/setting_tab.py
|
||||
# -*- coding: utf-8 -*-
|
||||
import os
|
||||
import customtkinter as ctk
|
||||
from tkinter import messagebox
|
||||
from utils import read_file, save_string_to_txt, clear_file_content
|
||||
from ui.context_menu import TextWidgetContextMenu
|
||||
|
||||
def build_setting_tab(self):
|
||||
self.setting_tab = self.tabview.add("Novel Architecture")
|
||||
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")
|
||||
|
||||
self.setting_word_count_label = ctk.CTkLabel(self.setting_tab, text="字数:0", font=("Microsoft YaHei", 12))
|
||||
self.setting_word_count_label.grid(row=0, column=1, 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=2, padx=5, pady=5, sticky="e")
|
||||
|
||||
self.setting_text = ctk.CTkTextbox(self.setting_tab, wrap="word", font=("Microsoft YaHei", 12))
|
||||
TextWidgetContextMenu(self.setting_text)
|
||||
self.setting_text.grid(row=1, column=0, sticky="nsew", padx=5, pady=5, columnspan=3)
|
||||
|
||||
def update_word_count(event=None):
|
||||
text = self.setting_text.get("0.0", "end")
|
||||
count = len(text) - 1
|
||||
self.setting_word_count_label.configure(text=f"字数:{count}")
|
||||
|
||||
self.setting_text.bind("<KeyRelease>", update_word_count)
|
||||
self.setting_text.bind("<ButtonRelease>", update_word_count)
|
||||
|
||||
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 的修改。")
|
||||
@@ -0,0 +1,57 @@
|
||||
# ui/summary_tab.py
|
||||
# -*- coding: utf-8 -*-
|
||||
import os
|
||||
import customtkinter as ctk
|
||||
from tkinter import messagebox
|
||||
from utils import read_file, save_string_to_txt, clear_file_content
|
||||
from ui.context_menu import TextWidgetContextMenu
|
||||
|
||||
def build_summary_tab(self):
|
||||
self.summary_tab = self.tabview.add("Global Summary")
|
||||
self.summary_tab.rowconfigure(0, weight=0)
|
||||
self.summary_tab.rowconfigure(1, weight=1)
|
||||
self.summary_tab.columnconfigure(0, weight=1)
|
||||
self.summary_tab.columnconfigure(1, weight=0)
|
||||
self.summary_tab.columnconfigure(2, weight=0)
|
||||
|
||||
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")
|
||||
|
||||
self.word_count_label = ctk.CTkLabel(self.summary_tab, text="字数:0", font=("Microsoft YaHei", 12))
|
||||
self.word_count_label.grid(row=0, column=1, 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=2, padx=5, pady=5, sticky="e")
|
||||
|
||||
self.summary_text = ctk.CTkTextbox(self.summary_tab, wrap="word", font=("Microsoft YaHei", 12))
|
||||
TextWidgetContextMenu(self.summary_text)
|
||||
self.summary_text.grid(row=1, column=0, sticky="nsew", padx=5, pady=5, columnspan=3)
|
||||
|
||||
def update_word_count(event=None):
|
||||
text = self.summary_text.get("0.0", "end")
|
||||
count = len(text) - 1
|
||||
self.word_count_label.configure(text=f"字数:{count}")
|
||||
|
||||
self.summary_text.bind("<KeyRelease>", update_word_count)
|
||||
self.summary_text.bind("<ButtonRelease>", update_word_count)
|
||||
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 的修改。")
|
||||
Reference in New Issue
Block a user