diff --git a/chapter_directory_parser.py b/chapter_directory_parser.py
index a2c6311..796b9f5 100644
--- a/chapter_directory_parser.py
+++ b/chapter_directory_parser.py
@@ -1,47 +1,126 @@
-# chapter_directory_parser.py
+# chapter_blueprint_parser.py
# -*- coding: utf-8 -*-
import re
-def get_chapter_info_from_directory(novel_directory_content: str, chapter_number: int):
+def parse_chapter_blueprint(blueprint_text: str):
"""
- 从给定的 novel_directory_content 文本中，解析 “第X章” 行，并提取本章的标题和可能的简述。
- 返回一个 dict: {
- "chapter_title": <字符串>,
- "chapter_brief": <字符串> (若没有则为空)
+ 解析整份章节蓝图文本，返回一个列表，每个元素是一个 dict：
+ {
+ "chapter_number": int,
+ "chapter_title": str,
+ "chapter_role": str, # 本章定位
+ "chapter_purpose": str, # 核心作用
+ "suspense_level": str, # 悬念密度
+ "foreshadowing": str, # 伏笔操作
+ "plot_twist_level": str, # 认知颠覆
+ "chapter_summary": str # 本章简述
}
- 注意：目录文本示例格式:
- 第1章 ：潮起
- 第2章 ：阴影浮现 - 主要角色冲突爆发
- ...
- 也可能没有简述，只有一个简单标题。
"""
- # 将文本逐行拆分
- lines = novel_directory_content.splitlines()
+ # 先按空行进行分块，以免多章之间混淆
+ chunks = re.split(r'\n\s*\n', blueprint_text.strip())
+ results = []
- # 章节匹配：形如 “第5章 ：xxx” or “第5章: xxx” or “第5章 xxx”
- pattern = re.compile(r'^第\s*(\d+)\s*章\s*[:：]?\s*(.*)$')
+ chapter_number_pattern = re.compile(r'^第\s*(\d+)\s*章\s*-\s*\[(.*?)\]') # 捕获章号与标题
+ role_pattern = re.compile(r'^本章定位：\s*(.*)$')
+ purpose_pattern = re.compile(r'^核心作用：\s*(.*)$')
+ suspense_pattern = re.compile(r'^悬念密度：\s*(.*)$')
+ foreshadow_pattern = re.compile(r'^伏笔操作：\s*(.*)$')
+ twist_pattern = re.compile(r'^认知颠覆：\s*(.*)$')
+ summary_pattern = re.compile(r'^本章简述：\s*\[(.*)\]$')
- for line in lines:
- match = pattern.match(line.strip())
- if match:
- chap_num = int(match.group(1))
- if chap_num == chapter_number:
- full_title = match.group(2).strip()
- if ' - ' in full_title:
- parts = full_title.split(' - ', 1)
- return {
- "chapter_title": parts[0].strip(),
- "chapter_brief": parts[1].strip()
- }
- else:
- return {
- "chapter_title": full_title,
- "chapter_brief": ""
- }
+ for chunk in chunks:
+ lines = chunk.strip().splitlines()
+ if not lines:
+ continue
- # 如果没有匹配到，返回默认
+ chapter_number = None
+ chapter_title = ""
+ chapter_role = ""
+ chapter_purpose = ""
+ suspense_level = ""
+ foreshadowing = ""
+ plot_twist_level = ""
+ chapter_summary = ""
+
+ # 先匹配第一行（或前几行），找到章号和标题
+ header_match = chapter_number_pattern.match(lines[0].strip()) if lines else None
+ if not header_match:
+ # 不符合格式，跳过
+ continue
+
+ chapter_number = int(header_match.group(1))
+ chapter_title = header_match.group(2).strip()
+
+ # 从后面的行匹配其他字段
+ for line in lines[1:]:
+ line_stripped = line.strip()
+ if not line_stripped:
+ continue
+
+ m_role = role_pattern.match(line_stripped)
+ if m_role:
+ chapter_role = m_role.group(1).strip()
+ continue
+
+ m_purpose = purpose_pattern.match(line_stripped)
+ if m_purpose:
+ chapter_purpose = m_purpose.group(1).strip()
+ continue
+
+ m_suspense = suspense_pattern.match(line_stripped)
+ if m_suspense:
+ suspense_level = m_suspense.group(1).strip()
+ continue
+
+ m_foreshadow = foreshadow_pattern.match(line_stripped)
+ if m_foreshadow:
+ foreshadowing = m_foreshadow.group(1).strip()
+ continue
+
+ m_twist = twist_pattern.match(line_stripped)
+ if m_twist:
+ plot_twist_level = m_twist.group(1).strip()
+ continue
+
+ m_summary = summary_pattern.match(line_stripped)
+ if m_summary:
+ chapter_summary = m_summary.group(1).strip()
+ continue
+
+ results.append({
+ "chapter_number": chapter_number,
+ "chapter_title": chapter_title,
+ "chapter_role": chapter_role,
+ "chapter_purpose": chapter_purpose,
+ "suspense_level": suspense_level,
+ "foreshadowing": foreshadowing,
+ "plot_twist_level": plot_twist_level,
+ "chapter_summary": chapter_summary
+ })
+
+ # 按照 chapter_number 排序后返回
+ results.sort(key=lambda x: x["chapter_number"])
+ return results
+
+
+def get_chapter_info_from_blueprint(blueprint_text: str, target_chapter_number: int):
+ """
+ 在已经加载好的章节蓝图文本中，找到对应章号的结构化信息，返回一个 dict。
+ 若找不到则返回一个默认的结构。
+ """
+ all_chapters = parse_chapter_blueprint(blueprint_text)
+ for ch in all_chapters:
+ if ch["chapter_number"] == target_chapter_number:
+ return ch
+ # 默认返回
return {
- "chapter_title": f"第{chapter_number}章",
- "chapter_brief": ""
+ "chapter_number": target_chapter_number,
+ "chapter_title": f"第{target_chapter_number}章",
+ "chapter_role": "",
+ "chapter_purpose": "",
+ "suspense_level": "",
+ "foreshadowing": "",
+ "plot_twist_level": "",
+ "chapter_summary": ""
}
diff --git a/novel_generator copy.py b/novel_generator copy.py
new file mode 100644
index 0000000..7505948
--- /dev/null
+++ b/novel_generator copy.py
@@ -0,0 +1,830 @@
+# novel_generator.py
+# -*- coding: utf-8 -*-
+import os
+import logging
+import re
+import time
+import traceback
+from typing import List, Optional
+
+# langchain 相关
+from langchain_openai import ChatOpenAI, OpenAIEmbeddings
+from langchain_chroma import Chroma
+from chromadb.config import Settings
+from langchain.docstore.document import Document
+
+# nltk、sentence_transformers 及文本处理相关
+import nltk
+import math
+from sentence_transformers import SentenceTransformer
+from sklearn.metrics.pairwise import cosine_similarity
+
+# 工具函数
+from utils import (
+ read_file, append_text_to_file, clear_file_content,
+ save_string_to_txt
+)
+
+# prompt模板
+from prompt_definitions import (
+ # 设定相关
+ set_prompt, character_prompt, dark_lines_prompt,
+ finalize_setting_prompt, novel_directory_prompt,
+
+ # 写作流程相关
+ summary_prompt, update_character_state_prompt,
+ chapter_outline_prompt, chapter_write_prompt
+)
+
+# Ollama嵌入 (如使用Ollama时需要)
+from embedding_ollama import OllamaEmbeddings
+
+# 用于目录解析章节标题/简介
+from chapter_directory_parser import get_chapter_info_from_directory
+
+
+logging.basicConfig(level=logging.INFO, format="%(asctime)s [%(levelname)s] %(message)s")
+
+
+# ============ 帮助函数 ============
+def remove_think_tags(text: str) -> str:
+ """移除 ... 包裹的内容"""
+ return re.sub(r'.*?', '', text, flags=re.DOTALL)
+
+def debug_log(prompt: str, response_content: str):
+ logging.info(f"\n[######################################### Prompt #########################################]\n {prompt}\n")
+ logging.info(f"\n[######################################### Response #########################################]\n {response_content}\n")
+
+def invoke_with_cleaning(model: ChatOpenAI, prompt: str) -> str:
+ """通用封装：调用模型并移除 ... 文本，记录日志后返回"""
+ response = model.invoke(prompt)
+ if not response:
+ logging.warning("No response from model.")
+ return ""
+ cleaned_text = remove_think_tags(response.content)
+ debug_log(prompt, cleaned_text)
+ return cleaned_text.strip()
+
+def ensure_openai_base_url_has_v1(url: str) -> str:
+ """
+ 若用户输入的 url 不包含 '/v1'，则在末尾追加 '/v1'。
+ """
+ import re
+ url = url.strip()
+ if not url:
+ return url
+ if not re.search(r'/v\d+$', url):
+ if '/v1' not in url:
+ url = url.rstrip('/') + '/v1'
+ return url
+
+def is_using_ollama_api(interface_format: str) -> bool:
+ return interface_format.lower() == "ollama"
+
+def is_using_ml_studio_api(interface_format: str) -> bool:
+ return interface_format.lower() == "ml studio"
+
+
+# ============ 获取 vectorstore 路径 ============
+def get_vectorstore_dir(filepath: str) -> str:
+ """
+ 返回存储向量库的本地路径：
+ 在用户指定的 `filepath` 下创建/使用 'vectorstore' 文件夹。
+ """
+ return os.path.join(filepath, "vectorstore")
+
+
+# ============ 创建 Embeddings 对象 ============
+def create_embeddings_object(
+ api_key: str,
+ base_url: str,
+ interface_format: str,
+ embedding_model_name: str
+):
+ """
+ 根据 embedding_interface_format，选择 Ollama 或 OpenAIEmbeddings 等不同后端。
+ """
+ if is_using_ollama_api(interface_format):
+ fixed_url = base_url.rstrip("/")
+ return OllamaEmbeddings(
+ model_name=embedding_model_name,
+ base_url=fixed_url
+ )
+ else:
+ # OpenAI 或 ML Studio 均使用 OpenAIEmbeddings，注意 base_url 可能需要 ensure /v1
+ fixed_url = ensure_openai_base_url_has_v1(base_url)
+ return OpenAIEmbeddings(
+ openai_api_key=api_key,
+ openai_api_base=fixed_url,
+ model=embedding_model_name
+ )
+
+
+# ============ 向量库相关操作 ============
+def clear_vector_store(filepath: str) -> bool:
+ """
+ 返回值表示是否成功清空向量库。
+ """
+ import shutil
+
+ store_dir = get_vectorstore_dir(filepath)
+ if not os.path.exists(store_dir):
+ logging.info("No vector store found to clear.")
+ return False
+
+ try:
+ if os.path.exists(store_dir):
+ shutil.rmtree(store_dir)
+ logging.info(f"Vector store directory '{store_dir}' removed.")
+ return True
+ except Exception as e:
+ logging.error(f"程序正在运行，无法删除，请在程序关闭后手动前往 {store_dir} 删除目录。\n {str(e)}")
+ traceback.print_exc()
+ return False
+
+def init_vector_store(
+ api_key: str,
+ base_url: str,
+ interface_format: str,
+ embedding_model_name: str,
+ texts: List[str],
+ filepath: str
+) -> Chroma:
+ """
+ 在 filepath 下创建/加载一个 Chroma 向量库并插入 texts。
+ """
+ store_dir = get_vectorstore_dir(filepath)
+ os.makedirs(store_dir, exist_ok=True)
+
+ embeddings = create_embeddings_object(
+ api_key=api_key,
+ base_url=base_url,
+ interface_format=interface_format,
+ embedding_model_name=embedding_model_name
+ )
+ documents = [Document(page_content=str(t)) for t in texts]
+ vectorstore = Chroma.from_documents(
+ documents,
+ embedding=embeddings,
+ persist_directory=store_dir,
+ client_settings=Settings(anonymized_telemetry=False),
+ collection_name="novel_collection"
+ )
+ return vectorstore
+
+
+def load_vector_store(
+ api_key: str,
+ base_url: str,
+ interface_format: str,
+ embedding_model_name: str,
+ filepath: str
+) -> Optional[Chroma]:
+ """
+ 读取已存在的 Chroma 向量库。若不存在则返回 None。
+ """
+ store_dir = get_vectorstore_dir(filepath)
+ if not os.path.exists(store_dir):
+ logging.info("Vector store not found. Will return None.")
+ return None
+
+ embeddings = create_embeddings_object(
+ api_key=api_key,
+ base_url=base_url,
+ interface_format=interface_format,
+ embedding_model_name=embedding_model_name
+ )
+ return Chroma(
+ persist_directory=store_dir,
+ embedding_function=embeddings,
+ client_settings=Settings(anonymized_telemetry=False),
+ collection_name="novel_collection"
+ )
+
+
+def split_by_length(text: str, max_length: int = 500) -> List[str]:
+ segments = []
+ start_idx = 0
+ while start_idx < len(text):
+ end_idx = min(start_idx + max_length, len(text))
+ segment = text[start_idx:end_idx]
+ segments.append(segment.strip())
+ start_idx = end_idx
+ return segments
+
+
+def split_text_for_vectorstore(chapter_text: str,
+ max_length: int = 500,
+ similarity_threshold: float = 0.7) -> List[str]:
+ """
+ 对新的章节文本进行分段后，再用于存入向量库。
+ """
+ if not chapter_text.strip():
+ return []
+
+ nltk.download('punkt', quiet=True)
+ nltk.download('punkt_tab', quiet=True)
+ sentences = nltk.sent_tokenize(chapter_text)
+ if not sentences:
+ return []
+
+ # 先对相近句子进行合并
+ model = SentenceTransformer('paraphrase-MiniLM-L6-v2')
+ embeddings = model.encode(sentences)
+
+ merged_paragraphs = []
+ current_sentences = [sentences[0]]
+ current_embedding = embeddings[0]
+
+ for i in range(1, len(sentences)):
+ sim = cosine_similarity([current_embedding], [embeddings[i]])[0][0]
+ if sim >= similarity_threshold:
+ current_sentences.append(sentences[i])
+ current_embedding = (current_embedding + embeddings[i]) / 2.0
+ else:
+ merged_paragraphs.append(" ".join(current_sentences))
+ current_sentences = [sentences[i]]
+ current_embedding = embeddings[i]
+
+ if current_sentences:
+ merged_paragraphs.append(" ".join(current_sentences))
+
+ # 再对合并好的段落做 max_length 切分
+ final_segments = []
+ for para in merged_paragraphs:
+ if len(para) > max_length:
+ sub_segments = split_by_length(para, max_length=max_length)
+ final_segments.extend(sub_segments)
+ else:
+ final_segments.append(para)
+
+ return final_segments
+
+
+def update_vector_store(
+ api_key: str,
+ base_url: str,
+ new_chapter: str,
+ interface_format: str,
+ embedding_model_name: str,
+ filepath: str
+):
+ """
+ 将最新章节文本插入到向量库中。若库不存在则初始化。
+ """
+ splitted_texts = split_text_for_vectorstore(new_chapter)
+ if not splitted_texts:
+ logging.warning("No valid text to insert into vector store. Skipping.")
+ return
+
+ store = load_vector_store(
+ api_key=api_key,
+ base_url=base_url,
+ interface_format=interface_format,
+ embedding_model_name=embedding_model_name,
+ filepath=filepath
+ )
+ if not store:
+ logging.info("Vector store does not exist. Initializing a new one for new chapter...")
+ init_vector_store(
+ api_key=api_key,
+ base_url=base_url,
+ interface_format=interface_format,
+ embedding_model_name=embedding_model_name,
+ texts=splitted_texts,
+ filepath=filepath
+ )
+ return
+
+ docs = [Document(page_content=str(t)) for t in splitted_texts]
+ store.add_documents(docs)
+ logging.info("Vector store updated with the new chapter splitted segments.")
+
+
+def get_relevant_context_from_vector_store(
+ api_key: str,
+ base_url: str,
+ query: str,
+ interface_format: str,
+ embedding_model_name: str,
+ filepath: str,
+ k: int = 2
+) -> str:
+ """
+ 从向量库中检索与 query 最相关的 k 条文本，拼接后返回。
+ """
+ store = load_vector_store(
+ api_key=api_key,
+ base_url=base_url,
+ interface_format=interface_format,
+ embedding_model_name=embedding_model_name,
+ filepath=filepath
+ )
+ if not store:
+ logging.info("No vector store found. Returning empty context.")
+ return ""
+
+ docs = store.similarity_search(query, k=k)
+ if not docs:
+ logging.info(f"No relevant documents found for query '{query}'. Returning empty context.")
+ return ""
+
+ combined = "\n".join([d.page_content for d in docs])
+ return combined
+
+
+# ============ 1. 生成小说“设定” (Novel_setting.txt) ============
+def Novel_setting_generate(
+ api_key: str,
+ base_url: str,
+ llm_model: str,
+ topic: str,
+ genre: str,
+ number_of_chapters: int,
+ word_number: int,
+ filepath: str,
+ temperature: float = 0.7
+) -> None:
+ os.makedirs(filepath, exist_ok=True)
+
+ model = ChatOpenAI(
+ model=llm_model,
+ api_key=api_key,
+ base_url=ensure_openai_base_url_has_v1(base_url),
+ temperature=temperature
+ )
+
+ # Step1: 基础设定
+ prompt_base = set_prompt.format(
+ topic=topic,
+ genre=genre,
+ number_of_chapters=number_of_chapters,
+ word_number=word_number
+ )
+ base_setting = invoke_with_cleaning(model, prompt_base)
+
+ # Step2: 角色设定
+ prompt_char = character_prompt.format(
+ novel_setting=base_setting
+ )
+ character_setting = invoke_with_cleaning(model, prompt_char)
+
+ # Step3: 暗线/雷点
+ prompt_dark = dark_lines_prompt.format(
+ character_info=character_setting
+ )
+ dark_lines = invoke_with_cleaning(model, prompt_dark)
+
+ # Step4: 最终整合
+ prompt_final = finalize_setting_prompt.format(
+ novel_setting_base=base_setting,
+ character_setting=character_setting,
+ dark_lines=dark_lines
+ )
+ final_novel_setting = invoke_with_cleaning(model, prompt_final)
+
+ filename_set = os.path.join(filepath, "Novel_setting.txt")
+ clear_file_content(filename_set)
+
+ final_novel_setting_cleaned = final_novel_setting.replace('#', '').replace('*', '')
+ save_string_to_txt(final_novel_setting_cleaned, filename_set)
+ logging.info("Novel_setting.txt has been generated successfully.")
+
+
+# ============ 2. 生成小说目录 (Novel_directory.txt) ============
+def Novel_directory_generate(
+ api_key: str,
+ base_url: str,
+ llm_model: str,
+ number_of_chapters: int,
+ filepath: str,
+ temperature: float = 0.7
+) -> None:
+ filename_set = os.path.join(filepath, "Novel_setting.txt")
+ final_novel_setting = read_file(filename_set).strip()
+ if not final_novel_setting:
+ logging.warning("Novel_setting.txt 内容为空，请先生成小说设定。")
+ return
+
+ model = ChatOpenAI(
+ model=llm_model,
+ api_key=api_key,
+ base_url=ensure_openai_base_url_has_v1(base_url),
+ temperature=temperature
+ )
+
+ prompt_dir = novel_directory_prompt.format(
+ final_novel_setting=final_novel_setting,
+ number_of_chapters=number_of_chapters
+ )
+ final_novel_directory = invoke_with_cleaning(model, prompt_dir)
+ if not final_novel_directory.strip():
+ logging.warning("Novel_directory生成结果为空。")
+ return
+
+ filename_dir = os.path.join(filepath, "Novel_directory.txt")
+ clear_file_content(filename_dir)
+
+ final_novel_directory_cleaned = final_novel_directory.replace('#', '').replace('*', '')
+ save_string_to_txt(final_novel_directory_cleaned, filename_dir)
+
+ logging.info("Novel_directory.txt has been generated successfully.")
+
+
+# ============ 获取最近 N 章内容，生成短期摘要 ============
+def get_last_n_chapters_text(chapters_dir: str, current_chapter_num: int, n: int = 3) -> List[str]:
+ texts = []
+ start_chap = max(1, current_chapter_num - n)
+ for c in range(start_chap, current_chapter_num):
+ chap_file = os.path.join(chapters_dir, f"chapter_{c}.txt")
+ if os.path.exists(chap_file):
+ text = read_file(chap_file).strip()
+ if text:
+ texts.append(text)
+ if len(texts) < n:
+ texts = [''] * (n - len(texts)) + texts
+ return texts
+
+def summarize_recent_chapters(
+ llm_model: str,
+ api_key: str,
+ base_url: str,
+ temperature: float,
+ chapters_text_list: List[str]
+) -> str:
+ if not chapters_text_list:
+ return ""
+ if all(not txt.strip() for txt in chapters_text_list):
+ return "暂无摘要。"
+
+ model = ChatOpenAI(
+ model=llm_model,
+ api_key=api_key,
+ base_url=ensure_openai_base_url_has_v1(base_url),
+ temperature=temperature
+ )
+
+ combined_text = "\n".join(chapters_text_list)
+ prompt = f"""你是一名资深长篇小说写作辅助AI，下面是最近几章的合并文本：
+{combined_text}
+
+请用中文输出不超过500字的摘要，只包含主要剧情进展、角色变化、冲突焦点等要点："""
+
+ summary_text = invoke_with_cleaning(model, prompt)
+ if not summary_text:
+ return (combined_text[:800] + "...") if len(combined_text) > 800 else combined_text
+ return summary_text
+
+
+# ============ 剧情要点/冲突 ============
+PLOT_ARCS_PROMPT = """\
+下面是新生成的章节内容:
+{chapter_text}
+
+这里是已记录的剧情要点/未解决冲突(可能为空):
+{old_plot_arcs}
+
+请基于新的章节内容，提炼本章引入或延续的悬念、冲突、角色暗线等，将其合并到旧的剧情要点中。
+若有新的冲突则添加，若有已解决/不再重要的冲突可标注或移除。
+最终输出更新后的剧情要点列表，以帮助后续保持故事整体的一致性和悬念延续。
+"""
+
+def update_plot_arcs(
+ chapter_text: str,
+ old_plot_arcs: str,
+ api_key: str,
+ base_url: str,
+ model_name: str,
+ temperature: float
+) -> str:
+ model = ChatOpenAI(
+ model=model_name,
+ api_key=api_key,
+ base_url=ensure_openai_base_url_has_v1(base_url),
+ temperature=temperature
+ )
+ prompt = PLOT_ARCS_PROMPT.format(
+ chapter_text=chapter_text,
+ old_plot_arcs=old_plot_arcs
+ )
+ arcs_text = invoke_with_cleaning(model, prompt)
+ if not arcs_text:
+ logging.warning("update_plot_arcs: No response or empty result.")
+ return old_plot_arcs
+ return arcs_text
+
+
+# ============ 生成章节草稿 ============
+def generate_chapter_draft(
+ novel_settings: str,
+ global_summary: str,
+ character_state: str,
+ recent_chapters_summary: str,
+ user_guidance: str,
+ api_key: str,
+ base_url: str,
+ model_name: str,
+ novel_number: int,
+ word_number: int,
+ temperature: float,
+ novel_novel_directory: str,
+ filepath: str,
+ interface_format: str,
+ embedding_model_name: str,
+ embedding_base_url: str,
+ embedding_retrieval_k: int = 4
+) -> str:
+ # 1) 根据目录解析标题、简介
+ chapter_info = get_chapter_info_from_directory(novel_novel_directory, novel_number)
+ chapter_title = chapter_info["chapter_title"]
+ chapter_brief = chapter_info["chapter_brief"]
+
+ # 合并要检索的文本（用户指导 + 章节简介 + 最近摘要）
+ combined_query_parts = []
+ if user_guidance.strip():
+ combined_query_parts.append(user_guidance)
+ if chapter_brief.strip():
+ combined_query_parts.append(chapter_brief)
+ if recent_chapters_summary.strip():
+ combined_query_parts.append(recent_chapters_summary)
+ # 额外加一个关键字
+ combined_query_parts.append("回顾剧情")
+
+ merged_query_str = "\n".join(combined_query_parts)
+
+ # 2) 从向量库检索上下文
+ relevant_context = get_relevant_context_from_vector_store(
+ api_key=api_key,
+ base_url=embedding_base_url if embedding_base_url else base_url,
+ query=merged_query_str,
+ interface_format=interface_format,
+ embedding_model_name=embedding_model_name,
+ filepath=filepath,
+ k=embedding_retrieval_k
+ )
+ if not relevant_context.strip():
+ relevant_context = "暂无相关内容。"
+
+ # 3) 生成本章大纲
+ model = ChatOpenAI(
+ model=model_name,
+ api_key=api_key,
+ base_url=ensure_openai_base_url_has_v1(base_url),
+ temperature=temperature
+ )
+
+ outline_prompt_text = chapter_outline_prompt.format(
+ novel_setting=novel_settings,
+ character_state=character_state + "\n\n【检索到的上下文】\n" + relevant_context,
+ global_summary=global_summary,
+ novel_number=novel_number,
+ chapter_title=chapter_title,
+ chapter_brief=chapter_brief
+ )
+ outline_prompt_text += f"\n\n【最近几章摘要】\n{recent_chapters_summary}"
+ outline_prompt_text += f"\n\n【用户指导】\n{user_guidance if user_guidance else '（无）'}"
+
+ chapter_outline = invoke_with_cleaning(model, outline_prompt_text)
+
+ outlines_dir = os.path.join(filepath, "outlines")
+ os.makedirs(outlines_dir, exist_ok=True)
+ outline_file = os.path.join(outlines_dir, f"outline_{novel_number}.txt")
+ clear_file_content(outline_file)
+ save_string_to_txt(chapter_outline, outline_file)
+
+ # 4) 生成正文草稿
+ writing_prompt_text = chapter_write_prompt.format(
+ novel_setting=novel_settings,
+ character_state=character_state + "\n\n【检索到的上下文】\n" + relevant_context,
+ global_summary=global_summary,
+ chapter_outline=chapter_outline,
+ word_number=word_number,
+ novel_number=novel_number,
+ chapter_title=chapter_title,
+ chapter_brief=chapter_brief
+ )
+ writing_prompt_text += f"\n\n【最近几章摘要】\n{recent_chapters_summary}"
+ writing_prompt_text += f"\n\n【用户指导】\n{user_guidance if user_guidance else '（无）'}"
+
+ chapter_content = invoke_with_cleaning(model, writing_prompt_text)
+
+ chapters_dir = os.path.join(filepath, "chapters")
+ os.makedirs(chapters_dir, exist_ok=True)
+ chapter_file = os.path.join(chapters_dir, f"chapter_{novel_number}.txt")
+ clear_file_content(chapter_file)
+ save_string_to_txt(chapter_content, chapter_file)
+
+ logging.info(f"[Draft] Chapter {novel_number} generated as a draft.")
+ return chapter_content
+
+
+# ============ 定稿章节 ============
+def finalize_chapter(
+ novel_number: int,
+ word_number: int,
+ api_key: str,
+ base_url: str,
+ interface_format: str,
+ embedding_model_name: str,
+ model_name: str,
+ temperature: float,
+ filepath: str,
+ embedding_base_url: str,
+ embedding_api_key: str
+):
+ chapters_dir = os.path.join(filepath, "chapters")
+ chapter_file = os.path.join(chapters_dir, f"chapter_{novel_number}.txt")
+ chapter_text = read_file(chapter_file).strip()
+ if not chapter_text:
+ logging.warning(f"Chapter {novel_number} is empty, cannot finalize.")
+ return
+
+ 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")
+
+ old_char_state = read_file(character_state_file)
+ old_global_summary = read_file(global_summary_file)
+ old_plot_arcs = read_file(plot_arcs_file)
+
+ # 篇幅不足，二次扩写
+ if len(chapter_text) < 0.8 * word_number:
+ logging.info("Chapter text is shorter than 80% of desired length. Enriching...")
+ chapter_text = enrich_chapter_text(
+ chapter_text=chapter_text,
+ word_number=word_number,
+ api_key=api_key,
+ base_url=base_url,
+ model_name=model_name,
+ temperature=temperature
+ )
+ clear_file_content(chapter_file)
+ save_string_to_txt(chapter_text, chapter_file)
+
+ # 更新全局摘要
+ model = ChatOpenAI(
+ model=model_name,
+ api_key=api_key,
+ base_url=ensure_openai_base_url_has_v1(base_url),
+ temperature=temperature
+ )
+
+ def update_global_summary(chapter_text: str, old_summary: str) -> str:
+ prompt = summary_prompt.format(
+ chapter_text=chapter_text,
+ global_summary=old_summary
+ )
+ return invoke_with_cleaning(model, prompt) or old_summary
+
+ new_global_summary = update_global_summary(chapter_text, old_global_summary)
+
+ # 更新角色状态
+ def update_character_state(chapter_text: str, old_state: str) -> str:
+ prompt = update_character_state_prompt.format(
+ chapter_text=chapter_text,
+ old_state=old_state
+ )
+ return invoke_with_cleaning(model, prompt) or old_state
+
+ new_char_state = update_character_state(chapter_text, old_char_state)
+
+ # 更新剧情要点
+ new_plot_arcs = update_plot_arcs(
+ chapter_text=chapter_text,
+ old_plot_arcs=old_plot_arcs,
+ api_key=api_key,
+ base_url=base_url,
+ model_name=model_name,
+ temperature=temperature
+ )
+
+ # 写回文件
+ clear_file_content(character_state_file)
+ save_string_to_txt(new_char_state, character_state_file)
+
+ clear_file_content(global_summary_file)
+ save_string_to_txt(new_global_summary, global_summary_file)
+
+ clear_file_content(plot_arcs_file)
+ save_string_to_txt(new_plot_arcs, plot_arcs_file)
+
+ # 更新向量库（此时用 embedding_api_key/embedding_base_url）
+ update_vector_store(
+ api_key=embedding_api_key,
+ base_url=embedding_base_url if embedding_base_url else base_url,
+ new_chapter=chapter_text,
+ interface_format=interface_format,
+ embedding_model_name=embedding_model_name,
+ filepath=filepath
+ )
+
+ logging.info(f"Chapter {novel_number} has been finalized.")
+
+
+def enrich_chapter_text(
+ chapter_text: str,
+ word_number: int,
+ api_key: str,
+ base_url: str,
+ model_name: str,
+ temperature: float
+) -> str:
+ model = ChatOpenAI(
+ model=model_name,
+ api_key=api_key,
+ base_url=ensure_openai_base_url_has_v1(base_url),
+ temperature=temperature
+ )
+ prompt = f"""以下是当前章节文本，可能篇幅较短，请在保持剧情连贯的前提下进行扩写，使其更充实、生动，并尽量靠近目标 {word_number} 字数。
+
+原章节内容：
+{chapter_text}"""
+ enriched_text = invoke_with_cleaning(model, prompt)
+ return enriched_text if enriched_text else chapter_text
+
+
+# ============ 导入外部知识文本到向量库 ============
+def advanced_split_content(content: str,
+ similarity_threshold: float = 0.7,
+ max_length: int = 500) -> List[str]:
+ """
+ 将文本先按句子切分，然后根据语义相似度进行合并，最后按 max_length 二次切分。
+ """
+ nltk.download('punkt', quiet=True)
+ sentences = nltk.sent_tokenize(content)
+ if not sentences:
+ return []
+
+ model = SentenceTransformer('paraphrase-MiniLM-L6-v2')
+ embeddings = model.encode(sentences)
+
+ merged_paragraphs = []
+ current_sentences = [sentences[0]]
+ current_embedding = embeddings[0]
+
+ for i in range(1, len(sentences)):
+ sim = cosine_similarity([current_embedding], [embeddings[i]])[0][0]
+ if sim >= similarity_threshold:
+ current_sentences.append(sentences[i])
+ current_embedding = (current_embedding + embeddings[i]) / 2.0
+ else:
+ merged_paragraphs.append(" ".join(current_sentences))
+ current_sentences = [sentences[i]]
+ current_embedding = embeddings[i]
+
+ if current_sentences:
+ merged_paragraphs.append(" ".join(current_sentences))
+
+ final_segments = []
+ for para in merged_paragraphs:
+ if len(para) > max_length:
+ sub_segments = split_by_length(para, max_length=max_length)
+ final_segments.extend(sub_segments)
+ else:
+ final_segments.append(para)
+
+ return final_segments
+
+def import_knowledge_file(
+ api_key: str,
+ base_url: str,
+ interface_format: str,
+ embedding_model_name: str,
+ file_path: str,
+ embedding_base_url: str,
+ filepath: str
+):
+ logging.info(f"开始导入知识库文件: {file_path}, 接口格式: {interface_format}, 模型: {embedding_model_name}")
+ if not os.path.exists(file_path):
+ logging.warning(f"知识库文件不存在: {file_path}")
+ return
+
+ content = read_file(file_path)
+ if not content.strip():
+ logging.warning("知识库文件内容为空。")
+ return
+
+ paragraphs = advanced_split_content(content)
+
+ # 若向量库不存在则初始化，否则追加
+ store = load_vector_store(
+ api_key=api_key,
+ base_url=base_url if base_url else "http://localhost:11434/v1",
+ interface_format=interface_format,
+ embedding_model_name=embedding_model_name,
+ filepath=filepath
+ )
+ if not store:
+ logging.info("Vector store does not exist. Initializing a new one for knowledge import...")
+ init_vector_store(
+ api_key=api_key,
+ base_url=base_url if base_url else "http://localhost:11434/v1",
+ interface_format=interface_format,
+ embedding_model_name=embedding_model_name,
+ texts=paragraphs,
+ filepath=filepath
+ )
+ else:
+ docs = [Document(page_content=str(p)) for p in paragraphs]
+ store.add_documents(docs)
+ logging.info("知识库文件已成功导入至向量库。")
diff --git a/novel_generator.py b/novel_generator.py
index 7505948..f29076c 100644
--- a/novel_generator.py
+++ b/novel_generator.py
@@ -15,7 +15,6 @@ from langchain.docstore.document import Document
# nltk、sentence_transformers 及文本处理相关
import nltk
-import math
from sentence_transformers import SentenceTransformer
from sklearn.metrics.pairwise import cosine_similarity
@@ -27,26 +26,26 @@ from utils import (
# prompt模板
from prompt_definitions import (
- # 设定相关
- set_prompt, character_prompt, dark_lines_prompt,
- finalize_setting_prompt, novel_directory_prompt,
-
- # 写作流程相关
- summary_prompt, update_character_state_prompt,
- chapter_outline_prompt, chapter_write_prompt
+ core_seed_prompt,
+ character_dynamics_prompt,
+ world_building_prompt,
+ plot_architecture_prompt,
+ chapter_blueprint_prompt,
+ summary_prompt,
+ update_character_state_prompt,
+ scene_dynamics_prompt
)
# Ollama嵌入 (如使用Ollama时需要)
from embedding_ollama import OllamaEmbeddings
# 用于目录解析章节标题/简介
-from chapter_directory_parser import get_chapter_info_from_directory
-
+from chapter_directory_parser import get_chapter_info_from_blueprint
logging.basicConfig(level=logging.INFO, format="%(asctime)s [%(levelname)s] %(message)s")
-# ============ 帮助函数 ============
+# ============ 基础工具 ============
def remove_think_tags(text: str) -> str:
"""移除 ... 包裹的内容"""
return re.sub(r'.*?', '', text, flags=re.DOTALL)
@@ -333,8 +332,8 @@ def get_relevant_context_from_vector_store(
return combined
-# ============ 1. 生成小说“设定” (Novel_setting.txt) ============
-def Novel_setting_generate(
+# ========== 1) 生成总体架构 (Novel_architecture.txt) ==========
+def Novel_architecture_generate(
api_key: str,
base_url: str,
llm_model: str,
@@ -345,8 +344,15 @@ def Novel_setting_generate(
filepath: str,
temperature: float = 0.7
) -> None:
+ """
+ 依次调用：
+ 1. core_seed_prompt
+ 2. character_dynamics_prompt
+ 3. world_building_prompt
+ 4. plot_architecture_prompt
+ 将结果整合为“Novel_architecture.txt”。
+ """
os.makedirs(filepath, exist_ok=True)
-
model = ChatOpenAI(
model=llm_model,
api_key=api_key,
@@ -354,58 +360,95 @@ def Novel_setting_generate(
temperature=temperature
)
- # Step1: 基础设定
- prompt_base = set_prompt.format(
+ # 1) 核心种子
+ prompt_core = core_seed_prompt.format(
topic=topic,
genre=genre,
number_of_chapters=number_of_chapters,
word_number=word_number
)
- base_setting = invoke_with_cleaning(model, prompt_base)
+ core_seed_result = invoke_with_cleaning(model, prompt_core)
+ core_seed_text = core_seed_result.strip()
- # Step2: 角色设定
- prompt_char = character_prompt.format(
- novel_setting=base_setting
+ # 2) 角色动力学
+ prompt_character = character_dynamics_prompt.format(core_seed=core_seed_text)
+ character_dynamics_result = invoke_with_cleaning(model, prompt_character)
+ character_dynamics_text = character_dynamics_result.strip()
+
+ # 3) 世界观
+ prompt_world = world_building_prompt.format(core_seed=core_seed_text)
+ world_building_result = invoke_with_cleaning(model, prompt_world)
+ world_building_text = world_building_result.strip()
+
+ # 4) 三幕式情节架构
+ prompt_plot = plot_architecture_prompt.format(
+ core_seed=core_seed_text,
+ character_dynamics=character_dynamics_text,
+ world_building=world_building_text
)
- character_setting = invoke_with_cleaning(model, prompt_char)
+ plot_arch_result = invoke_with_cleaning(model, prompt_plot)
+ plot_arch_text = plot_arch_result.strip()
- # Step3: 暗线/雷点
- prompt_dark = dark_lines_prompt.format(
- character_info=character_setting
+ # 整合并写入 Novel_architecture.txt
+ final_content = (
+ "#=== 1) 核心种子 ===\n"
+ f"{core_seed_text}\n\n"
+ "#=== 2) 角色动力学 ===\n"
+ f"{character_dynamics_text}\n\n"
+ "#=== 3) 世界观 ===\n"
+ f"{world_building_text}\n\n"
+ "#=== 4) 三幕式情节架构 ===\n"
+ f"{plot_arch_text}\n"
)
- dark_lines = invoke_with_cleaning(model, prompt_dark)
- # Step4: 最终整合
- prompt_final = finalize_setting_prompt.format(
- novel_setting_base=base_setting,
- character_setting=character_setting,
- dark_lines=dark_lines
- )
- final_novel_setting = invoke_with_cleaning(model, prompt_final)
+ arch_file = os.path.join(filepath, "Novel_architecture.txt")
+ clear_file_content(arch_file)
+ save_string_to_txt(final_content, arch_file)
- filename_set = os.path.join(filepath, "Novel_setting.txt")
- clear_file_content(filename_set)
-
- final_novel_setting_cleaned = final_novel_setting.replace('#', '').replace('*', '')
- save_string_to_txt(final_novel_setting_cleaned, filename_set)
- logging.info("Novel_setting.txt has been generated successfully.")
+ logging.info("Novel_architecture.txt has been generated successfully.")
-# ============ 2. 生成小说目录 (Novel_directory.txt) ============
-def Novel_directory_generate(
+# ========== 2) 生成章节蓝图 (Novel_directory.txt) ==========
+def Chapter_blueprint_generate(
api_key: str,
base_url: str,
llm_model: str,
- number_of_chapters: int,
filepath: str,
temperature: float = 0.7
) -> None:
- filename_set = os.path.join(filepath, "Novel_setting.txt")
- final_novel_setting = read_file(filename_set).strip()
- if not final_novel_setting:
- logging.warning("Novel_setting.txt 内容为空，请先生成小说设定。")
+ """
+ 基于“Novel_architecture.txt”中的三幕式情节架构，调用 chapter_blueprint_prompt，
+ 生成章节蓝图并写入 Novel_directory.txt。
+ """
+ arch_file = os.path.join(filepath, "Novel_architecture.txt")
+ if not os.path.exists(arch_file):
+ logging.warning("Novel_architecture.txt not found. Please generate architecture first.")
return
+ architecture_text = read_file(arch_file).strip()
+ if not architecture_text:
+ logging.warning("Novel_architecture.txt is empty.")
+ return
+
+ # 从内容中尽量提取 number_of_chapters
+ # 如果之前已经存储了 number_of_chapters，可以在外面传入，这里做简化：
+ # 这里用正则或者其他逻辑提取，但演示时直接写 10 也可
+ match_chaps = re.search(r'约(\d+)章', architecture_text)
+ if match_chaps:
+ number_of_chapters = int(match_chaps.group(1))
+ else:
+ number_of_chapters = 10 # fallback
+
+ # 提取三幕式文本
+ # 在写入时，我们将 4) 三幕式情节架构 作为传给 prompt 的核心
+ # 这里做一个简易匹配
+ plot_arch_text = ""
+ # 假设 "#=== 4) 三幕式情节架构 ===" 是分隔点
+ pat_plot = r'#=== 4\) 三幕式情节架构 ===\n([\s\S]+)$'
+ m = re.search(pat_plot, architecture_text)
+ if m:
+ plot_arch_text = m.group(1).strip()
+
model = ChatOpenAI(
model=llm_model,
api_key=api_key,
@@ -413,22 +456,20 @@ def Novel_directory_generate(
temperature=temperature
)
- prompt_dir = novel_directory_prompt.format(
- final_novel_setting=final_novel_setting,
+ prompt = chapter_blueprint_prompt.format(
+ plot_architecture=plot_arch_text,
number_of_chapters=number_of_chapters
)
- final_novel_directory = invoke_with_cleaning(model, prompt_dir)
- if not final_novel_directory.strip():
- logging.warning("Novel_directory生成结果为空。")
+ blueprint_text = invoke_with_cleaning(model, prompt)
+ if not blueprint_text.strip():
+ logging.warning("Chapter blueprint generation result is empty.")
return
filename_dir = os.path.join(filepath, "Novel_directory.txt")
clear_file_content(filename_dir)
+ save_string_to_txt(blueprint_text, filename_dir)
- final_novel_directory_cleaned = final_novel_directory.replace('#', '').replace('*', '')
- save_string_to_txt(final_novel_directory_cleaned, filename_dir)
-
- logging.info("Novel_directory.txt has been generated successfully.")
+ logging.info("Novel_directory.txt (chapter blueprint) has been generated successfully.")
# ============ 获取最近 N 章内容，生成短期摘要 ============
@@ -514,58 +555,114 @@ def update_plot_arcs(
return arcs_text
-# ============ 生成章节草稿 ============
+# ========== 3) 生成章节草稿 ==========
+
def generate_chapter_draft(
- novel_settings: str,
- global_summary: str,
- character_state: str,
- recent_chapters_summary: str,
- user_guidance: str,
api_key: str,
base_url: str,
model_name: str,
+ filepath: str,
novel_number: int,
word_number: int,
temperature: float,
- novel_novel_directory: str,
- filepath: str,
- interface_format: str,
- embedding_model_name: str,
- embedding_base_url: str,
- embedding_retrieval_k: int = 4
+ user_guidance: str,
+ characters_involved: str,
+ key_items: str,
+ scene_location: str,
+ time_constraint: str,
+ embedding_retrieval_k: int = 2
) -> str:
- # 1) 根据目录解析标题、简介
- chapter_info = get_chapter_info_from_directory(novel_novel_directory, novel_number)
+ """
+ 根据 scene_dynamics_prompt，生成本章草稿。
+ - novel_architecture 取自 Novel_architecture.txt
+ - blueprint 取自 Novel_directory.txt
+ - global_summary, character_state 分别取自全局摘要、角色状态文件
+ - 向量库检索上下文
+ - 用户还可以额外提供四个可选元素：核心人物、关键道具、空间坐标、时间压力
+ """
+
+ # 1) 读取相关文件
+ arch_file = os.path.join(filepath, "Novel_architecture.txt")
+ novel_architecture_text = read_file(arch_file)
+
+ directory_file = os.path.join(filepath, "Novel_directory.txt")
+ blueprint_text = read_file(directory_file)
+
+ global_summary_file = os.path.join(filepath, "global_summary.txt")
+ global_summary_text = read_file(global_summary_file)
+
+ character_state_file = os.path.join(filepath, "character_state.txt")
+ character_state_text = read_file(character_state_file)
+
+ # 2) 解析 blueprint，得到本章所需的字段
+ chapter_info = get_chapter_info_from_blueprint(blueprint_text, novel_number)
chapter_title = chapter_info["chapter_title"]
- chapter_brief = chapter_info["chapter_brief"]
+ chapter_role = chapter_info["chapter_role"]
+ chapter_purpose = chapter_info["chapter_purpose"]
+ suspense_level = chapter_info["suspense_level"]
+ foreshadowing = chapter_info["foreshadowing"]
+ plot_twist_level = chapter_info["plot_twist_level"]
+ chapter_summary = chapter_info["chapter_summary"]
- # 合并要检索的文本（用户指导 + 章节简介 + 最近摘要）
- combined_query_parts = []
- if user_guidance.strip():
- combined_query_parts.append(user_guidance)
- if chapter_brief.strip():
- combined_query_parts.append(chapter_brief)
- if recent_chapters_summary.strip():
- combined_query_parts.append(recent_chapters_summary)
- # 额外加一个关键字
- combined_query_parts.append("回顾剧情")
+ # 3) 取最近3章文本，拼成查询语句 => 用于向量库检索
+ chapters_dir = os.path.join(filepath, "chapters")
+ recent_3_texts = get_last_n_chapters_text(chapters_dir, novel_number, n=3)
+ merged_query_str = "回顾剧情：\n" + "\n".join(recent_3_texts) + "\n" + user_guidance
- merged_query_str = "\n".join(combined_query_parts)
-
- # 2) 从向量库检索上下文
+ # 4) 检索向量库上下文
relevant_context = get_relevant_context_from_vector_store(
api_key=api_key,
- base_url=embedding_base_url if embedding_base_url else base_url,
+ base_url=base_url,
query=merged_query_str,
- interface_format=interface_format,
- embedding_model_name=embedding_model_name,
+ embedding_model_name=model_name,
filepath=filepath,
k=embedding_retrieval_k
)
- if not relevant_context.strip():
- relevant_context = "暂无相关内容。"
- # 3) 生成本章大纲
+ if not relevant_context.strip():
+ relevant_context = "（无检索到的上下文）"
+
+ # 5) 构造prompt，调用 scene_dynamics_prompt
+ # 在这里，我们拆分架构文本，以便给模型提供：
+ # - “世界观”与“小说设定”可以从 arch_file 中的相应片段读取
+ # 这里为了简化，直接把 novel_architecture_text 整体塞入 novel_setting
+ # 也可更精细地拆分 "#=== 3) 世界观 ===" 片段给 world_building
+ # 下方仅作示例。
+ world_building_text = ""
+ match_world = re.search(r'#=== 3\) 世界观 ===\n([\s\S]+?)\n#===', novel_architecture_text)
+ if match_world:
+ world_building_text = match_world.group(1).strip()
+ else:
+ world_building_text = "暂无世界观信息"
+
+ novel_setting_text = novel_architecture_text # 整份当做“小说设定”参考
+
+ prompt_text = scene_dynamics_prompt.format(
+ novel_number=novel_number,
+ chapter_title=chapter_title,
+ chapter_role=chapter_role,
+ chapter_purpose=chapter_purpose,
+ suspense_level=suspense_level,
+ foreshadowing=foreshadowing,
+ plot_twist_level=plot_twist_level,
+ chapter_summary=chapter_summary,
+
+ characters_involved=characters_involved,
+ key_items=key_items,
+ scene_location=scene_location,
+ time_constraint=time_constraint,
+
+ world_building=world_building_text,
+ novel_setting=novel_setting_text,
+ global_summary=global_summary_text,
+ character_state=character_state_text
+ )
+
+ # 因为我们还想让模型了解向量库检索到的上下文，可以合并到最后
+ prompt_text += f"\n\n【检索到的上下文】\n{relevant_context}"
+ # 也可合并用户指导
+ prompt_text += f"\n\n【用户指导】\n{user_guidance}\n"
+
model = ChatOpenAI(
model=model_name,
api_key=api_key,
@@ -573,44 +670,15 @@ def generate_chapter_draft(
temperature=temperature
)
- outline_prompt_text = chapter_outline_prompt.format(
- novel_setting=novel_settings,
- character_state=character_state + "\n\n【检索到的上下文】\n" + relevant_context,
- global_summary=global_summary,
- novel_number=novel_number,
- chapter_title=chapter_title,
- chapter_brief=chapter_brief
- )
- outline_prompt_text += f"\n\n【最近几章摘要】\n{recent_chapters_summary}"
- outline_prompt_text += f"\n\n【用户指导】\n{user_guidance if user_guidance else '（无）'}"
-
- chapter_outline = invoke_with_cleaning(model, outline_prompt_text)
-
- outlines_dir = os.path.join(filepath, "outlines")
- os.makedirs(outlines_dir, exist_ok=True)
- outline_file = os.path.join(outlines_dir, f"outline_{novel_number}.txt")
- clear_file_content(outline_file)
- save_string_to_txt(chapter_outline, outline_file)
-
- # 4) 生成正文草稿
- writing_prompt_text = chapter_write_prompt.format(
- novel_setting=novel_settings,
- character_state=character_state + "\n\n【检索到的上下文】\n" + relevant_context,
- global_summary=global_summary,
- chapter_outline=chapter_outline,
- word_number=word_number,
- novel_number=novel_number,
- chapter_title=chapter_title,
- chapter_brief=chapter_brief
- )
- writing_prompt_text += f"\n\n【最近几章摘要】\n{recent_chapters_summary}"
- writing_prompt_text += f"\n\n【用户指导】\n{user_guidance if user_guidance else '（无）'}"
-
- chapter_content = invoke_with_cleaning(model, writing_prompt_text)
+ chapter_content = invoke_with_cleaning(model, prompt_text)
+ if not chapter_content.strip():
+ logging.warning("Generated chapter draft is empty.")
+ # 6) 写入 chapters 目录
chapters_dir = os.path.join(filepath, "chapters")
os.makedirs(chapters_dir, exist_ok=True)
chapter_file = os.path.join(chapters_dir, f"chapter_{novel_number}.txt")
+
clear_file_content(chapter_file)
save_string_to_txt(chapter_content, chapter_file)
@@ -618,20 +686,20 @@ def generate_chapter_draft(
return chapter_content
-# ============ 定稿章节 ============
+# ========== 4) 定稿章节 ==========
def finalize_chapter(
novel_number: int,
word_number: int,
api_key: str,
base_url: str,
- interface_format: str,
- embedding_model_name: str,
model_name: str,
temperature: float,
filepath: str,
- embedding_base_url: str,
- embedding_api_key: str
+ embedding_model_name: str
):
+ """
+ 定稿：更新全局摘要、角色状态，并将本章文本插入向量库。
+ """
chapters_dir = os.path.join(filepath, "chapters")
chapter_file = os.path.join(chapters_dir, f"chapter_{novel_number}.txt")
chapter_text = read_file(chapter_file).strip()
@@ -639,82 +707,55 @@ def finalize_chapter(
logging.warning(f"Chapter {novel_number} is empty, cannot finalize.")
return
- 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")
-
- old_char_state = read_file(character_state_file)
- old_global_summary = read_file(global_summary_file)
- old_plot_arcs = read_file(plot_arcs_file)
-
- # 篇幅不足，二次扩写
- if len(chapter_text) < 0.8 * word_number:
- logging.info("Chapter text is shorter than 80% of desired length. Enriching...")
- chapter_text = enrich_chapter_text(
- chapter_text=chapter_text,
- word_number=word_number,
- api_key=api_key,
- base_url=base_url,
- model_name=model_name,
- temperature=temperature
- )
+ # 如果长度比目标少很多，可考虑在此扩写
+ if len(chapter_text) < 0.6 * word_number:
+ chapter_text = enrich_chapter_text(chapter_text, word_number, api_key, base_url, model_name, temperature)
clear_file_content(chapter_file)
save_string_to_txt(chapter_text, chapter_file)
- # 更新全局摘要
+ # 读取全局摘要、角色状态
+ global_summary_file = os.path.join(filepath, "global_summary.txt")
+ old_global_summary = read_file(global_summary_file)
+ character_state_file = os.path.join(filepath, "character_state.txt")
+ old_character_state = read_file(character_state_file)
+
+ # 1) 更新全局摘要
model = ChatOpenAI(
model=model_name,
api_key=api_key,
base_url=ensure_openai_base_url_has_v1(base_url),
temperature=temperature
)
-
- def update_global_summary(chapter_text: str, old_summary: str) -> str:
- prompt = summary_prompt.format(
- chapter_text=chapter_text,
- global_summary=old_summary
- )
- return invoke_with_cleaning(model, prompt) or old_summary
-
- new_global_summary = update_global_summary(chapter_text, old_global_summary)
-
- # 更新角色状态
- def update_character_state(chapter_text: str, old_state: str) -> str:
- prompt = update_character_state_prompt.format(
- chapter_text=chapter_text,
- old_state=old_state
- )
- return invoke_with_cleaning(model, prompt) or old_state
-
- new_char_state = update_character_state(chapter_text, old_char_state)
-
- # 更新剧情要点
- new_plot_arcs = update_plot_arcs(
+ prompt_summary = summary_prompt.format(
chapter_text=chapter_text,
- old_plot_arcs=old_plot_arcs,
- api_key=api_key,
- base_url=base_url,
- model_name=model_name,
- temperature=temperature
+ global_summary=old_global_summary
)
+ new_global_summary = invoke_with_cleaning(model, prompt_summary)
+ if not new_global_summary.strip():
+ new_global_summary = old_global_summary
+
+ # 2) 更新角色状态
+ prompt_char_state = update_character_state_prompt.format(
+ chapter_text=chapter_text,
+ old_state=old_character_state
+ )
+ new_char_state = invoke_with_cleaning(model, prompt_char_state)
+ if not new_char_state.strip():
+ new_char_state = old_character_state
# 写回文件
- clear_file_content(character_state_file)
- save_string_to_txt(new_char_state, character_state_file)
-
clear_file_content(global_summary_file)
save_string_to_txt(new_global_summary, global_summary_file)
- clear_file_content(plot_arcs_file)
- save_string_to_txt(new_plot_arcs, plot_arcs_file)
+ clear_file_content(character_state_file)
+ save_string_to_txt(new_char_state, character_state_file)
- # 更新向量库（此时用 embedding_api_key/embedding_base_url）
+ # 3) 更新向量库
update_vector_store(
- api_key=embedding_api_key,
- base_url=embedding_base_url if embedding_base_url else base_url,
+ api_key=api_key,
+ base_url=base_url,
new_chapter=chapter_text,
- interface_format=interface_format,
- embedding_model_name=embedding_model_name,
+ model_name=embedding_model_name, # 用于embedding
filepath=filepath
)
diff --git a/prompt_definitions.py b/prompt_definitions.py
index ebe588c..989236d 100644
--- a/prompt_definitions.py
+++ b/prompt_definitions.py
@@ -186,7 +186,7 @@ update_character_state_prompt = """\
仅返回更新后的角色状态文本，不要解释任何内容。
"""
-# =============== 7. 章节正文写作 ===================
+# =============== 8. 章节正文写作 ===================
scene_dynamics_prompt = """\
即将创作：第{novel_number}章《{chapter_title}》
本章定位：{chapter_role}
@@ -227,4 +227,4 @@ scene_dynamics_prompt = """\
最后设置一个"钩链转折"：结尾同时回收旧悬念/创造新悬念/抛出新危机/颠覆某个认知预设/神转折等。
仅返回章节正文文本，不要解释任何内容。
-"""
\ No newline at end of file
+"""
diff --git a/ui.py b/ui.py
index 7e2f461..2a4fb1d 100644
--- a/ui.py
+++ b/ui.py
@@ -7,17 +7,17 @@ import threading
import customtkinter as ctk
from tkinter import filedialog, messagebox
import traceback
+
from config_manager import load_config, save_config
from utils import read_file, save_string_to_txt, clear_file_content
from novel_generator import (
- Novel_setting_generate,
- Novel_directory_generate,
+ Novel_architecture_generate,
+ Chapter_blueprint_generate,
generate_chapter_draft,
finalize_chapter,
import_knowledge_file,
clear_vector_store,
get_last_n_chapters_text,
- summarize_recent_chapters
)
from consistency_checker import check_consistency
@@ -28,8 +28,6 @@ def log_error(message: str):
"""
logging.error(f"{message}\n{traceback.format_exc()}")
-
-# 设置全局主题和颜色
ctk.set_appearance_mode("System")
ctk.set_default_color_theme("blue")
@@ -39,79 +37,65 @@ class NovelGeneratorGUI:
self.master = master
self.master.title("Novel Generator GUI")
- # 防止因 icon.ico 不存在导致程序崩溃
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)
- # ========== 主要的属性变量 ==========
-
- # LLM 接口相关
+ # 主要属性变量
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.base_url_var = ctk.StringVar(value=self.loaded_config.get("base_url", "https://api.openai.com/v1"))
self.interface_format_var = ctk.StringVar(value=self.loaded_config.get("interface_format", "OpenAI"))
self.model_name_var = ctk.StringVar(value=self.loaded_config.get("model_name", "gpt-4o-mini"))
-
- # 仍然用 DoubleVar，但因为是 Slider，不会让用户手动清空文本，一般不会出现空字符串问题
self.temperature_var = ctk.DoubleVar(value=self.loaded_config.get("temperature", 0.7))
- # Embedding 接口相关
+ # Embedding相关
self.embedding_interface_format_var = ctk.StringVar(value=self.loaded_config.get("embedding_interface_format", "OpenAI"))
self.embedding_api_key_var = ctk.StringVar(value=self.loaded_config.get("embedding_api_key", ""))
- self.embedding_url_var = ctk.StringVar(value=self.loaded_config.get("embedding_url", ""))
- self.embedding_model_name_var = ctk.StringVar(value=self.loaded_config.get("embedding_model_name", ""))
-
- # ### CHANGED：将 IntVar 改为 StringVar，避免用户清空输入时抛错
+ self.embedding_url_var = ctk.StringVar(value=self.loaded_config.get("embedding_url", "https://api.openai.com/v1"))
+ self.embedding_model_name_var = ctk.StringVar(value=self.loaded_config.get("embedding_model_name", "text-embedding-ada-002"))
self.embedding_retrieval_k_var = ctk.StringVar(value=str(self.loaded_config.get("embedding_retrieval_k", 4)))
- # 小说通用参数
self.topic_default = self.loaded_config.get("topic", "")
self.genre_var = ctk.StringVar(value=self.loaded_config.get("genre", "玄幻"))
-
- # ### CHANGED：将章节数、每章字数改为 StringVar
self.num_chapters_var = ctk.StringVar(value=str(self.loaded_config.get("num_chapters", 10)))
self.word_number_var = ctk.StringVar(value=str(self.loaded_config.get("word_number", 3000)))
-
self.filepath_var = ctk.StringVar(value=self.loaded_config.get("filepath", ""))
- # ### CHANGED：章节号也改为 StringVar
self.chapter_num_var = ctk.StringVar(value="1")
- # ========== 主容器使用 TabView ==========
+ # 新增四个可选要素
+ self.characters_involved_var = ctk.StringVar(value="")
+ self.key_items_var = ctk.StringVar(value="")
+ self.scene_location_var = ctk.StringVar(value="")
+ self.time_constraint_var = ctk.StringVar(value="")
+
+ # UI 布局
self.tabview = ctk.CTkTabview(self.master, width=1200, height=800)
self.tabview.pack(fill="both", expand=True)
- # 创建各个Tab
self.main_tab = self.tabview.add("Main Functions")
- self.setting_tab = self.tabview.add("Novel Settings")
- self.directory_tab = self.tabview.add("Novel Directory")
+ self.setting_tab = self.tabview.add("Novel Architecture")
+ self.directory_tab = self.tabview.add("Chapter Blueprint")
self.character_tab = self.tabview.add("Character State")
self.summary_tab = self.tabview.add("Global Summary")
self.chapters_view_tab = self.tabview.add("Chapters Manage")
- # 构建各个 Tab 的布局
self.build_main_tab()
self.build_setting_tab()
self.build_directory_tab()
self.build_character_tab()
self.build_summary_tab()
- self.build_chapters_tab() # 新增
+ self.build_chapters_tab()
- # ------------------ 工具方法：安全获取 IntVar (现已兼容 StringVar) ------------------
def safe_get_int(self, var, default=1):
- """
- 尝试把 StringVar 或 IntVar 中的值转换为 int；
- 若失败则将其重置为 default 并返回 default。
- """
try:
val_str = str(var.get()).strip()
return int(val_str)
@@ -119,102 +103,51 @@ class NovelGeneratorGUI:
var.set(str(default))
return default
- # ------------------ 主功能 Tab ------------------
+ # ------------------ 主 Tab ------------------
def build_main_tab(self):
- """
- 主Tab分为左右两栏：
- 左侧：本章内容、Step按钮、日志
- 右侧：配置区域(带边框) + 保存/加载配置 + 小说参数 + 可选功能按钮
- """
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):
- """
- 左侧布局:
- row=0 -> “本章内容”文本框 (chapter_result)
- row=1 -> Step1~4按钮
- row=2 -> “输出日志”标题
- row=3 -> “输出日志”文本框 (log_text)
- """
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.grid_columnconfigure(0, weight=1)
+ self.left_frame.columnconfigure(0, weight=1)
- # ========== 本章内容 ==========
- chapter_label = ctk.CTkLabel(
- self.left_frame,
- text="本章内容 (可编辑)",
- font=("Microsoft YaHei", 12)
- )
+ chapter_label = ctk.CTkLabel(self.left_frame, text="本章内容 (可编辑)", font=("Microsoft YaHei", 12))
chapter_label.grid(row=0, column=0, padx=5, pady=(5, 0), sticky="w")
- # 章节草稿：可编辑
- self.chapter_result = ctk.CTkTextbox(
- self.left_frame,
- wrap="word",
- font=("Microsoft YaHei", 14)
- )
+ self.chapter_result = ctk.CTkTextbox(self.left_frame, wrap="word", font=("Microsoft YaHei", 14))
self.chapter_result.grid(row=1, column=0, sticky="nsew", padx=5, pady=(0, 5))
- # ========== Step1~4按钮 ==========
- self.build_step_buttons_area()
-
- # ========== 输出日志 label ==========
- log_label = ctk.CTkLabel(
- self.left_frame,
- text="输出日志 (只读)",
- font=("Microsoft YaHei", 12)
- )
- log_label.grid(row=3, column=0, padx=5, pady=(5, 0), sticky="w")
-
- # ========== 日志：只读 ==========
- self.log_text = ctk.CTkTextbox(
- self.left_frame,
- wrap="word",
- font=("Microsoft YaHei", 12)
- )
- self.log_text.grid(row=4, column=0, sticky="nsew", padx=5, pady=(0, 5))
- self.log_text.configure(state="disabled")
-
- def build_step_buttons_area(self):
- """
- 在左侧，仅放 Step1~Step4 四个按钮
- """
+ # 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_setting = ctk.CTkButton(
+ self.btn_generate_architecture = ctk.CTkButton(
self.step_buttons_frame,
- text="Step1. 生成设定",
- command=self.generate_novel_setting_ui,
+ text="Step1. 生成架构",
+ command=self.generate_novel_architecture_ui,
font=("Microsoft YaHei", 12)
)
- self.btn_generate_setting.grid(row=0, column=0, padx=5, pady=2, sticky="ew")
+ 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_novel_directory_ui,
+ 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")
@@ -235,42 +168,240 @@ class NovelGeneratorGUI:
)
self.btn_finalize_chapter.grid(row=0, column=3, padx=5, pady=2, sticky="ew")
+ # 日志
+ log_label = ctk.CTkLabel(self.left_frame, text="输出日志 (只读)", font=("Microsoft YaHei", 12))
+ log_label.grid(row=3, column=0, padx=5, pady=(5, 0), sticky="w")
+
+ self.log_text = ctk.CTkTextbox(self.left_frame, wrap="word", font=("Microsoft YaHei", 12))
+ self.log_text.grid(row=4, column=0, sticky="nsew", padx=5, pady=(0, 5))
+ self.log_text.configure(state="disabled")
+
def build_right_layout(self):
- """
- 右侧布局，包含：
- row=0 -> 带边框的配置区 (TabView + 保存/加载配置按钮)
- row=1 -> 小说参数区域
- row=2 -> 可选功能按钮 (一致性审校 / 导入知识库 / 清空向量库 / 查看剧情要点)
- """
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)
- # 1) 配置区
- self.config_frame = ctk.CTkFrame(
- self.right_frame,
- corner_radius=10,
- border_width=2,
- border_color="gray"
- )
+ # 配置区
+ self.config_frame = ctk.CTkFrame(self.right_frame, corner_radius=10, border_width=2, border_color="gray")
self.config_frame.grid(row=0, column=0, sticky="ew", padx=5, pady=5)
self.config_frame.columnconfigure(0, weight=1)
- self.build_config_tabview() # LLM、Embedding等配置
- self.build_main_buttons_area() # 保存/加载配置按钮
+ self.build_config_tabview()
+ self.build_main_buttons_area()
- # 2) 小说参数
+ # 小说参数
self.build_novel_params_area(start_row=1)
- # 3) 可选功能按钮
+ # 可选功能按钮
self.build_optional_buttons_area(start_row=2)
- # ------------------ 可选功能按钮区域（右下） ------------------
+ def build_config_tabview(self):
+ self.config_tabview = ctk.CTkTabview(self.config_frame, width=600, height=200)
+ self.config_tabview.grid(row=0, column=0, sticky="we", padx=5, pady=5)
+
+ self.ai_config_tab = self.config_tabview.add("LLM Model settings")
+ self.embeddings_config_tab = self.config_tabview.add("Embedding settings")
+
+ self.build_ai_config_tab()
+ self.build_embeddings_config_tab()
+
+ def build_ai_config_tab(self):
+ def on_interface_format_changed(new_value):
+ if new_value == "OpenAI":
+ self.base_url_var.set("https://api.openai.com/v1")
+ # 可根据需要扩展 Ollama / ML Studio 等
+
+ for i in range(5):
+ 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)
+
+ api_key_label = ctk.CTkLabel(self.ai_config_tab, text="LLM API Key:", font=("Microsoft YaHei", 12))
+ api_key_label.grid(row=0, column=0, padx=5, pady=5, sticky="e")
+ api_key_entry = ctk.CTkEntry(self.ai_config_tab, textvariable=self.api_key_var, font=("Microsoft YaHei", 12))
+ api_key_entry.grid(row=0, column=1, padx=5, pady=5, sticky="nsew")
+
+ base_url_label = ctk.CTkLabel(self.ai_config_tab, text="LLM Base URL:", font=("Microsoft YaHei", 12))
+ base_url_label.grid(row=1, column=0, padx=5, pady=5, sticky="e")
+ 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, sticky="nsew")
+
+ interface_label = ctk.CTkLabel(self.ai_config_tab, text="LLM 接口格式:", font=("Microsoft YaHei", 12))
+ interface_label.grid(row=2, column=0, padx=5, pady=5, sticky="e")
+ interface_options = ["OpenAI"]
+ 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, sticky="nsew")
+
+ model_name_label = ctk.CTkLabel(self.ai_config_tab, text="Model Name:", font=("Microsoft YaHei", 12))
+ model_name_label.grid(row=3, column=0, padx=5, pady=5, sticky="e")
+ 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, sticky="nsew")
+
+ temp_label = ctk.CTkLabel(self.ai_config_tab, text="Temperature:", font=("Microsoft YaHei", 12))
+ temp_label.grid(row=4, 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.ai_config_tab,
+ from_=0.0, to=1.0,
+ number_of_steps=100,
+ 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=1, pady=1, sticky="w")
+
+ def build_embeddings_config_tab(self):
+ 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)
+
+ emb_api_key_label = ctk.CTkLabel(self.embeddings_config_tab, text="Embedding API Key:", font=("Microsoft YaHei", 12))
+ emb_api_key_label.grid(row=0, column=0, padx=5, pady=5, sticky="e")
+ 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")
+
+ emb_interface_label = ctk.CTkLabel(self.embeddings_config_tab, text="Embedding 接口格式:", font=("Microsoft YaHei", 12))
+ emb_interface_label.grid(row=1, column=0, padx=5, pady=5, sticky="e")
+ emb_interface_options = ["OpenAI"] # 可扩展
+ emb_interface_dropdown = ctk.CTkOptionMenu(
+ self.embeddings_config_tab,
+ values=emb_interface_options,
+ variable=self.embedding_interface_format_var,
+ font=("Microsoft YaHei", 12)
+ )
+ emb_interface_dropdown.grid(row=1, column=1, padx=5, pady=5, sticky="nsew")
+
+ emb_url_label = ctk.CTkLabel(self.embeddings_config_tab, text="Embedding Base URL:", font=("Microsoft YaHei", 12))
+ emb_url_label.grid(row=2, column=0, padx=5, pady=5, sticky="e")
+ 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")
+
+ emb_model_name_label = ctk.CTkLabel(self.embeddings_config_tab, text="Embedding Model Name:", font=("Microsoft YaHei", 12))
+ emb_model_name_label.grid(row=3, column=0, padx=5, pady=5, sticky="e")
+ 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")
+
+ emb_retrieval_k_label = ctk.CTkLabel(self.embeddings_config_tab, text="Retrieval Top-K:", font=("Microsoft YaHei", 12))
+ emb_retrieval_k_label.grid(row=4, column=0, padx=5, pady=5, sticky="e")
+ emb_retrieval_k_entry = ctk.CTkEntry(self.embeddings_config_tab, textvariable=self.embedding_retrieval_k_var, font=("Microsoft YaHei", 12))
+ emb_retrieval_k_entry.grid(row=4, column=1, padx=5, pady=5, sticky="nsew")
+
+ def build_main_buttons_area(self):
+ self.btn_frame_config = ctk.CTkFrame(self.config_frame)
+ self.btn_frame_config.grid(row=1, column=0, padx=5, pady=5, sticky="ew")
+ self.btn_frame_config.columnconfigure(0, weight=1)
+ self.btn_frame_config.columnconfigure(1, weight=1)
+
+ save_config_btn = ctk.CTkButton(self.btn_frame_config, text="保存配置", command=self.save_config_btn, font=("Microsoft YaHei", 12))
+ save_config_btn.grid(row=0, column=0, padx=5, pady=5, sticky="ew")
+
+ load_config_btn = ctk.CTkButton(self.btn_frame_config, text="加载配置", command=self.load_config_btn, font=("Microsoft YaHei", 12))
+ load_config_btn.grid(row=0, column=1, padx=5, pady=5, sticky="ew")
+
+ def build_novel_params_area(self, start_row=1):
+ self.params_frame = ctk.CTkScrollableFrame(self.right_frame, orientation="vertical")
+ self.params_frame.grid(row=start_row, column=0, sticky="nsew", padx=5, pady=5)
+ self.params_frame.columnconfigure(1, weight=1)
+
+ topic_label = ctk.CTkLabel(self.params_frame, text="主题(Topic):", font=("Microsoft YaHei", 12))
+ topic_label.grid(row=0, column=0, padx=5, pady=5, sticky="e")
+ self.topic_text = ctk.CTkTextbox(self.params_frame, width=200, height=80, wrap="word", font=("Microsoft YaHei", 12))
+ self.topic_text.grid(row=0, column=1, padx=5, pady=5, sticky="nsew")
+ if self.topic_default:
+ self.topic_text.insert("0.0", self.topic_default)
+
+ genre_label = ctk.CTkLabel(self.params_frame, text="类型(Genre):", font=("Microsoft YaHei", 12))
+ genre_label.grid(row=1, column=0, padx=5, pady=5, sticky="e")
+ 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")
+
+ row_for_chapter_and_word = 2
+ num_chapters_label = ctk.CTkLabel(self.params_frame, text="章节数:", font=("Microsoft YaHei", 12))
+ num_chapters_label.grid(row=row_for_chapter_and_word, column=0, padx=5, pady=5, sticky="e")
+
+ ch_word_frame = ctk.CTkFrame(self.params_frame)
+ ch_word_frame.grid(row=row_for_chapter_and_word, column=1, padx=5, pady=5, sticky="ew")
+ ch_word_frame.columnconfigure((0, 1, 2, 3), weight=0)
+
+ num_chapters_entry = ctk.CTkEntry(ch_word_frame, textvariable=self.num_chapters_var, width=60, font=("Microsoft YaHei", 12))
+ num_chapters_entry.grid(row=0, column=0, padx=5, pady=5, sticky="w")
+
+ word_number_label = ctk.CTkLabel(ch_word_frame, text="每章字数:", font=("Microsoft YaHei", 12))
+ word_number_label.grid(row=0, column=1, padx=(15, 5), pady=5, sticky="e")
+
+ word_number_entry = ctk.CTkEntry(ch_word_frame, textvariable=self.word_number_var, width=60, font=("Microsoft YaHei", 12))
+ word_number_entry.grid(row=0, column=2, padx=5, pady=5, sticky="w")
+
+ # 保存路径
+ filepath_label = ctk.CTkLabel(self.params_frame, text="保存路径:", font=("Microsoft YaHei", 12))
+ filepath_label.grid(row=3, column=0, padx=5, pady=5, sticky="e")
+
+ self.filepath_frame = ctk.CTkFrame(self.params_frame)
+ self.filepath_frame.grid(row=3, 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")
+
+ # 章节号
+ chapter_num_label = ctk.CTkLabel(self.params_frame, text="章节号:", font=("Microsoft YaHei", 12))
+ chapter_num_label.grid(row=4, column=0, padx=5, pady=5, sticky="e")
+ chapter_num_entry = ctk.CTkEntry(self.params_frame, textvariable=self.chapter_num_var, width=80, font=("Microsoft YaHei", 12))
+ chapter_num_entry.grid(row=4, column=1, padx=5, pady=5, sticky="w")
+
+ # 用户指导
+ guide_label = ctk.CTkLabel(self.params_frame, text="本章指导:", font=("Microsoft YaHei", 12))
+ guide_label.grid(row=5, column=0, padx=5, pady=5, sticky="ne")
+ self.user_guide_text = ctk.CTkTextbox(self.params_frame, width=200, height=80, wrap="word", font=("Microsoft YaHei", 12))
+ self.user_guide_text.grid(row=5, column=1, padx=5, pady=5, sticky="nsew")
+
+ # 新增：四个可选元素
+ row_index = 6
+
+ char_inv_label = ctk.CTkLabel(self.params_frame, text="核心人物:", font=("Microsoft YaHei", 12))
+ char_inv_label.grid(row=row_index, column=0, padx=5, pady=5, sticky="e")
+ char_inv_entry = ctk.CTkEntry(self.params_frame, textvariable=self.characters_involved_var, font=("Microsoft YaHei", 12))
+ char_inv_entry.grid(row=row_index, column=1, padx=5, pady=5, sticky="ew")
+
+ row_index += 1
+ key_items_label = ctk.CTkLabel(self.params_frame, text="关键道具:", font=("Microsoft YaHei", 12))
+ key_items_label.grid(row=row_index, column=0, padx=5, pady=5, sticky="e")
+ key_items_entry = ctk.CTkEntry(self.params_frame, textvariable=self.key_items_var, font=("Microsoft YaHei", 12))
+ key_items_entry.grid(row=row_index, column=1, padx=5, pady=5, sticky="ew")
+
+ row_index += 1
+ scene_loc_label = ctk.CTkLabel(self.params_frame, text="空间坐标:", font=("Microsoft YaHei", 12))
+ scene_loc_label.grid(row=row_index, column=0, padx=5, pady=5, sticky="e")
+ scene_loc_entry = ctk.CTkEntry(self.params_frame, textvariable=self.scene_location_var, font=("Microsoft YaHei", 12))
+ scene_loc_entry.grid(row=row_index, column=1, padx=5, pady=5, sticky="ew")
+
+ row_index += 1
+ time_const_label = ctk.CTkLabel(self.params_frame, text="时间压力:", font=("Microsoft YaHei", 12))
+ time_const_label.grid(row=row_index, column=0, padx=5, pady=5, sticky="e")
+ time_const_entry = ctk.CTkEntry(self.params_frame, textvariable=self.time_constraint_var, font=("Microsoft YaHei", 12))
+ time_const_entry.grid(row=row_index, column=1, padx=5, pady=5, sticky="ew")
+
def build_optional_buttons_area(self, start_row=2):
- """
- 放在右侧的最下方：包括 一致性审校、导入知识库、清空向量库、查看剧情要点
- """
self.optional_btn_frame = ctk.CTkFrame(self.right_frame)
self.optional_btn_frame.grid(row=start_row, column=0, sticky="ew", padx=5, pady=5)
self.optional_btn_frame.columnconfigure((0, 1, 2, 3), weight=1)
@@ -308,363 +439,385 @@ class NovelGeneratorGUI:
)
self.plot_arcs_btn.grid(row=0, column=3, padx=5, pady=5, sticky="ew")
- # ========== 配置区域（TabView） ==========
- def build_config_tabview(self):
- self.config_tabview = ctk.CTkTabview(self.config_frame, width=600, height=200)
- self.config_tabview.grid(row=0, column=0, sticky="we", padx=5, pady=5)
+ def load_config_btn(self):
+ cfg = load_config(self.config_file)
+ if cfg:
+ self.api_key_var.set(cfg.get("api_key", ""))
+ self.base_url_var.set(cfg.get("base_url", ""))
+ self.interface_format_var.set(cfg.get("interface_format", "OpenAI"))
+ self.model_name_var.set(cfg.get("model_name", ""))
+ self.temperature_var.set(cfg.get("temperature", 0.7))
+ self.embedding_api_key_var.set(cfg.get("embedding_api_key", ""))
+ self.embedding_interface_format_var.set(cfg.get("embedding_interface_format", "OpenAI"))
+ self.embedding_url_var.set(cfg.get("embedding_url", ""))
+ self.embedding_model_name_var.set(cfg.get("embedding_model_name", ""))
+ self.embedding_retrieval_k_var.set(str(cfg.get("embedding_retrieval_k", 4)))
+ self.genre_var.set(cfg.get("genre", ""))
+ self.num_chapters_var.set(str(cfg.get("num_chapters", 10)))
+ self.word_number_var.set(str(cfg.get("word_number", 3000)))
+ self.filepath_var.set(cfg.get("filepath", ""))
- self.ai_config_tab = self.config_tabview.add("LLM Model settings")
- self.embeddings_config_tab = self.config_tabview.add("Embedding settings")
+ topic_value = cfg.get("topic", "")
+ self.topic_text.delete("0.0", "end")
+ self.topic_text.insert("0.0", topic_value)
- self.build_ai_config_tab()
- self.build_embeddings_config_tab()
+ self.log("已加载配置。")
+ else:
+ messagebox.showwarning("提示", "未找到或无法读取配置文件。")
- def build_ai_config_tab(self):
- def on_interface_format_changed(new_value):
- if new_value == "Ollama":
- self.base_url_var.set("http://localhost:11434/v1")
- elif new_value == "ML Studio":
- self.base_url_var.set("http://localhost:1234/v1")
- elif new_value == "OpenAI":
- self.base_url_var.set("https://api.openai.com/v1")
+ def save_config_btn(self):
+ config_data = {
+ "api_key": self.api_key_var.get(),
+ "base_url": self.base_url_var.get(),
+ "interface_format": self.interface_format_var.get(),
+ "model_name": self.model_name_var.get(),
+ "temperature": self.temperature_var.get(),
- for i in range(5):
- 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) # for temp label
+ "embedding_api_key": self.embedding_api_key_var.get(),
+ "embedding_interface_format": self.embedding_interface_format_var.get(),
+ "embedding_url": self.embedding_url_var.get(),
+ "embedding_model_name": self.embedding_model_name_var.get(),
+ "embedding_retrieval_k": self.safe_get_int(self.embedding_retrieval_k_var, 4),
- api_key_label = ctk.CTkLabel(
- self.ai_config_tab,
- text="LLM API Key:",
- font=("Microsoft YaHei", 12)
- )
- api_key_label.grid(row=0, column=0, padx=5, pady=5, sticky="e")
- api_key_entry = ctk.CTkEntry(
- self.ai_config_tab,
- textvariable=self.api_key_var,
- font=("Microsoft YaHei", 12)
- )
- api_key_entry.grid(row=0, column=1, padx=5, pady=5, sticky="nsew")
+ "topic": self.topic_text.get("0.0", "end").strip(),
+ "genre": self.genre_var.get(),
+ "num_chapters": self.safe_get_int(self.num_chapters_var, 10),
+ "word_number": self.safe_get_int(self.word_number_var, 3000),
+ "filepath": self.filepath_var.get()
+ }
+ if save_config(config_data, self.config_file):
+ messagebox.showinfo("提示", "配置已保存至 config.json")
+ self.log("配置已保存。")
+ else:
+ messagebox.showerror("错误", "保存配置失败。")
- base_url_label = ctk.CTkLabel(
- self.ai_config_tab,
- text="LLM Base URL:",
- font=("Microsoft YaHei", 12)
- )
- base_url_label.grid(row=1, column=0, padx=5, pady=5, sticky="e")
- 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, sticky="nsew")
+ def browse_folder(self):
+ selected_dir = filedialog.askdirectory()
+ if selected_dir:
+ self.filepath_var.set(selected_dir)
- interface_label = ctk.CTkLabel(
- self.ai_config_tab,
- text="LLM 接口格式:",
- font=("Microsoft YaHei", 12)
- )
- interface_label.grid(row=2, column=0, padx=5, pady=5, sticky="e")
- interface_options = ["OpenAI", "Ollama", "ML Studio"]
- interface_dropdown = ctk.CTkOptionMenu(
- self.ai_config_tab,
- values=interface_options,
- variable=self.interface_format_var,
- command=on_interface_format_changed,
- font=("Microsoft YaHei", 12)
- )
- interface_dropdown.grid(row=2, column=1, padx=5, pady=5, sticky="nsew")
+ 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")
- model_name_label = ctk.CTkLabel(
- self.ai_config_tab,
- text="Model Name:",
- font=("Microsoft YaHei", 12)
- )
- model_name_label.grid(row=3, column=0, padx=5, pady=5, sticky="e")
- 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, sticky="nsew")
+ def safe_log(self, message: str):
+ self.master.after(0, lambda: self.log(message))
- temp_label = ctk.CTkLabel(
- self.ai_config_tab,
- text="Temperature:",
- font=("Microsoft YaHei", 12)
- )
- temp_label.grid(row=4, column=0, padx=5, pady=5, sticky="e")
+ def disable_button_safe(self, btn):
+ self.master.after(0, lambda: btn.configure(state="disabled"))
- def update_temp_label(value):
- self.temp_value_label.configure(text=f"{float(value):.2f}")
+ def enable_button_safe(self, btn):
+ self.master.after(0, lambda: btn.configure(state="normal"))
- temp_scale = ctk.CTkSlider(
- self.ai_config_tab,
- from_=0.0, to=1.0,
- number_of_steps=100,
- command=update_temp_label,
- variable=self.temperature_var
- )
- temp_scale.grid(row=4, column=1, padx=5, pady=5, sticky="we")
+ def handle_exception(self, context: str):
+ full_message = f"{context}\n{traceback.format_exc()}"
+ logging.error(full_message)
+ self.safe_log(full_message)
- 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=1, pady=1, sticky="w")
+ # ------------------ Step1: 生成架构 ------------------
+ def generate_novel_architecture_ui(self):
+ filepath = self.filepath_var.get().strip()
+ if not filepath:
+ messagebox.showwarning("警告", "请先选择保存文件路径")
+ return
- def build_embeddings_config_tab(self):
- def on_embedding_interface_changed(new_value):
- if new_value == "Ollama":
- self.embedding_url_var.set("http://localhost:11434/api")
- elif new_value == "ML Studio":
- self.embedding_url_var.set("http://localhost:1234/v1")
- elif new_value == "OpenAI":
- self.embedding_url_var.set("https://api.openai.com/v1")
+ def task():
+ self.disable_button_safe(self.btn_generate_architecture)
+ 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()
- 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)
+ 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)
- emb_api_key_label = ctk.CTkLabel(
- self.embeddings_config_tab,
- text="Embedding API Key:",
- font=("Microsoft YaHei", 12)
- )
- emb_api_key_label.grid(row=0, column=0, padx=5, pady=5, sticky="e")
- 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")
+ self.safe_log("开始生成小说架构...")
+ Novel_architecture_generate(
+ 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
+ )
+ self.safe_log("✅ 小说架构生成完成。请在 'Novel Architecture' 标签页查看或编辑。")
+ except Exception:
+ self.handle_exception("生成小说架构时出错")
+ finally:
+ self.enable_button_safe(self.btn_generate_architecture)
- emb_interface_label = ctk.CTkLabel(
- self.embeddings_config_tab,
- text="Embedding 接口格式:",
- font=("Microsoft YaHei", 12)
- )
- emb_interface_label.grid(row=1, column=0, padx=5, pady=5, sticky="e")
- emb_interface_options = ["OpenAI", "Ollama", "ML Studio"]
- emb_interface_dropdown = ctk.CTkOptionMenu(
- self.embeddings_config_tab,
- values=emb_interface_options,
- variable=self.embedding_interface_format_var,
- command=on_embedding_interface_changed,
- font=("Microsoft YaHei", 12)
- )
- emb_interface_dropdown.grid(row=1, column=1, padx=5, pady=5, sticky="nsew")
+ threading.Thread(target=task, daemon=True).start()
- emb_url_label = ctk.CTkLabel(
- self.embeddings_config_tab,
- text="Embedding Base URL:",
- font=("Microsoft YaHei", 12)
- )
- emb_url_label.grid(row=2, column=0, padx=5, pady=5, sticky="e")
- 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")
+ # ------------------ Step2: 生成章节蓝图 ------------------
+ def generate_chapter_blueprint_ui(self):
+ filepath = self.filepath_var.get().strip()
+ if not filepath:
+ messagebox.showwarning("警告", "请先选择保存文件路径")
+ return
- emb_model_name_label = ctk.CTkLabel(
- self.embeddings_config_tab,
- text="Embedding Model Name:",
- font=("Microsoft YaHei", 12)
- )
- emb_model_name_label.grid(row=3, column=0, padx=5, pady=5, sticky="e")
- 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")
+ def task():
+ self.disable_button_safe(self.btn_generate_directory)
+ 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()
- emb_retrieval_k_label = ctk.CTkLabel(
- self.embeddings_config_tab,
- text="Retrieval Top-K:",
- font=("Microsoft YaHei", 12)
- )
- emb_retrieval_k_label.grid(row=4, column=0, padx=5, pady=5, sticky="e")
- 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")
+ self.safe_log("开始生成章节蓝图...")
+ Chapter_blueprint_generate(
+ api_key=api_key,
+ base_url=base_url,
+ llm_model=model_name,
+ filepath=filepath,
+ temperature=temperature
+ )
+ self.safe_log("✅ 章节蓝图生成完成。请在 'Chapter Blueprint' 标签页查看或编辑。")
+ except Exception:
+ self.handle_exception("生成章节蓝图时出错")
+ finally:
+ self.enable_button_safe(self.btn_generate_directory)
- # ========== 保存/加载 配置按钮区域 ==========
- def build_main_buttons_area(self):
- """
- 放置在带边框配置区(config_frame)内部，位于TabView下方
- """
- 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)
+ threading.Thread(target=task, daemon=True).start()
- 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")
+ # ------------------ Step3: 生成草稿 ------------------
+ def generate_chapter_draft_ui(self):
+ filepath = self.filepath_var.get().strip()
+ if not filepath:
+ messagebox.showwarning("警告", "请先配置保存文件路径。")
+ return
- 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 task():
+ self.disable_button_safe(self.btn_generate_chapter)
+ 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()
- # ========== 小说参数区域 ==========
- 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)
+ 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()
- # 主题(Topic)
- topic_label = ctk.CTkLabel(
- self.params_frame,
- text="主题(Topic):",
- font=("Microsoft YaHei", 12)
- )
- topic_label.grid(row=0, column=0, padx=5, pady=5, sticky="e")
- self.topic_text = ctk.CTkTextbox(
- self.params_frame,
- width=200,
- height=80,
- wrap="word",
- font=("Microsoft YaHei", 12)
- )
- self.topic_text.grid(row=0, column=1, padx=5, pady=5, sticky="nsew")
- if self.topic_default:
- self.topic_text.insert("0.0", self.topic_default)
+ # 新增四个可选要素
+ 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()
- # 类型(Genre)
- genre_label = ctk.CTkLabel(
- self.params_frame,
- text="类型(Genre):",
- font=("Microsoft YaHei", 12)
- )
- genre_label.grid(row=1, column=0, padx=5, pady=5, sticky="e")
- 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")
+ embedding_model_name = self.embedding_model_name_var.get().strip()
+ embedding_k = self.safe_get_int(self.embedding_retrieval_k_var, 4)
- # 章节数、每章字数 放在同一行
- row_for_chapter_and_word = 2
- num_chapters_label = ctk.CTkLabel(
- self.params_frame,
- text="章节数:",
- font=("Microsoft YaHei", 12)
- )
- num_chapters_label.grid(row=row_for_chapter_and_word, column=0, padx=5, pady=5, sticky="e")
+ self.safe_log(f"开始生成第{chap_num}章草稿...")
+ draft_text = generate_chapter_draft(
+ api_key=api_key,
+ base_url=base_url,
+ model_name=model_name,
+ filepath=filepath,
+ novel_number=chap_num,
+ word_number=word_number,
+ temperature=temperature,
+ user_guidance=user_guidance,
+ characters_involved=char_inv,
+ key_items=key_items,
+ scene_location=scene_loc,
+ time_constraint=time_constr,
+ embedding_retrieval_k=embedding_k
+ )
+ 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("⚠️ 本章草稿生成失败或无内容。")
- ch_word_frame = ctk.CTkFrame(self.params_frame)
- ch_word_frame.grid(row=row_for_chapter_and_word, column=1, padx=5, pady=5, sticky="ew")
- ch_word_frame.columnconfigure(0, weight=0)
- ch_word_frame.columnconfigure(1, weight=0)
- ch_word_frame.columnconfigure(2, weight=0)
- ch_word_frame.columnconfigure(3, weight=1)
+ except Exception:
+ self.handle_exception("生成章节草稿时出错")
+ finally:
+ self.enable_button_safe(self.btn_generate_chapter)
- num_chapters_entry = ctk.CTkEntry(
- ch_word_frame,
- textvariable=self.num_chapters_var,
- width=60,
- font=("Microsoft YaHei", 12)
- )
- num_chapters_entry.grid(row=0, column=0, padx=5, pady=5, sticky="w")
+ threading.Thread(target=task, daemon=True).start()
- word_number_label = ctk.CTkLabel(
- ch_word_frame,
- text="每章字数:",
- font=("Microsoft YaHei", 12)
- )
- word_number_label.grid(row=0, column=1, padx=(15, 5), pady=5, sticky="e")
+ 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")
- word_number_entry = ctk.CTkEntry(
- ch_word_frame,
- textvariable=self.word_number_var,
- width=60,
- font=("Microsoft YaHei", 12)
- )
- word_number_entry.grid(row=0, column=2, padx=5, pady=5, sticky="w")
+ # ------------------ Step4: 定稿章节 ------------------
+ def finalize_chapter_ui(self):
+ filepath = self.filepath_var.get().strip()
+ if not filepath:
+ messagebox.showwarning("警告", "请先配置保存文件路径。")
+ return
- # 保存路径
- filepath_label = ctk.CTkLabel(
- self.params_frame,
- text="保存路径:",
- font=("Microsoft YaHei", 12)
- )
- filepath_label.grid(row=3, column=0, padx=5, pady=5, sticky="e")
+ def task():
+ self.disable_button_safe(self.btn_finalize_chapter)
+ 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()
- self.filepath_frame = ctk.CTkFrame(self.params_frame)
- self.filepath_frame.grid(row=3, column=1, padx=5, pady=5, sticky="nsew")
- self.filepath_frame.columnconfigure(0, weight=1)
+ embedding_model_name = self.embedding_model_name_var.get().strip()
- 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")
+ chap_num = self.safe_get_int(self.chapter_num_var, 1)
+ word_number = self.safe_get_int(self.word_number_var, 3000)
- # 章节号
- chapter_num_label = ctk.CTkLabel(
- self.params_frame,
- text="章节号:",
- font=("Microsoft YaHei", 12)
- )
- chapter_num_label.grid(row=4, column=0, padx=5, pady=5, sticky="e")
- chapter_num_entry = ctk.CTkEntry(
- self.params_frame,
- textvariable=self.chapter_num_var,
- width=80,
- font=("Microsoft YaHei", 12)
- )
- chapter_num_entry.grid(row=4, column=1, padx=5, pady=5, sticky="w")
+ self.safe_log(f"开始定稿第{chap_num}章...")
+ # 先保存用户在左侧编辑框中的修改
+ chapters_dir = os.path.join(filepath, "chapters")
+ os.makedirs(chapters_dir, exist_ok=True)
+ chapter_file = os.path.join(chapters_dir, f"chapter_{chap_num}.txt")
+ edited_text = self.chapter_result.get("0.0", "end").strip()
+ clear_file_content(chapter_file)
+ save_string_to_txt(edited_text, chapter_file)
- # 用户指导
- guide_label = ctk.CTkLabel(
- self.params_frame,
- text="本章指导:",
- font=("Microsoft YaHei", 12)
- )
- guide_label.grid(row=5, column=0, padx=5, pady=5, sticky="ne")
- self.user_guide_text = ctk.CTkTextbox(
- self.params_frame,
- width=200,
- height=80,
- wrap="word",
- font=("Microsoft YaHei", 12)
- )
- self.user_guide_text.grid(row=5, column=1, padx=5, pady=5, sticky="nsew")
+ 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_model_name=embedding_model_name
+ )
+ self.safe_log(f"✅ 第{chap_num}章定稿完成（已更新全局摘要、角色状态、向量库）。")
- # ------------------ 其他Tab的构建 ------------------
+ final_text = read_file(chapter_file)
+ self.master.after(0, lambda: self.show_chapter_in_textbox(final_text))
+
+ except Exception:
+ self.handle_exception("定稿章节时出错")
+ finally:
+ self.enable_button_safe(self.btn_finalize_chapter)
+
+ threading.Thread(target=task, daemon=True).start()
+
+ # ------------------ 一致性审校 ------------------
+ def do_consistency_check(self):
+ filepath = self.filepath_var.get().strip()
+ if not filepath:
+ messagebox.showwarning("警告", "请先配置保存文件路径。")
+ return
+
+ def task():
+ self.disable_button_safe(self.btn_check_consistency)
+ try:
+ api_key = self.api_key_var.get().strip()
+ base_url = self.base_url_var.get().strip()
+ model_name = self.model_name_var.get().strip()
+ temperature = self.temperature_var.get()
+
+ chap_num = self.safe_get_int(self.chapter_num_var, 1)
+ chap_file = os.path.join(filepath, "chapters", f"chapter_{chap_num}.txt")
+ chapter_text = read_file(chap_file)
+
+ if not chapter_text.strip():
+ self.safe_log("⚠️ 当前章节文件为空或不存在，无法审校。")
+ return
+
+ self.safe_log("开始一致性审校...")
+ result = check_consistency(
+ novel_setting="", # 如果需要，可传入最新的 Novel_architecture 内容
+ character_state=read_file(os.path.join(filepath, "character_state.txt")),
+ global_summary=read_file(os.path.join(filepath, "global_summary.txt")),
+ chapter_text=chapter_text,
+ api_key=api_key,
+ base_url=base_url,
+ model_name=model_name,
+ temperature=temperature,
+ plot_arcs=""
+ )
+ self.safe_log("审校结果：")
+ self.safe_log(result)
+
+ except Exception:
+ self.handle_exception("审校时出错")
+ finally:
+ self.enable_button_safe(self.btn_check_consistency)
+
+ threading.Thread(target=task, daemon=True).start()
+
+ def import_knowledge_handler(self):
+ selected_file = filedialog.askopenfilename(
+ title="选择要导入的知识库文件",
+ filetypes=[("Text Files", "*.txt"), ("All Files", "*.*")]
+ )
+ if selected_file:
+ def task():
+ self.disable_button_safe(self.btn_import_knowledge)
+ try:
+ 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()
+ embedding_model_name = self.embedding_model_name_var.get().strip()
+
+ self.safe_log(f"开始导入知识库文件: {selected_file}")
+ import_knowledge_file(
+ api_key=api_key,
+ base_url=base_url,
+ interface_format=interface_format,
+ embedding_model_name=embedding_model_name,
+ file_path=selected_file,
+ embedding_base_url=base_url,
+ filepath=self.filepath_var.get().strip()
+ )
+ self.safe_log("✅ 知识库文件导入完成。")
+ except Exception:
+ self.handle_exception("导入知识库时出错")
+ finally:
+ self.enable_button_safe(self.btn_import_knowledge)
+
+ threading.Thread(target=task, daemon=True).start()
+
+ def clear_vectorstore_handler(self):
+ filepath = self.filepath_var.get().strip()
+ if not filepath:
+ messagebox.showwarning("警告", "请先配置保存文件路径。")
+ return
+
+ first_confirm = messagebox.askyesno("警告", "确定要清空本地向量库吗？此操作不可恢复！")
+ if first_confirm:
+ second_confirm = messagebox.askyesno("二次确认", "你确定真的要删除所有向量数据吗？此操作不可恢复！")
+ if second_confirm:
+ if clear_vector_store(filepath):
+ self.log("已清空向量库。")
+ else:
+ self.log(f"未能清空向量库，请关闭程序后手动删除 {filepath} 下的 vectorstore 文件夹。")
+
+ def show_plot_arcs_ui(self):
+ filepath = self.filepath_var.get().strip()
+ if not filepath:
+ messagebox.showwarning("警告", "请先在主Tab中设置保存文件路径")
+ return
+
+ plot_arcs_file = os.path.join(filepath, "plot_arcs.txt")
+ if not os.path.exists(plot_arcs_file):
+ messagebox.showinfo("剧情要点", "当前还未生成任何剧情要点或冲突记录。")
+ return
+
+ arcs_text = read_file(plot_arcs_file).strip()
+ if not arcs_text:
+ arcs_text = "当前没有记录的剧情要点或冲突。"
+
+ top = ctk.CTkToplevel(self.master)
+ top.title("剧情要点/未解决冲突")
+ top.geometry("600x400")
+
+ text_area = ctk.CTkTextbox(top, wrap="word", font=("Microsoft YaHei", 12))
+ text_area.pack(fill="both", expand=True, padx=10, pady=10)
+
+ text_area.insert("0.0", arcs_text)
+ text_area.configure(state="disabled")
+
+ # ------------------ 其他标签页: Novel Architecture, Chapter Blueprint, Character State, Summary ------------------
def build_setting_tab(self):
self.setting_tab.rowconfigure(0, weight=0)
self.setting_tab.rowconfigure(1, weight=1)
@@ -672,8 +825,8 @@ class NovelGeneratorGUI:
load_btn = ctk.CTkButton(
self.setting_tab,
- text="加载 Novel_setting.txt",
- command=self.load_novel_setting,
+ 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")
@@ -681,18 +834,36 @@ class NovelGeneratorGUI:
save_btn = ctk.CTkButton(
self.setting_tab,
text="保存修改",
- command=self.save_novel_setting,
+ command=self.save_novel_architecture,
font=("Microsoft YaHei", 12)
)
save_btn.grid(row=0, column=0, padx=5, pady=5, sticky="e")
- self.setting_text = ctk.CTkTextbox(
- self.setting_tab,
- wrap="word",
- font=("Microsoft YaHei", 12)
- )
+ self.setting_text = ctk.CTkTextbox(self.setting_tab, wrap="word", font=("Microsoft YaHei", 12))
self.setting_text.grid(row=1, column=0, sticky="nsew", padx=5, pady=5)
+ def load_novel_architecture(self):
+ filepath = self.filepath_var.get().strip()
+ if not filepath:
+ messagebox.showwarning("警告", "请先设置保存文件路径")
+ return
+ filename = os.path.join(filepath, "Novel_architecture.txt")
+ content = read_file(filename)
+ self.setting_text.delete("0.0", "end")
+ self.setting_text.insert("0.0", content)
+ self.log("已加载 Novel_architecture.txt 内容到编辑区。")
+
+ def save_novel_architecture(self):
+ filepath = self.filepath_var.get().strip()
+ if not filepath:
+ messagebox.showwarning("警告", "请先设置保存文件路径")
+ return
+ content = self.setting_text.get("0.0", "end").strip()
+ filename = os.path.join(filepath, "Novel_architecture.txt")
+ clear_file_content(filename)
+ save_string_to_txt(content, filename)
+ self.log("已保存对 Novel_architecture.txt 的修改。")
+
def build_directory_tab(self):
self.directory_tab.rowconfigure(0, weight=0)
self.directory_tab.rowconfigure(1, weight=1)
@@ -701,7 +872,7 @@ class NovelGeneratorGUI:
load_btn = ctk.CTkButton(
self.directory_tab,
text="加载 Novel_directory.txt",
- command=self.load_novel_directory,
+ command=self.load_chapter_blueprint,
font=("Microsoft YaHei", 12)
)
load_btn.grid(row=0, column=0, padx=5, pady=5, sticky="w")
@@ -709,18 +880,36 @@ class NovelGeneratorGUI:
save_btn = ctk.CTkButton(
self.directory_tab,
text="保存修改",
- command=self.save_novel_directory,
+ command=self.save_chapter_blueprint,
font=("Microsoft YaHei", 12)
)
save_btn.grid(row=0, column=0, padx=5, pady=5, sticky="e")
- self.directory_text = ctk.CTkTextbox(
- self.directory_tab,
- wrap="word",
- font=("Microsoft YaHei", 12)
- )
+ self.directory_text = ctk.CTkTextbox(self.directory_tab, wrap="word", font=("Microsoft YaHei", 12))
self.directory_text.grid(row=1, column=0, sticky="nsew", padx=5, pady=5)
+ def load_chapter_blueprint(self):
+ filepath = self.filepath_var.get().strip()
+ if not filepath:
+ messagebox.showwarning("警告", "请先设置保存文件路径")
+ return
+ filename = os.path.join(filepath, "Novel_directory.txt")
+ content = read_file(filename)
+ self.directory_text.delete("0.0", "end")
+ self.directory_text.insert("0.0", content)
+ self.log("已加载 Novel_directory.txt 内容到编辑区。")
+
+ def save_chapter_blueprint(self):
+ filepath = self.filepath_var.get().strip()
+ if not filepath:
+ messagebox.showwarning("警告", "请先设置保存文件路径")
+ return
+ content = self.directory_text.get("0.0", "end").strip()
+ filename = os.path.join(filepath, "Novel_directory.txt")
+ clear_file_content(filename)
+ save_string_to_txt(content, filename)
+ self.log("已保存对 Novel_directory.txt 的修改。")
+
def build_character_tab(self):
self.character_tab.rowconfigure(0, weight=0)
self.character_tab.rowconfigure(1, weight=1)
@@ -742,13 +931,31 @@ class NovelGeneratorGUI:
)
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 = ctk.CTkTextbox(self.character_tab, wrap="word", font=("Microsoft YaHei", 12))
self.character_text.grid(row=1, column=0, sticky="nsew", padx=5, pady=5)
+ def load_character_state(self):
+ filepath = self.filepath_var.get().strip()
+ if not filepath:
+ messagebox.showwarning("警告", "请先设置保存文件路径")
+ return
+ filename = os.path.join(filepath, "character_state.txt")
+ content = read_file(filename)
+ self.character_text.delete("0.0", "end")
+ self.character_text.insert("0.0", content)
+ self.log("已加载 character_state.txt 到编辑区。")
+
+ def save_character_state(self):
+ filepath = self.filepath_var.get().strip()
+ if not filepath:
+ messagebox.showwarning("警告", "请先设置保存文件路径")
+ return
+ content = self.character_text.get("0.0", "end").strip()
+ filename = os.path.join(filepath, "character_state.txt")
+ clear_file_content(filename)
+ save_string_to_txt(content, filename)
+ self.log("已保存对 character_state.txt 的修改。")
+
def build_summary_tab(self):
self.summary_tab.rowconfigure(0, weight=0)
self.summary_tab.rowconfigure(1, weight=1)
@@ -770,17 +977,33 @@ class NovelGeneratorGUI:
)
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 = ctk.CTkTextbox(self.summary_tab, wrap="word", font=("Microsoft YaHei", 12))
self.summary_text.grid(row=1, column=0, sticky="nsew", padx=5, pady=5)
+ def load_global_summary(self):
+ filepath = self.filepath_var.get().strip()
+ if not filepath:
+ messagebox.showwarning("警告", "请先设置保存文件路径")
+ return
+ filename = os.path.join(filepath, "global_summary.txt")
+ content = read_file(filename)
+ self.summary_text.delete("0.0", "end")
+ self.summary_text.insert("0.0", content)
+ self.log("已加载 global_summary.txt 到编辑区。")
+
+ def save_global_summary(self):
+ filepath = self.filepath_var.get().strip()
+ if not filepath:
+ messagebox.showwarning("警告", "请先设置保存文件路径")
+ return
+ content = self.summary_text.get("0.0", "end").strip()
+ filename = os.path.join(filepath, "global_summary.txt")
+ clear_file_content(filename)
+ save_string_to_txt(content, filename)
+ self.log("已保存对 global_summary.txt 的修改。")
+
+ # ------------------ 章节管理标签页 ------------------
def build_chapters_tab(self):
- """
- 新增的 Tab，用于查看、编辑和保存已生成的各章节内容。
- """
self.chapters_view_tab.rowconfigure(0, weight=0)
self.chapters_view_tab.rowconfigure(1, weight=1)
self.chapters_view_tab.columnconfigure(0, weight=1)
@@ -821,7 +1044,6 @@ class NovelGeneratorGUI:
self.chapters_list = []
self.refresh_chapters_list()
- # ------------------ 章节管理辅助方法 ------------------
def refresh_chapters_list(self):
filepath = self.filepath_var.get().strip()
chapters_dir = os.path.join(filepath, "chapters")
@@ -857,7 +1079,6 @@ class NovelGeneratorGUI:
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):
@@ -914,528 +1135,7 @@ class NovelGeneratorGUI:
else:
messagebox.showinfo("提示", "已经是最后一章了。")
- # ------------------ 配置管理 ------------------
- def load_config_btn(self):
- cfg = load_config(self.config_file)
- if cfg:
- # LLM
- self.api_key_var.set(cfg.get("api_key", ""))
- self.base_url_var.set(cfg.get("base_url", ""))
- self.interface_format_var.set(cfg.get("interface_format", "OpenAI"))
- self.model_name_var.set(cfg.get("model_name", ""))
- self.temperature_var.set(cfg.get("temperature", 0.7))
- # Embedding
- self.embedding_api_key_var.set(cfg.get("embedding_api_key", ""))
- self.embedding_interface_format_var.set(cfg.get("embedding_interface_format", "OpenAI"))
- self.embedding_url_var.set(cfg.get("embedding_url", ""))
- self.embedding_model_name_var.set(cfg.get("embedding_model_name", ""))
- # ### CHANGED：用字符串形式设值
- self.embedding_retrieval_k_var.set(str(cfg.get("embedding_retrieval_k", 4)))
-
- # Novel
- self.genre_var.set(cfg.get("genre", ""))
-
- # ### CHANGED：用字符串形式设值
- self.num_chapters_var.set(str(cfg.get("num_chapters", 10)))
- self.word_number_var.set(str(cfg.get("word_number", 3000)))
- self.filepath_var.set(cfg.get("filepath", ""))
-
- topic_value = cfg.get("topic", "")
- self.topic_text.delete("0.0", "end")
- self.topic_text.insert("0.0", topic_value)
-
- self.log("已加载配置。")
- else:
- messagebox.showwarning("提示", "未找到或无法读取配置文件。")
-
- def save_config_btn(self):
- config_data = {
- # LLM
- "api_key": self.api_key_var.get(),
- "base_url": self.base_url_var.get(),
- "interface_format": self.interface_format_var.get(),
- "model_name": self.model_name_var.get(),
- "temperature": self.temperature_var.get(),
-
- # Embedding
- "embedding_api_key": self.embedding_api_key_var.get(),
- "embedding_interface_format": self.embedding_interface_format_var.get(),
- "embedding_url": self.embedding_url_var.get(),
- "embedding_model_name": self.embedding_model_name_var.get(),
- "embedding_retrieval_k": self.safe_get_int(self.embedding_retrieval_k_var, 4),
-
- # Novel
- "topic": self.topic_text.get("0.0", "end").strip(),
- "genre": self.genre_var.get(),
- "num_chapters": self.safe_get_int(self.num_chapters_var, 10),
- "word_number": self.safe_get_int(self.word_number_var, 3000),
- "filepath": self.filepath_var.get()
- }
- if save_config(config_data, self.config_file):
- messagebox.showinfo("提示", "配置已保存至 config.json")
- self.log("配置已保存。")
- else:
- messagebox.showerror("错误", "保存配置失败。")
-
- def browse_folder(self):
- selected_dir = filedialog.askdirectory()
- if selected_dir:
- self.filepath_var.set(selected_dir)
-
- # ------------------ 日志输出（主线程安全） ------------------
- def log(self, message: str):
- self.log_text.configure(state="normal")
- self.log_text.insert("end", message + "\n")
- self.log_text.see("end")
- self.log_text.configure(state="disabled")
-
- def safe_log(self, message: str):
- self.master.after(0, lambda: self.log(message))
-
- def disable_button_safe(self, btn):
- self.master.after(0, lambda: btn.configure(state="disabled"))
-
- def enable_button_safe(self, btn):
- self.master.after(0, lambda: btn.configure(state="normal"))
-
- def handle_exception(self, context: str):
- full_message = f"{context}\n{traceback.format_exc()}"
- logging.error(full_message)
- self.safe_log(full_message)
-
- # ------------------ 分步操作：生成设定、目录、章节草稿、定稿 ------------------
- def generate_novel_setting_ui(self):
- """Step1. 生成小说设定（Novel_setting.txt）"""
- filepath = self.filepath_var.get().strip()
- if not filepath:
- messagebox.showwarning("警告", "请先选择保存文件路径")
- return
-
- def task():
- self.disable_button_safe(self.btn_generate_setting)
- 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()
-
- 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)
- temperature = self.temperature_var.get()
-
- self.safe_log("开始生成小说设定...")
- Novel_setting_generate(
- 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
- )
- self.safe_log("✅ 小说设定生成完成。请在 'Novel Settings' 标签页进行查看或编辑。")
- except Exception:
- self.handle_exception("生成小说设定时出错")
- finally:
- self.enable_button_safe(self.btn_generate_setting)
-
- threading.Thread(target=task, daemon=True).start()
-
- def generate_novel_directory_ui(self):
- """Step2. 基于已有 Novel_setting.txt 生成 Novel_directory.txt"""
- filepath = self.filepath_var.get().strip()
- if not filepath:
- messagebox.showwarning("警告", "请先选择保存文件路径")
- return
-
- def task():
- self.disable_button_safe(self.btn_generate_directory)
- 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()
- num_chapters = self.safe_get_int(self.num_chapters_var, 10)
- temperature = self.temperature_var.get()
-
- self.safe_log("开始生成小说目录...")
- Novel_directory_generate(
- api_key=api_key,
- base_url=base_url,
- llm_model=model_name,
- number_of_chapters=num_chapters,
- filepath=filepath,
- temperature=temperature
- )
- self.safe_log("✅ 小说目录生成完成。请在 'Novel Directory' 标签页查看或编辑。")
- 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):
- """Step3. 生成当前章节草稿"""
- filepath = self.filepath_var.get().strip()
- if not filepath:
- messagebox.showwarning("警告", "请先配置保存文件路径。")
- return
-
- def task():
- self.disable_button_safe(self.btn_generate_chapter)
- 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()
-
- novel_settings_file = os.path.join(filepath, "Novel_setting.txt")
- novel_settings = read_file(novel_settings_file)
- if not novel_settings.strip():
- self.safe_log("⚠️ 未找到 Novel_setting.txt，请先生成设定。")
- return
-
- character_state_file = os.path.join(filepath, "character_state.txt")
- character_state = read_file(character_state_file)
- global_summary_file = os.path.join(filepath, "global_summary.txt")
- global_summary = read_file(global_summary_file)
- novel_directory_file = os.path.join(filepath, "Novel_directory.txt")
- novel_directory = read_file(novel_directory_file)
-
- 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()
-
- # 获取最近3章文本
- chapters_dir = os.path.join(filepath, "chapters")
- recent_3_texts = get_last_n_chapters_text(chapters_dir, chap_num, n=3)
-
- # 生成最近章节摘要
- recent_chapters_summary = summarize_recent_chapters(
- llm_model=model_name,
- api_key=api_key,
- base_url=base_url,
- temperature=temperature,
- chapters_text_list=recent_3_texts
- )
-
- self.safe_log(f"开始生成第{chap_num}章草稿...")
- draft_text = generate_chapter_draft(
- novel_settings=novel_settings,
- global_summary=global_summary,
- character_state=character_state,
- recent_chapters_summary=recent_chapters_summary,
- user_guidance=user_guidance,
- api_key=api_key,
- base_url=base_url,
- model_name=model_name,
- novel_number=chap_num,
- word_number=word_number,
- temperature=temperature,
- novel_novel_directory=novel_directory,
- filepath=filepath,
-
- # Embedding 配置
- interface_format=self.embedding_interface_format_var.get().strip(),
- embedding_model_name=self.embedding_model_name_var.get().strip(),
- embedding_base_url=self.embedding_url_var.get().strip(),
- # 新增：检索 K 值
- embedding_retrieval_k=self.safe_get_int(self.embedding_retrieval_k_var, 4)
- )
- if draft_text:
- self.safe_log(f"✅ 第{chap_num}章草稿生成完成。请在左侧查看或编辑。")
- self.master.after(0, lambda: self.show_chapter_in_textbox(draft_text))
- else:
- self.safe_log("⚠️ 本章草稿生成失败或无内容。")
-
- except Exception:
- self.handle_exception("生成章节草稿时出错")
- finally:
- self.enable_button_safe(self.btn_generate_chapter)
-
- threading.Thread(target=task, daemon=True).start()
-
- def show_chapter_in_textbox(self, text: str):
- self.chapter_result.delete("0.0", "end")
- self.chapter_result.insert("0.0", text)
- self.chapter_result.see("end")
-
- def finalize_chapter_ui(self):
- """Step4. 定稿当前章节：更新全局摘要、角色状态、向量库等"""
- filepath = self.filepath_var.get().strip()
- if not filepath:
- messagebox.showwarning("警告", "请先配置保存文件路径。")
- return
-
- def task():
- self.disable_button_safe(self.btn_finalize_chapter)
- 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.embedding_interface_format_var.get().strip()
- embedding_model_name = self.embedding_model_name_var.get().strip()
- embedding_base_url = self.embedding_url_var.get().strip()
- embedding_api_key = self.embedding_api_key_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")
- chapter_file = os.path.join(chapters_dir, f"chapter_{chap_num}.txt")
- edited_text = self.chapter_result.get("0.0", "end").strip()
- clear_file_content(chapter_file)
- save_string_to_txt(edited_text, chapter_file)
-
- finalize_chapter(
- novel_number=chap_num,
- word_number=word_number,
- api_key=api_key,
- base_url=base_url,
- interface_format=interface_format,
- embedding_model_name=embedding_model_name,
- model_name=model_name,
- temperature=temperature,
- filepath=filepath,
- embedding_base_url=embedding_base_url,
- embedding_api_key=embedding_api_key
- )
- self.safe_log(f"✅ 第{chap_num}章定稿完成（已更新全局摘要、角色状态、剧情要点、向量库）。")
-
- # 读取定稿后的文本显示
- chap_file = os.path.join(filepath, "chapters", f"chapter_{chap_num}.txt")
- final_text = read_file(chap_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):
- """使用审校Agent对最新章节进行简单一致性或冲突检查"""
- 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()
-
- 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")
-
- 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)
-
- 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=novel_setting,
- character_state=character_state,
- global_summary=global_summary,
- chapter_text=chapter_text,
- api_key=api_key,
- base_url=base_url,
- model_name=model_name,
- temperature=temperature,
- plot_arcs=plot_arcs
- )
- self.safe_log("审校结果：")
- self.safe_log(result)
-
- except Exception:
- self.handle_exception("审校时出错")
- finally:
- self.enable_button_safe(self.btn_check_consistency)
-
- threading.Thread(target=task, daemon=True).start()
-
- # ------------------ 导入知识库/清空向量库/查看剧情要点 ------------------
- def import_knowledge_handler(self):
- selected_file = filedialog.askopenfilename(
- title="选择要导入的知识库文件",
- filetypes=[("Text Files", "*.txt"), ("All Files", "*.*")]
- )
- if selected_file:
- def task():
- self.disable_button_safe(self.btn_import_knowledge)
- try:
- self.safe_log(f"开始导入知识库文件: {selected_file}")
- import_knowledge_file(
- 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(),
- embedding_model_name=self.embedding_model_name_var.get().strip(),
- file_path=selected_file,
- embedding_base_url=self.embedding_url_var.get().strip(),
- filepath=self.filepath_var.get().strip()
- )
- self.safe_log("✅ 知识库文件导入完成。")
- except Exception:
- self.handle_exception("导入知识库时出错")
- finally:
- self.enable_button_safe(self.btn_import_knowledge)
-
- threading.Thread(target=task, daemon=True).start()
-
- def clear_vectorstore_handler(self):
- filepath = self.filepath_var.get().strip()
- if not filepath:
- messagebox.showwarning("警告", "请先配置保存文件路径。")
- return
-
- first_confirm = messagebox.askyesno("警告", "确定要清空本地向量库吗？此操作不可恢复！")
- if first_confirm:
- second_confirm = messagebox.askyesno("二次确认", "你确定真的要删除所有向量数据吗？此操作不可恢复！")
- if second_confirm:
- if clear_vector_store(filepath):
- self.log("已清空向量库。")
- else:
- self.log(f"未能清空向量库。请关闭程序后前往{filepath}文件夹手动删除vectorstore目录。")
-
- def show_plot_arcs_ui(self):
- filepath = self.filepath_var.get().strip()
- if not filepath:
- messagebox.showwarning("警告", "请先在主Tab中设置保存文件路径")
- return
-
- plot_arcs_file = os.path.join(filepath, "plot_arcs.txt")
- if not os.path.exists(plot_arcs_file):
- messagebox.showinfo("剧情要点", "当前还未生成任何剧情要点或未解决冲突。")
- return
-
- arcs_text = read_file(plot_arcs_file).strip()
- if not arcs_text:
- arcs_text = "当前没有记录的剧情要点或冲突。"
-
- top = ctk.CTkToplevel(self.master)
- top.title("剧情要点/未解决冲突")
- top.geometry("600x400")
-
- text_area = ctk.CTkTextbox(top, wrap="word", font=("Microsoft YaHei", 12))
- text_area.pack(fill="both", expand=True, padx=10, pady=10)
-
- text_area.insert("0.0", arcs_text)
- text_area.configure(state="disabled")
-
- # ------------------ Novel Settings/Directory/Character/Global Summary 的加载与保存 ------------------
- 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")
- clear_file_content(setting_file)
- save_string_to_txt(content, setting_file)
- self.log("已保存对 Novel_setting.txt 的修改。")
-
- 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")
- clear_file_content(directory_file)
- save_string_to_txt(content, directory_file)
- self.log("已保存对 Novel_directory.txt 的修改。")
-
- 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")
- clear_file_content(char_file)
- save_string_to_txt(content, char_file)
- self.log("已保存对 character_state.txt 的修改。")
-
- 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")
- clear_file_content(summary_file)
- save_string_to_txt(content, summary_file)
- self.log("已保存对 global_summary.txt 的修改。")
-
-
-# 入口
if __name__ == "__main__":
app = ctk.CTk()
gui = NovelGeneratorGUI(app)