770 lines
36 KiB
Python
770 lines
36 KiB
Python
# 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
|
||
import glob
|
||
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,
|
||
build_chapter_prompt
|
||
)
|
||
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.loaded_config["llm_configs"][self.architecture_llm_var.get()]["interface_format"]
|
||
api_key = self.loaded_config["llm_configs"][self.architecture_llm_var.get()]["api_key"]
|
||
base_url = self.loaded_config["llm_configs"][self.architecture_llm_var.get()]["base_url"]
|
||
model_name = self.loaded_config["llm_configs"][self.architecture_llm_var.get()]["model_name"]
|
||
temperature = self.loaded_config["llm_configs"][self.architecture_llm_var.get()]["temperature"]
|
||
max_tokens = self.loaded_config["llm_configs"][self.architecture_llm_var.get()]["max_tokens"]
|
||
timeout_val = self.loaded_config["llm_configs"][self.architecture_llm_var.get()]["timeout"]
|
||
|
||
|
||
|
||
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:
|
||
|
||
number_of_chapters = self.safe_get_int(self.num_chapters_var, 10)
|
||
|
||
interface_format = self.loaded_config["llm_configs"][self.chapter_outline_llm_var.get()]["interface_format"]
|
||
api_key = self.loaded_config["llm_configs"][self.chapter_outline_llm_var.get()]["api_key"]
|
||
base_url = self.loaded_config["llm_configs"][self.chapter_outline_llm_var.get()]["base_url"]
|
||
model_name = self.loaded_config["llm_configs"][self.chapter_outline_llm_var.get()]["model_name"]
|
||
temperature = self.loaded_config["llm_configs"][self.chapter_outline_llm_var.get()]["temperature"]
|
||
max_tokens = self.loaded_config["llm_configs"][self.chapter_outline_llm_var.get()]["max_tokens"]
|
||
timeout_val = self.loaded_config["llm_configs"][self.chapter_outline_llm_var.get()]["timeout"]
|
||
|
||
|
||
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.loaded_config["llm_configs"][self.prompt_draft_llm_var.get()]["interface_format"]
|
||
api_key = self.loaded_config["llm_configs"][self.prompt_draft_llm_var.get()]["api_key"]
|
||
base_url = self.loaded_config["llm_configs"][self.prompt_draft_llm_var.get()]["base_url"]
|
||
model_name = self.loaded_config["llm_configs"][self.prompt_draft_llm_var.get()]["model_name"]
|
||
temperature = self.loaded_config["llm_configs"][self.prompt_draft_llm_var.get()]["temperature"]
|
||
max_tokens = self.loaded_config["llm_configs"][self.prompt_draft_llm_var.get()]["max_tokens"]
|
||
timeout_val = self.loaded_config["llm_configs"][self.prompt_draft_llm_var.get()]["timeout"]
|
||
|
||
|
||
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 函数构造初始提示词
|
||
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.loaded_config["llm_configs"][self.final_chapter_llm_var.get()]["interface_format"]
|
||
api_key = self.loaded_config["llm_configs"][self.final_chapter_llm_var.get()]["api_key"]
|
||
base_url = self.loaded_config["llm_configs"][self.final_chapter_llm_var.get()]["base_url"]
|
||
model_name = self.loaded_config["llm_configs"][self.final_chapter_llm_var.get()]["model_name"]
|
||
temperature = self.loaded_config["llm_configs"][self.final_chapter_llm_var.get()]["temperature"]
|
||
max_tokens = self.loaded_config["llm_configs"][self.final_chapter_llm_var.get()]["max_tokens"]
|
||
timeout_val = self.loaded_config["llm_configs"][self.final_chapter_llm_var.get()]["timeout"]
|
||
|
||
|
||
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:
|
||
interface_format = self.loaded_config["llm_configs"][self.consistency_review_llm_var.get()]["interface_format"]
|
||
api_key = self.loaded_config["llm_configs"][self.consistency_review_llm_var.get()]["api_key"]
|
||
base_url = self.loaded_config["llm_configs"][self.consistency_review_llm_var.get()]["base_url"]
|
||
model_name = self.loaded_config["llm_configs"][self.consistency_review_llm_var.get()]["model_name"]
|
||
temperature = self.loaded_config["llm_configs"][self.consistency_review_llm_var.get()]["temperature"]
|
||
max_tokens = self.loaded_config["llm_configs"][self.consistency_review_llm_var.get()]["max_tokens"]
|
||
timeout = self.loaded_config["llm_configs"][self.consistency_review_llm_var.get()]["timeout"]
|
||
|
||
|
||
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 generate_batch_ui(self):
|
||
def open_batch_dialog():
|
||
dialog = tk.Toplevel()
|
||
chapter_file = os.path.join(self.filepath_var.get().strip(), "chapters")
|
||
files = glob.glob(os.path.join(chapter_file, "chapter_*.txt"))
|
||
if not files:
|
||
num = 1
|
||
else:
|
||
num = max(int(os.path.basename(f).split('_')[1].split('.')[0]) for f in files) + 1
|
||
dialog.geometry("+500+400")
|
||
tk.Label(dialog, text="起始章节").grid(row=0, column=0)
|
||
entry_start = tk.Entry(dialog)
|
||
entry_start.grid(row=0, column=1)
|
||
entry_start.insert(0, str(num))
|
||
tk.Label(dialog, text="结束章节").grid(row=0, column=2)
|
||
entry_end = tk.Entry(dialog)
|
||
entry_end.grid(row=0, column=3)
|
||
tk.Label(dialog, text="期望字数").grid(row=1, column=0)
|
||
entry_word = tk.Entry(dialog)
|
||
entry_word.grid(row=1, column=1)
|
||
entry_word.insert(0, self.word_number_var.get())
|
||
tk.Label(dialog, text="最低字数").grid(row=1, column=2)
|
||
entry_min = tk.Entry(dialog)
|
||
entry_min.grid(row=1, column=3)
|
||
entry_min.insert(0, self.word_number_var.get())
|
||
|
||
auto_enrich_bool = tk.BooleanVar()
|
||
auto_enrich_bool_ck = tk.Checkbutton(dialog, text="低于最低字数时自动扩写", variable=auto_enrich_bool)
|
||
auto_enrich_bool_ck.grid(row=2, column=0)
|
||
|
||
result = {"start": None, "end": None, "word": None, "min": None, "auto_enrich": None, "close": False}
|
||
|
||
|
||
def on_confirm():
|
||
nonlocal result
|
||
if not entry_start.get() or not entry_end.get() or not entry_word.get() or not entry_min.get():
|
||
messagebox.showwarning("警告", "请填写完整信息。")
|
||
return
|
||
|
||
result = {
|
||
"start": entry_start.get(),
|
||
"end": entry_end.get(),
|
||
"word": entry_word.get(),
|
||
"min": entry_min.get(),
|
||
"auto_enrich": auto_enrich_bool.get(),
|
||
"close": False
|
||
}
|
||
dialog.destroy()
|
||
|
||
def on_cancel():
|
||
nonlocal result
|
||
result["close"] = True
|
||
dialog.destroy()
|
||
tk.Button(dialog, text="确认", command=on_confirm).grid(row=2, column=1)
|
||
dialog.protocol("WM_DELETE_WINDOW", on_cancel)
|
||
dialog.transient(self.master)
|
||
dialog.grab_set()
|
||
dialog.wait_window(dialog)
|
||
return result
|
||
|
||
def generate_chapter_batch(self ,i ,word, min, auto_enrich):
|
||
draft_interface_format = self.loaded_config["llm_configs"][self.prompt_draft_llm_var.get()]["interface_format"]
|
||
draft_api_key = self.loaded_config["llm_configs"][self.prompt_draft_llm_var.get()]["api_key"]
|
||
draft_base_url = self.loaded_config["llm_configs"][self.prompt_draft_llm_var.get()]["base_url"]
|
||
draft_model_name = self.loaded_config["llm_configs"][self.prompt_draft_llm_var.get()]["model_name"]
|
||
draft_temperature = self.loaded_config["llm_configs"][self.prompt_draft_llm_var.get()]["temperature"]
|
||
draft_max_tokens = self.loaded_config["llm_configs"][self.prompt_draft_llm_var.get()]["max_tokens"]
|
||
draft_timeout = self.loaded_config["llm_configs"][self.prompt_draft_llm_var.get()]["timeout"]
|
||
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)
|
||
|
||
prompt_text = build_chapter_prompt(
|
||
api_key=draft_api_key,
|
||
base_url=draft_base_url,
|
||
model_name=draft_model_name,
|
||
filepath=self.filepath_var.get().strip(),
|
||
novel_number=i,
|
||
word_number=word,
|
||
temperature=draft_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=draft_interface_format,
|
||
max_tokens=draft_max_tokens,
|
||
timeout=draft_timeout,
|
||
)
|
||
final_prompt = prompt_text
|
||
role_names = [name.strip() for name in self.char_inv_text.get("0.0", "end").split("\n")]
|
||
role_lib_path = os.path.join(self.filepath_var.get().strip(), "角色库")
|
||
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)
|
||
draft_text = generate_chapter_draft(
|
||
api_key=draft_api_key,
|
||
base_url=draft_base_url,
|
||
model_name=draft_model_name,
|
||
filepath=self.filepath_var.get().strip(),
|
||
novel_number=i,
|
||
word_number=word,
|
||
temperature=draft_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=draft_interface_format,
|
||
max_tokens=draft_max_tokens,
|
||
timeout=draft_timeout,
|
||
custom_prompt_text=final_prompt
|
||
)
|
||
|
||
finalize_interface_format = self.loaded_config["llm_configs"][self.final_chapter_llm_var.get()]["interface_format"]
|
||
finalize_api_key = self.loaded_config["llm_configs"][self.final_chapter_llm_var.get()]["api_key"]
|
||
finalize_base_url = self.loaded_config["llm_configs"][self.final_chapter_llm_var.get()]["base_url"]
|
||
finalize_model_name = self.loaded_config["llm_configs"][self.final_chapter_llm_var.get()]["model_name"]
|
||
finalize_temperature = self.loaded_config["llm_configs"][self.final_chapter_llm_var.get()]["temperature"]
|
||
finalize_max_tokens = self.loaded_config["llm_configs"][self.final_chapter_llm_var.get()]["max_tokens"]
|
||
finalize_timeout = self.loaded_config["llm_configs"][self.final_chapter_llm_var.get()]["timeout"]
|
||
|
||
chapters_dir = os.path.join(self.filepath_var.get().strip(), "chapters")
|
||
os.makedirs(chapters_dir, exist_ok=True)
|
||
chapter_path = os.path.join(chapters_dir, f"chapter_{i}.txt")
|
||
if len(draft_text) < 0.7 * min and auto_enrich:
|
||
self.safe_log(f"第{i}章草稿字数 ({len(draft_text)}) 低于目标字数({min})的70%，正在扩写...")
|
||
enriched = enrich_chapter_text(
|
||
chapter_text=draft_text,
|
||
word_number=word,
|
||
api_key=draft_api_key,
|
||
base_url=draft_base_url,
|
||
model_name=draft_model_name,
|
||
temperature=draft_temperature,
|
||
interface_format=draft_interface_format,
|
||
max_tokens=draft_max_tokens,
|
||
timeout=draft_timeout
|
||
)
|
||
draft_text = enriched
|
||
clear_file_content(chapter_path)
|
||
save_string_to_txt(draft_text, chapter_path)
|
||
finalize_chapter(
|
||
novel_number=i,
|
||
word_number=word,
|
||
api_key=finalize_api_key,
|
||
base_url=finalize_base_url,
|
||
model_name=finalize_model_name,
|
||
temperature=finalize_temperature,
|
||
filepath=self.filepath_var.get().strip(),
|
||
embedding_api_key=embedding_api_key,
|
||
embedding_url=embedding_url,
|
||
embedding_interface_format=embedding_interface_format,
|
||
embedding_model_name=embedding_model_name,
|
||
interface_format=finalize_interface_format,
|
||
max_tokens=finalize_max_tokens,
|
||
timeout=finalize_timeout
|
||
)
|
||
|
||
|
||
result = open_batch_dialog()
|
||
if result["close"]:
|
||
return
|
||
|
||
for i in range(int(result["start"]), int(result["end"]) + 1):
|
||
generate_chapter_batch(self, i, int(result["word"]), int(result["min"]), result["auto_enrich"])
|
||
|
||
|
||
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")
|