记录一下，便于待会核对更改
This commit is contained in:
YILING0013
@@ -1,12 +1,13 @@
|
||||
# ui.py
|
||||
# -*- coding: utf-8 -*-
|
||||
|
||||
import os
|
||||
import tkinter as tk
|
||||
from tkinter import ttk, filedialog, scrolledtext, messagebox
|
||||
import threading
|
||||
import customtkinter as ctk
|
||||
from tkinter import filedialog, messagebox
|
||||
|
||||
from config_manager import load_config, save_config
|
||||
from utils import read_file
|
||||
from utils import read_file, save_string_to_txt
|
||||
from novel_generator import (
|
||||
Novel_novel_directory_generate,
|
||||
generate_chapter_draft,
|
||||
@@ -18,181 +19,456 @@ from novel_generator import (
|
||||
)
|
||||
from consistency_checker import check_consistency
|
||||
|
||||
# 设置全局主题和颜色
|
||||
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")
|
||||
self.master.title("Novel Generator GUI (CustomTkinter)")
|
||||
self.master.iconbitmap("icon.ico")
|
||||
# 窗口最大化
|
||||
self.master.state("zoomed")
|
||||
# 配置窗口大小
|
||||
self.master.geometry("1344x896")
|
||||
|
||||
# 配置持久化
|
||||
self.config_file = "config.json"
|
||||
self.loaded_config = load_config(self.config_file)
|
||||
|
||||
# 使用 PanedWindow 实现左右布局
|
||||
main_pane = ttk.PanedWindow(master, orient="horizontal")
|
||||
main_pane.pack(fill="both", expand=True)
|
||||
# ========== 主要的属性变量 ==========
|
||||
# 右侧参数区 - 各种输入
|
||||
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.agicto.cn/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.embedding_url_var = ctk.StringVar(value=self.loaded_config.get("embedding_url", "")) # 新增：可选Embedding模型URL
|
||||
|
||||
self.temperature_var = ctk.DoubleVar(value=self.loaded_config.get("temperature", 0.7))
|
||||
self.topic_default = self.loaded_config.get("topic", "")
|
||||
self.genre_var = ctk.StringVar(value=self.loaded_config.get("genre", "玄幻"))
|
||||
self.num_chapters_var = ctk.IntVar(value=self.loaded_config.get("num_chapters", 10))
|
||||
self.word_number_var = ctk.IntVar(value=self.loaded_config.get("word_number", 3000))
|
||||
self.filepath_var = ctk.StringVar(value=self.loaded_config.get("filepath", ""))
|
||||
|
||||
# 左侧：显示区(上下分区)
|
||||
self.left_frame = ttk.Frame(main_pane)
|
||||
main_pane.add(self.left_frame, weight=3)
|
||||
self.chapter_num_var = ctk.IntVar(value=1)
|
||||
|
||||
# 右侧：参数输入区
|
||||
self.right_frame = ttk.Frame(main_pane, padding="10 10 10 10")
|
||||
main_pane.add(self.right_frame, weight=1)
|
||||
# ========== 主容器使用 TabView ==========
|
||||
self.tabview = ctk.CTkTabview(self.master, width=1200, height=800)
|
||||
self.tabview.pack(fill="both", expand=True)
|
||||
|
||||
# 左侧布局：日志区 + 章节内容
|
||||
# 创建各个Tab
|
||||
self.main_tab = self.tabview.add("主功能")
|
||||
self.setting_tab = self.tabview.add("Novel Settings")
|
||||
self.directory_tab = self.tabview.add("Novel Directory")
|
||||
self.character_tab = self.tabview.add("Character State")
|
||||
self.summary_tab = self.tabview.add("Global Summary")
|
||||
|
||||
# 构建主功能Tab的布局
|
||||
self.build_main_tab()
|
||||
|
||||
# 构建“Novel Settings”编辑Tab
|
||||
self.build_setting_tab()
|
||||
|
||||
# 构建“Novel Directory”编辑Tab
|
||||
self.build_directory_tab()
|
||||
|
||||
# 角色状态 Tab
|
||||
self.build_character_tab()
|
||||
|
||||
# 全局摘要 Tab
|
||||
self.build_summary_tab()
|
||||
|
||||
# ------------------ 主功能 Tab ------------------
|
||||
def build_main_tab(self):
|
||||
"""
|
||||
主Tab: 左侧显示日志 / 本章内容, 右侧显示主要功能操作区
|
||||
"""
|
||||
self.main_tab.rowconfigure(0, weight=1)
|
||||
self.main_tab.columnconfigure(0, weight=1)
|
||||
self.main_tab.columnconfigure(1, weight=0)
|
||||
|
||||
# 左侧Frame
|
||||
self.left_frame = ctk.CTkFrame(self.main_tab)
|
||||
self.left_frame.grid(row=0, column=0, sticky="nsew", padx=2, pady=2)
|
||||
# 右侧Frame
|
||||
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.rowconfigure(0, weight=1)
|
||||
self.left_frame.rowconfigure(1, weight=1)
|
||||
self.left_frame.columnconfigure(0, weight=1)
|
||||
"""
|
||||
左侧包含两个区域：
|
||||
1. 输出日志（下半部分）
|
||||
2. 本章内容（上半部分）
|
||||
"""
|
||||
self.left_frame.grid_rowconfigure(0, weight=3)
|
||||
self.left_frame.grid_rowconfigure(1, weight=1)
|
||||
self.left_frame.grid_columnconfigure(0, weight=1)
|
||||
|
||||
# 日志区
|
||||
log_frame = ttk.LabelFrame(self.left_frame, text="输出日志")
|
||||
log_frame.grid(row=0, column=0, sticky="nsew", padx=5, pady=5)
|
||||
log_frame.rowconfigure(0, weight=1)
|
||||
log_frame.columnconfigure(0, weight=1)
|
||||
# 本章内容
|
||||
chapter_label = ctk.CTkLabel(self.left_frame, text="本章内容", font=("Microsoft YaHei", 14))
|
||||
chapter_label.grid(row=0, column=0, padx=5, pady=(5, 0), sticky="w")
|
||||
|
||||
self.log_text = scrolledtext.ScrolledText(log_frame, width=80, height=10)
|
||||
self.log_text.grid(row=0, column=0, sticky="nsew")
|
||||
self.chapter_result = ctk.CTkTextbox(self.left_frame, wrap="word", font=("Microsoft YaHei", 14))
|
||||
self.chapter_result.grid(row=0, column=0, sticky="nsew", padx=5, pady=(0, 5))
|
||||
|
||||
# 章节内容区
|
||||
chapter_frame = ttk.LabelFrame(self.left_frame, text="本章内容")
|
||||
chapter_frame.grid(row=1, column=0, sticky="nsew", padx=5, pady=5)
|
||||
chapter_frame.rowconfigure(0, weight=1)
|
||||
chapter_frame.columnconfigure(0, weight=1)
|
||||
# 输出日志
|
||||
log_label = ctk.CTkLabel(self.left_frame, text="输出日志", font=("Microsoft YaHei", 14))
|
||||
log_label.grid(row=1, column=0, padx=5, pady=(5, 0), sticky="w")
|
||||
|
||||
self.chapter_result = scrolledtext.ScrolledText(chapter_frame, width=80, height=10, foreground="blue")
|
||||
self.chapter_result.grid(row=0, column=0, sticky="nsew")
|
||||
self.log_text = ctk.CTkTextbox(self.left_frame, wrap="word", font=("Microsoft YaHei", 12))
|
||||
self.log_text.grid(row=1, column=0, sticky="nsew", padx=5, pady=(0, 5))
|
||||
|
||||
def build_right_layout(self):
|
||||
# 行列配置
|
||||
for i in range(20):
|
||||
self.right_frame.rowconfigure(i, weight=0)
|
||||
self.right_frame.columnconfigure(1, weight=1)
|
||||
"""
|
||||
右侧用于显示一系列参数输入和功能按钮。
|
||||
"""
|
||||
# 配置网格
|
||||
for i in range(25):
|
||||
self.right_frame.grid_rowconfigure(i, weight=0)
|
||||
self.right_frame.grid_columnconfigure(0, weight=0)
|
||||
self.right_frame.grid_columnconfigure(1, weight=1)
|
||||
|
||||
# 1. API Key
|
||||
ttk.Label(self.right_frame, text="API Key:").grid(row=0, column=0, padx=5, pady=5, sticky="e")
|
||||
self.api_key_var = tk.StringVar(value=self.loaded_config.get("api_key", ""))
|
||||
ttk.Entry(self.right_frame, textvariable=self.api_key_var, width=32).grid(row=0, column=1, padx=5, pady=5, sticky="w")
|
||||
api_key_label = ctk.CTkLabel(self.right_frame, text="API Key:")
|
||||
api_key_label.grid(row=0, column=0, padx=5, pady=5, sticky="e")
|
||||
api_key_entry = ctk.CTkEntry(self.right_frame, textvariable=self.api_key_var)
|
||||
api_key_entry.grid(row=0, column=1, columnspan=2, padx=5, pady=5, sticky="nsew")
|
||||
|
||||
# 2. Base URL
|
||||
ttk.Label(self.right_frame, text="Base URL:").grid(row=1, column=0, padx=5, pady=5, sticky="e")
|
||||
self.base_url_var = tk.StringVar(value=self.loaded_config.get("base_url", "https://api.agicto.cn/v1"))
|
||||
ttk.Entry(self.right_frame, textvariable=self.base_url_var, width=32).grid(row=1, column=1, padx=5, pady=5, sticky="w")
|
||||
base_url_label = ctk.CTkLabel(self.right_frame, text="Base URL:")
|
||||
base_url_label.grid(row=1, column=0, padx=5, pady=5, sticky="e")
|
||||
base_url_entry = ctk.CTkEntry(self.right_frame, textvariable=self.base_url_var)
|
||||
base_url_entry.grid(row=1, column=1, columnspan=2, padx=5, pady=5, sticky="nsew")
|
||||
|
||||
# 2.1 接口格式 下拉菜单
|
||||
interface_label = ctk.CTkLabel(self.right_frame, text="接口格式:")
|
||||
interface_label.grid(row=2, column=0, padx=5, pady=5, sticky="e")
|
||||
interface_options = ["OpenAI", "Ollama", "ML Studio", "Local"]
|
||||
interface_dropdown = ctk.CTkOptionMenu(self.right_frame, values=interface_options, variable=self.interface_format_var)
|
||||
interface_dropdown.grid(row=2, column=1, columnspan=2, padx=5, pady=5, sticky="nsew")
|
||||
|
||||
# 3. Model Name
|
||||
ttk.Label(self.right_frame, text="Model Name:").grid(row=2, column=0, padx=5, pady=5, sticky="e")
|
||||
self.model_name_var = tk.StringVar(value=self.loaded_config.get("model_name", "gpt-4o-mini"))
|
||||
ttk.Entry(self.right_frame, textvariable=self.model_name_var, width=32).grid(row=2, column=1, padx=5, pady=5, sticky="w")
|
||||
model_name_label = ctk.CTkLabel(self.right_frame, text="Model Name:")
|
||||
model_name_label.grid(row=3, column=0, padx=5, pady=5, sticky="e")
|
||||
model_name_entry = ctk.CTkEntry(self.right_frame, textvariable=self.model_name_var)
|
||||
model_name_entry.grid(row=3, column=1, columnspan=2, padx=5, pady=5, sticky="nsew")
|
||||
|
||||
# 3.1 Embedding Model URL (可选)
|
||||
embedding_url_label = ctk.CTkLabel(self.right_frame, text="Embedding URL:")
|
||||
embedding_url_label.grid(row=4, column=0, padx=5, pady=5, sticky="e")
|
||||
embedding_url_entry = ctk.CTkEntry(self.right_frame, textvariable=self.embedding_url_var)
|
||||
embedding_url_entry.grid(row=4, column=1, columnspan=2, padx=5, pady=5, sticky="nsew")
|
||||
|
||||
# 4. Temperature
|
||||
ttk.Label(self.right_frame, text="Temperature:").grid(row=3, column=0, padx=5, pady=5, sticky="e")
|
||||
self.temperature_var = tk.DoubleVar(value=self.loaded_config.get("temperature", 0.7))
|
||||
self.temp_value_label = ttk.Label(self.right_frame, text=f"{self.temperature_var.get():.2f}")
|
||||
self.temp_value_label.grid(row=3, column=2, padx=5, pady=5, sticky="w")
|
||||
|
||||
temp_scale = ttk.Scale(self.right_frame, from_=0.0, to=1.0, orient=tk.HORIZONTAL, variable=self.temperature_var)
|
||||
temp_scale.grid(row=3, column=1, padx=5, pady=5, sticky="we")
|
||||
def update_temp_label(*args):
|
||||
self.temp_value_label.config(text=f"{self.temperature_var.get():.2f}")
|
||||
self.temperature_var.trace("w", update_temp_label)
|
||||
temp_label = ctk.CTkLabel(self.right_frame, text="Temperature:")
|
||||
temp_label.grid(row=5, column=0, padx=5, pady=5, sticky="e")
|
||||
|
||||
def update_temp_label(value):
|
||||
self.temp_value_label.configure(text=f"{float(value):.2f}")
|
||||
temp_scale = ctk.CTkSlider(self.right_frame, from_=0.0, to=1.0, number_of_steps=100,
|
||||
command=update_temp_label, variable=self.temperature_var)
|
||||
temp_scale.grid(row=5, column=1, padx=5, pady=5, sticky="we")
|
||||
|
||||
self.temp_value_label = ctk.CTkLabel(self.right_frame, text=f"{self.temperature_var.get():.2f}")
|
||||
self.temp_value_label.grid(row=5, column=2, padx=1, pady=1, sticky="w")
|
||||
|
||||
# 5. 主题(Topic) 多行输入
|
||||
ttk.Label(self.right_frame, text="主题(Topic):").grid(row=4, column=0, padx=5, pady=5, sticky="ne")
|
||||
self.topic_text = scrolledtext.ScrolledText(self.right_frame, width=32, height=4)
|
||||
self.topic_text.grid(row=4, column=1, padx=5, pady=5, sticky="w")
|
||||
topic_default = self.loaded_config.get("topic", "")
|
||||
if topic_default:
|
||||
self.topic_text.insert(tk.END, topic_default)
|
||||
topic_label = ctk.CTkLabel(self.right_frame, text="主题(Topic):", font=("Microsoft YaHei", 12))
|
||||
topic_label.grid(row=6, column=0, padx=5, pady=5, sticky="e")
|
||||
self.topic_text = ctk.CTkTextbox(self.right_frame, width=200, height=80, wrap="word", font=("Microsoft YaHei", 12))
|
||||
self.topic_text.grid(row=6, column=1, columnspan=2, padx=5, pady=5, sticky="nsew")
|
||||
if self.topic_default:
|
||||
self.topic_text.insert("0.0", self.topic_default)
|
||||
|
||||
# 6. 类型(Genre)
|
||||
ttk.Label(self.right_frame, text="类型(Genre):").grid(row=5, column=0, padx=5, pady=5, sticky="e")
|
||||
self.genre_var = tk.StringVar(value=self.loaded_config.get("genre", "玄幻"))
|
||||
ttk.Entry(self.right_frame, textvariable=self.genre_var, width=32).grid(row=5, column=1, padx=5, pady=5, sticky="w")
|
||||
genre_label = ctk.CTkLabel(self.right_frame, text="类型(Genre):", font=("Microsoft YaHei", 12))
|
||||
genre_label.grid(row=7, column=0, padx=5, pady=5, sticky="e")
|
||||
genre_entry = ctk.CTkEntry(self.right_frame, textvariable=self.genre_var, font=("Microsoft YaHei", 12))
|
||||
genre_entry.grid(row=7, column=1, columnspan=2, padx=5, pady=5, sticky="nsew")
|
||||
|
||||
# 7. 章节数
|
||||
ttk.Label(self.right_frame, text="章节数:").grid(row=6, column=0, padx=5, pady=5, sticky="e")
|
||||
self.num_chapters_var = tk.IntVar(value=self.loaded_config.get("num_chapters", 10))
|
||||
ttk.Entry(self.right_frame, textvariable=self.num_chapters_var, width=8).grid(row=6, column=1, padx=5, pady=5, sticky="w")
|
||||
num_chapters_label = ctk.CTkLabel(self.right_frame, text="章节数:", font=("Microsoft YaHei", 12))
|
||||
num_chapters_label.grid(row=8, column=0, padx=5, pady=5, sticky="e")
|
||||
num_chapters_entry = ctk.CTkEntry(self.right_frame, textvariable=self.num_chapters_var, width=80)
|
||||
num_chapters_entry.grid(row=8, column=1, padx=5, pady=5, sticky="w")
|
||||
|
||||
# 8. 每章字数
|
||||
ttk.Label(self.right_frame, text="每章字数:").grid(row=7, column=0, padx=5, pady=5, sticky="e")
|
||||
self.word_number_var = tk.IntVar(value=self.loaded_config.get("word_number", 3000))
|
||||
ttk.Entry(self.right_frame, textvariable=self.word_number_var, width=8).grid(row=7, column=1, padx=5, pady=5, sticky="w")
|
||||
word_number_label = ctk.CTkLabel(self.right_frame, text="每章字数:", font=("Microsoft YaHei", 12))
|
||||
word_number_label.grid(row=9, column=0, padx=5, pady=5, sticky="e")
|
||||
word_number_entry = ctk.CTkEntry(self.right_frame, textvariable=self.word_number_var, width=80)
|
||||
word_number_entry.grid(row=9, column=1, padx=5, pady=5, sticky="w")
|
||||
|
||||
# 9. 文件保存路径
|
||||
ttk.Label(self.right_frame, text="保存路径:").grid(row=8, column=0, padx=5, pady=5, sticky="e")
|
||||
self.filepath_var = tk.StringVar(value=self.loaded_config.get("filepath", ""))
|
||||
ttk.Entry(self.right_frame, textvariable=self.filepath_var, width=32).grid(row=8, column=1, padx=5, pady=5, sticky="w")
|
||||
ttk.Button(self.right_frame, text="浏览...", command=self.browse_folder).grid(row=8, column=2, padx=5, pady=5, sticky="w")
|
||||
filepath_label = ctk.CTkLabel(self.right_frame, text="保存路径:", font=("Microsoft YaHei", 12))
|
||||
filepath_label.grid(row=10, column=0, padx=5, pady=5, sticky="e")
|
||||
filepath_entry = ctk.CTkEntry(self.right_frame, textvariable=self.filepath_var)
|
||||
filepath_entry.grid(row=10, column=1, padx=5, pady=5, sticky="nsew")
|
||||
browse_btn = ctk.CTkButton(self.right_frame, text="浏览...", command=self.browse_folder, width=60, font=("Microsoft YaHei", 12))
|
||||
browse_btn.grid(row=10, column=2, padx=1, pady=1, sticky="w")
|
||||
|
||||
# 保存/加载配置按钮
|
||||
config_frame = ttk.Frame(self.right_frame)
|
||||
config_frame.grid(row=9, column=1, sticky="w")
|
||||
ttk.Button(config_frame, text="保存配置", command=self.save_config_btn).grid(row=0, column=0, padx=5)
|
||||
ttk.Button(config_frame, text="加载配置", command=self.load_config_btn).grid(row=0, column=1, padx=5)
|
||||
config_frame = ctk.CTkFrame(self.right_frame)
|
||||
config_frame.grid(row=11, column=1, columnspan=2, sticky="nsew")
|
||||
|
||||
save_config_btn = ctk.CTkButton(config_frame, text="保存配置", command=self.save_config_btn, font=("Microsoft YaHei", 12))
|
||||
save_config_btn.grid(row=0, column=0, padx=5, pady=5, sticky="nsew")
|
||||
|
||||
load_config_btn = ctk.CTkButton(config_frame, text="加载配置", command=self.load_config_btn, font=("Microsoft YaHei", 12))
|
||||
load_config_btn.grid(row=0, column=1, padx=5, pady=5, sticky="nsew")
|
||||
|
||||
# 10. 章节号
|
||||
ttk.Label(self.right_frame, text="章节号:").grid(row=10, column=0, sticky="e")
|
||||
self.chapter_num_var = tk.IntVar(value=1)
|
||||
ttk.Entry(self.right_frame, textvariable=self.chapter_num_var, width=6).grid(row=10, column=1, padx=5, pady=5, sticky="w")
|
||||
chapter_num_label = ctk.CTkLabel(self.right_frame, text="章节号:", font=("Microsoft YaHei", 12))
|
||||
chapter_num_label.grid(row=12, column=0, padx=5, pady=5, sticky="e")
|
||||
chapter_num_entry = ctk.CTkEntry(self.right_frame, textvariable=self.chapter_num_var, width=80)
|
||||
chapter_num_entry.grid(row=12, column=1, padx=5, pady=5, sticky="w")
|
||||
|
||||
# 11. “用户指导” 多行输入
|
||||
ttk.Label(self.right_frame, text="本章指导:").grid(row=11, column=0, padx=5, pady=5, sticky="ne")
|
||||
self.user_guide_text = scrolledtext.ScrolledText(self.right_frame, width=32, height=4)
|
||||
self.user_guide_text.grid(row=11, column=1, padx=5, pady=5, sticky="w")
|
||||
guide_label = ctk.CTkLabel(self.right_frame, text="本章指导:", font=("Microsoft YaHei", 12))
|
||||
guide_label.grid(row=13, column=0, padx=5, pady=5, sticky="ne")
|
||||
self.user_guide_text = ctk.CTkTextbox(self.right_frame, width=200, height=80, wrap="word", font=("Microsoft YaHei", 12))
|
||||
self.user_guide_text.grid(row=13, column=1, columnspan=2, padx=5, pady=5, sticky="nsew")
|
||||
|
||||
row_base = 12
|
||||
# ============ 功能按钮 ============
|
||||
|
||||
# (1) 生成设定 & 目录
|
||||
self.btn_generate_full = ttk.Button(self.right_frame, text="Step1. 生成设定 & 目录", command=self.generate_full_novel)
|
||||
self.btn_generate_full.grid(row=row_base, column=0, columnspan=2, padx=5, pady=5, sticky="ew")
|
||||
|
||||
# (2) 生成章节草稿
|
||||
self.btn_generate_chapter = ttk.Button(self.right_frame, text="Step2. 生成章节草稿", command=self.generate_chapter_draft_ui)
|
||||
self.btn_generate_chapter.grid(row=row_base+1, column=0, columnspan=2, padx=5, pady=5, sticky="ew")
|
||||
|
||||
# (3) 定稿当前章节
|
||||
self.btn_finalize_chapter = ttk.Button(self.right_frame, text="Step3. 定稿当前章节", command=self.finalize_chapter_ui)
|
||||
self.btn_finalize_chapter.grid(row=row_base+2, column=0, columnspan=2, padx=5, pady=5, sticky="ew")
|
||||
|
||||
# (4) 一致性审校
|
||||
self.btn_check_consistency = ttk.Button(self.right_frame, text="[可选]一致性审校", command=self.do_consistency_check)
|
||||
self.btn_check_consistency.grid(row=row_base+3, column=0, columnspan=2, padx=5, pady=5, sticky="ew")
|
||||
|
||||
# (5) 导入知识库文件
|
||||
self.btn_import_knowledge = ttk.Button(self.right_frame, text="[可选]导入知识库", command=self.import_knowledge_handler)
|
||||
self.btn_import_knowledge.grid(row=row_base+4, column=0, columnspan=2, padx=5, pady=5, sticky="ew")
|
||||
|
||||
# (6) 清空向量库
|
||||
self.btn_clear_vectorstore = ttk.Button(self.right_frame, text="清空向量库", command=self.clear_vectorstore_handler)
|
||||
self.btn_clear_vectorstore.grid(row=row_base+5, column=0, columnspan=2, padx=5, pady=5, sticky="ew")
|
||||
|
||||
# (7) 查看剧情要点
|
||||
ttk.Button(self.right_frame, text="[查看] 剧情要点", command=self.show_plot_arcs_ui).grid(
|
||||
row=row_base+6, column=0, columnspan=2, padx=5, pady=5, sticky="ew"
|
||||
# 功能按钮区域
|
||||
row_base = 14
|
||||
self.btn_generate_full = ctk.CTkButton(
|
||||
self.right_frame, text="Step1. 生成设定 & 目录",
|
||||
command=self.generate_full_novel,
|
||||
font=("Microsoft YaHei", 12)
|
||||
)
|
||||
self.btn_generate_full.grid(row=row_base, column=0, columnspan=3, padx=5, pady=5, sticky="ew")
|
||||
|
||||
# -------------- 配置管理 --------------
|
||||
self.btn_generate_chapter = ctk.CTkButton(
|
||||
self.right_frame, text="Step2. 生成章节草稿",
|
||||
command=self.generate_chapter_draft_ui,
|
||||
font=("Microsoft YaHei", 12)
|
||||
)
|
||||
self.btn_generate_chapter.grid(row=row_base+1, column=0, columnspan=3, padx=5, pady=5, sticky="ew")
|
||||
|
||||
self.btn_finalize_chapter = ctk.CTkButton(
|
||||
self.right_frame, text="Step3. 定稿当前章节",
|
||||
command=self.finalize_chapter_ui,
|
||||
font=("Microsoft YaHei", 12)
|
||||
)
|
||||
self.btn_finalize_chapter.grid(row=row_base+2, column=0, columnspan=3, padx=5, pady=5, sticky="ew")
|
||||
|
||||
self.btn_check_consistency = ctk.CTkButton(
|
||||
self.right_frame, text="[可选]一致性审校",
|
||||
command=self.do_consistency_check,
|
||||
font=("Microsoft YaHei", 12)
|
||||
)
|
||||
self.btn_check_consistency.grid(row=row_base+3, column=0, columnspan=3, padx=5, pady=5, sticky="ew")
|
||||
|
||||
self.btn_import_knowledge = ctk.CTkButton(
|
||||
self.right_frame, text="[可选]导入知识库",
|
||||
command=self.import_knowledge_handler,
|
||||
font=("Microsoft YaHei", 12)
|
||||
)
|
||||
self.btn_import_knowledge.grid(row=row_base+4, column=0, columnspan=3, padx=5, pady=5, sticky="ew")
|
||||
|
||||
self.btn_clear_vectorstore = ctk.CTkButton(
|
||||
self.right_frame, text="清空向量库",
|
||||
fg_color="red", # 让按钮显眼一些
|
||||
command=self.clear_vectorstore_handler,
|
||||
font=("Microsoft YaHei", 12)
|
||||
)
|
||||
self.btn_clear_vectorstore.grid(row=row_base+5, column=0, columnspan=3, padx=5, pady=5, sticky="ew")
|
||||
|
||||
plot_arcs_btn = ctk.CTkButton(
|
||||
self.right_frame, text="[查看] 剧情要点",
|
||||
command=self.show_plot_arcs_ui,
|
||||
font=("Microsoft YaHei", 12)
|
||||
)
|
||||
plot_arcs_btn.grid(row=row_base+6, column=0, columnspan=3, padx=5, pady=5, sticky="ew")
|
||||
|
||||
# ------------------ Novel Settings Tab ------------------
|
||||
def build_setting_tab(self):
|
||||
"""
|
||||
可查看/编辑 Novel_setting.txt 并保存
|
||||
"""
|
||||
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_setting.txt", command=self.load_novel_setting)
|
||||
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_setting)
|
||||
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_setting(self):
|
||||
filepath = self.filepath_var.get().strip()
|
||||
if not filepath:
|
||||
messagebox.showwarning("警告", "请先在主Tab中设置保存文件路径")
|
||||
return
|
||||
|
||||
setting_file = os.path.join(filepath, "Novel_setting.txt")
|
||||
content = read_file(setting_file)
|
||||
self.setting_text.delete("0.0", "end")
|
||||
self.setting_text.insert("0.0", content)
|
||||
self.log("已加载 Novel_setting.txt 内容到编辑区。")
|
||||
|
||||
def save_novel_setting(self):
|
||||
filepath = self.filepath_var.get().strip()
|
||||
if not filepath:
|
||||
messagebox.showwarning("警告", "请先在主Tab中设置保存文件路径")
|
||||
return
|
||||
|
||||
content = self.setting_text.get("0.0", "end").strip()
|
||||
setting_file = os.path.join(filepath, "Novel_setting.txt")
|
||||
save_string_to_txt(content, setting_file)
|
||||
self.log("已保存对 Novel_setting.txt 的修改。")
|
||||
|
||||
# ------------------ Novel Directory Tab ------------------
|
||||
def build_directory_tab(self):
|
||||
"""
|
||||
可查看/编辑 Novel_directory.txt 并保存
|
||||
"""
|
||||
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_novel_directory)
|
||||
load_btn.grid(row=0, column=0, padx=5, pady=5, sticky="w")
|
||||
|
||||
save_btn = ctk.CTkButton(self.directory_tab, text="保存修改", command=self.save_novel_directory)
|
||||
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_novel_directory(self):
|
||||
filepath = self.filepath_var.get().strip()
|
||||
if not filepath:
|
||||
messagebox.showwarning("警告", "请先在主Tab中设置保存文件路径")
|
||||
return
|
||||
|
||||
directory_file = os.path.join(filepath, "Novel_directory.txt")
|
||||
content = read_file(directory_file)
|
||||
self.directory_text.delete("0.0", "end")
|
||||
self.directory_text.insert("0.0", content)
|
||||
self.log("已加载 Novel_directory.txt 内容到编辑区。")
|
||||
|
||||
def save_novel_directory(self):
|
||||
filepath = self.filepath_var.get().strip()
|
||||
if not filepath:
|
||||
messagebox.showwarning("警告", "请先在主Tab中设置保存文件路径")
|
||||
return
|
||||
|
||||
content = self.directory_text.get("0.0", "end").strip()
|
||||
directory_file = os.path.join(filepath, "Novel_directory.txt")
|
||||
save_string_to_txt(content, directory_file)
|
||||
self.log("已保存对 Novel_directory.txt 的修改。")
|
||||
|
||||
# ------------------ Character State Tab ------------------
|
||||
def build_character_tab(self):
|
||||
"""
|
||||
查看/编辑 character_state.txt
|
||||
"""
|
||||
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)
|
||||
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)
|
||||
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("警告", "请先在主Tab中设置保存文件路径")
|
||||
return
|
||||
|
||||
char_file = os.path.join(filepath, "character_state.txt")
|
||||
content = read_file(char_file)
|
||||
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("警告", "请先在主Tab中设置保存文件路径")
|
||||
return
|
||||
|
||||
content = self.character_text.get("0.0", "end").strip()
|
||||
char_file = os.path.join(filepath, "character_state.txt")
|
||||
save_string_to_txt(content, char_file)
|
||||
self.log("已保存对 character_state.txt 的修改。")
|
||||
|
||||
# ------------------ Global Summary Tab ------------------
|
||||
def build_summary_tab(self):
|
||||
"""
|
||||
查看/编辑 global_summary.txt
|
||||
"""
|
||||
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)
|
||||
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)
|
||||
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("警告", "请先在主Tab中设置保存文件路径")
|
||||
return
|
||||
|
||||
summary_file = os.path.join(filepath, "global_summary.txt")
|
||||
content = read_file(summary_file)
|
||||
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("警告", "请先在主Tab中设置保存文件路径")
|
||||
return
|
||||
|
||||
content = self.summary_text.get("0.0", "end").strip()
|
||||
summary_file = os.path.join(filepath, "global_summary.txt")
|
||||
save_string_to_txt(content, summary_file)
|
||||
self.log("已保存对 global_summary.txt 的修改。")
|
||||
|
||||
# ------------------ 配置管理 ------------------
|
||||
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.embedding_url_var.set(cfg.get("embedding_url", ""))
|
||||
self.temperature_var.set(cfg.get("temperature", 0.7))
|
||||
self.genre_var.set(cfg.get("genre", ""))
|
||||
self.num_chapters_var.set(cfg.get("num_chapters", 10))
|
||||
self.word_number_var.set(cfg.get("word_number", 3000))
|
||||
self.filepath_var.set(cfg.get("filepath", ""))
|
||||
|
||||
# 多行文本
|
||||
self.topic_text.delete("1.0", tk.END)
|
||||
self.topic_text.insert(tk.END, cfg.get("topic", ""))
|
||||
# 主题
|
||||
self.topic_text.delete("0.0", "end")
|
||||
self.topic_text.insert("0.0", cfg.get("topic", ""))
|
||||
|
||||
self.log("已加载配置。")
|
||||
else:
|
||||
@@ -202,9 +478,11 @@ class NovelGeneratorGUI:
|
||||
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(),
|
||||
"embedding_url": self.embedding_url_var.get(),
|
||||
"temperature": self.temperature_var.get(),
|
||||
"topic": self.topic_text.get("1.0", tk.END).strip(),
|
||||
"topic": self.topic_text.get("0.0", "end").strip(),
|
||||
"genre": self.genre_var.get(),
|
||||
"num_chapters": self.num_chapters_var.get(),
|
||||
"word_number": self.word_number_var.get(),
|
||||
@@ -221,17 +499,17 @@ class NovelGeneratorGUI:
|
||||
if selected_dir:
|
||||
self.filepath_var.set(selected_dir)
|
||||
|
||||
# -------------- 日志输出 --------------
|
||||
# ------------------ 日志输出 ------------------
|
||||
def log(self, message: str):
|
||||
self.log_text.insert(tk.END, message + "\n")
|
||||
self.log_text.see(tk.END)
|
||||
self.log_text.insert("end", message + "\n")
|
||||
self.log_text.see("end")
|
||||
|
||||
# -------------- 功能 --------------
|
||||
# ------------------ 功能区 --------------------
|
||||
def disable_button(self, btn):
|
||||
btn.config(state=tk.DISABLED)
|
||||
btn.configure(state="disabled")
|
||||
|
||||
def enable_button(self, btn):
|
||||
btn.config(state=tk.NORMAL)
|
||||
btn.configure(state="normal")
|
||||
|
||||
def generate_full_novel(self):
|
||||
"""生成小说设定 & 目录"""
|
||||
@@ -241,7 +519,7 @@ class NovelGeneratorGUI:
|
||||
api_key = self.api_key_var.get().strip()
|
||||
base_url = self.base_url_var.get().strip()
|
||||
model_name = self.model_name_var.get().strip()
|
||||
topic = self.topic_text.get("1.0", tk.END).strip()
|
||||
topic = self.topic_text.get("0.0", "end").strip()
|
||||
genre = self.genre_var.get().strip()
|
||||
num_chapters = self.num_chapters_var.get()
|
||||
word_number = self.word_number_var.get()
|
||||
@@ -284,6 +562,10 @@ class NovelGeneratorGUI:
|
||||
temperature = self.temperature_var.get()
|
||||
filepath = self.filepath_var.get().strip()
|
||||
|
||||
if not filepath:
|
||||
self.log("请先配置保存文件路径。")
|
||||
return
|
||||
|
||||
novel_settings_file = os.path.join(filepath, "Novel_setting.txt")
|
||||
novel_settings = read_file(novel_settings_file)
|
||||
if not novel_settings.strip():
|
||||
@@ -299,15 +581,17 @@ class NovelGeneratorGUI:
|
||||
|
||||
chap_num = self.chapter_num_var.get()
|
||||
word_number = self.word_number_var.get()
|
||||
user_guidance = self.user_guide_text.get("1.0", tk.END).strip()
|
||||
user_guidance = self.user_guide_text.get("0.0", "end").strip()
|
||||
|
||||
# 获取最近3章文本，生成短期摘要
|
||||
# 获取最近3章文本
|
||||
chapters_dir = os.path.join(filepath, "chapters")
|
||||
recent_3_texts = get_last_n_chapters_text(chapters_dir, chap_num, n=3)
|
||||
|
||||
# 用当前模型生成一个较为详细的最近剧情摘要
|
||||
model_obj = self.get_llm_model(model_name, api_key, base_url, temperature)
|
||||
recent_chapters_summary = summarize_recent_chapters(model_obj, recent_3_texts)
|
||||
recent_chapters_summary = summarize_recent_chapters(
|
||||
None, # 这里只是示例，实际可根据你的需求传入对应的模型
|
||||
recent_3_texts
|
||||
)
|
||||
|
||||
self.log(f"开始生成第{chap_num}章草稿...")
|
||||
draft_text = generate_chapter_draft(
|
||||
@@ -327,9 +611,9 @@ class NovelGeneratorGUI:
|
||||
)
|
||||
if draft_text:
|
||||
self.log(f"✅ 第{chap_num}章草稿生成完成。请在下方查看。")
|
||||
self.chapter_result.delete("1.0", tk.END)
|
||||
self.chapter_result.insert(tk.END, draft_text)
|
||||
self.chapter_result.see(tk.END)
|
||||
self.chapter_result.delete("0.0", "end")
|
||||
self.chapter_result.insert("0.0", draft_text)
|
||||
self.chapter_result.see("end")
|
||||
else:
|
||||
self.log("⚠️ 本章草稿生成失败或无内容。")
|
||||
|
||||
@@ -352,6 +636,10 @@ class NovelGeneratorGUI:
|
||||
temperature = self.temperature_var.get()
|
||||
filepath = self.filepath_var.get().strip()
|
||||
|
||||
if not filepath:
|
||||
self.log("请先配置保存文件路径。")
|
||||
return
|
||||
|
||||
chap_num = self.chapter_num_var.get()
|
||||
word_number = self.word_number_var.get()
|
||||
|
||||
@@ -370,9 +658,9 @@ class NovelGeneratorGUI:
|
||||
# 读取定稿后的文本显示
|
||||
chap_file = os.path.join(filepath, "chapters", f"chapter_{chap_num}.txt")
|
||||
final_text = read_file(chap_file)
|
||||
self.chapter_result.delete("1.0", tk.END)
|
||||
self.chapter_result.insert(tk.END, final_text)
|
||||
self.chapter_result.see(tk.END)
|
||||
self.chapter_result.delete("0.0", "end")
|
||||
self.chapter_result.insert("0.0", final_text)
|
||||
self.chapter_result.see("end")
|
||||
|
||||
except Exception as e:
|
||||
self.log(f"❌ 定稿章节时出错: {e}")
|
||||
@@ -393,16 +681,19 @@ class NovelGeneratorGUI:
|
||||
temperature = self.temperature_var.get()
|
||||
filepath = self.filepath_var.get().strip()
|
||||
|
||||
# 读取关键文件
|
||||
if not filepath:
|
||||
self.log("请先配置保存文件路径。")
|
||||
return
|
||||
|
||||
novel_settings_file = os.path.join(filepath, "Novel_setting.txt")
|
||||
character_state_file = os.path.join(filepath, "character_state.txt")
|
||||
global_summary_file = os.path.join(filepath, "global_summary.txt")
|
||||
plot_arcs_file = os.path.join(filepath, "plot_arcs.txt") # 新增
|
||||
plot_arcs_file = os.path.join(filepath, "plot_arcs.txt")
|
||||
|
||||
novel_setting = read_file(novel_settings_file)
|
||||
character_state = read_file(character_state_file)
|
||||
global_summary = read_file(global_summary_file)
|
||||
plot_arcs = read_file(plot_arcs_file) # 新增
|
||||
plot_arcs = read_file(plot_arcs_file)
|
||||
|
||||
# 获取当前章节文本
|
||||
chap_num = self.chapter_num_var.get()
|
||||
@@ -423,7 +714,7 @@ class NovelGeneratorGUI:
|
||||
base_url=base_url,
|
||||
model_name=model_name,
|
||||
temperature=temperature,
|
||||
plot_arcs=plot_arcs # 新增传入
|
||||
plot_arcs=plot_arcs
|
||||
)
|
||||
self.log("审校结果：")
|
||||
self.log(result)
|
||||
@@ -450,6 +741,7 @@ class NovelGeneratorGUI:
|
||||
import_knowledge_file(
|
||||
api_key=self.api_key_var.get().strip(),
|
||||
base_url=self.base_url_var.get().strip(),
|
||||
embedding_base_url=self.embedding_url_var.get().strip(), # 新增：传入embedding url
|
||||
file_path=selected_file
|
||||
)
|
||||
self.log("✅ 知识库文件导入完成。")
|
||||
@@ -462,42 +754,43 @@ class NovelGeneratorGUI:
|
||||
thread.start()
|
||||
|
||||
def clear_vectorstore_handler(self):
|
||||
"""
|
||||
清空向量库按钮：弹出二次确认，若确认则执行 clear_vector_store()。
|
||||
"""
|
||||
def confirmed_clear():
|
||||
"""清空向量库按钮：弹出二次确认。"""
|
||||
first_confirm = messagebox.askyesno("警告", "确定要清空本地向量库吗？此操作不可恢复！")
|
||||
if first_confirm:
|
||||
second_confirm = messagebox.askyesno("二次确认", "你确定真的要删除所有向量数据吗？此操作不可恢复！")
|
||||
if second_confirm:
|
||||
clear_vector_store()
|
||||
self.log("已清空向量库。")
|
||||
|
||||
first_confirm = messagebox.askyesno("警告", "确定要清空本地向量库吗？此操作不可恢复！")
|
||||
if first_confirm:
|
||||
confirmed_clear()
|
||||
|
||||
# =========== 新增：在 UI 中查看当前剧情要点 =============
|
||||
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 = tk.Toplevel(self.master)
|
||||
top.title("剧情要点/未解决冲突")
|
||||
text_area = scrolledtext.ScrolledText(top, width=60, height=20)
|
||||
text_area.pack(fill="both", expand=True)
|
||||
text_area.insert(tk.END, arcs_text)
|
||||
text_area.config(state=tk.DISABLED)
|
||||
|
||||
def get_llm_model(self, model_name, api_key, base_url, temperature):
|
||||
from langchain_openai import ChatOpenAI
|
||||
return ChatOpenAI(
|
||||
model=model_name,
|
||||
api_key=api_key,
|
||||
base_url=base_url,
|
||||
temperature=temperature
|
||||
)
|
||||
# 弹窗显示
|
||||
top = ctk.CTkToplevel(self.master)
|
||||
top.title("剧情要点/未解决冲突")
|
||||
top.geometry("600x400")
|
||||
|
||||
text_area = ctk.CTkTextbox(top, wrap="word")
|
||||
text_area.pack(fill="both", expand=True, padx=10, pady=10)
|
||||
|
||||
text_area.insert("0.0", arcs_text)
|
||||
text_area.configure(state="disabled")
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
app = ctk.CTk()
|
||||
gui = NovelGeneratorGUI(app)
|
||||
app.mainloop()
|
||||
|
||||
Reference in New Issue
Block a user