记录一下,便于待会核对更改

This commit is contained in:
YILING0013
2025-02-02 11:55:49 +08:00
parent eb627fbec9
commit bcd13b8ca3
6 changed files with 664 additions and 193 deletions
+101
View File
@@ -0,0 +1,101 @@
第1章 :星河坠落提瓦特 - 穿越与初遇派蒙,危机初现端倪
第2章 :蒙德的初印象 - 初探蒙德城,与丽莎的首次相遇
第3章 :酒馆的闲谈 - 了解提瓦特局势,派蒙展现惊人一面
第4章 :丽莎的试探 - 星河展现科技实力,引来丽莎的警惕
第5章 :委托与冒险 - 接受委托,初次展现星穹铁道科技的运用
第6章 :风魔龙的挑战 - 运用科技对抗风魔龙,展现策略与智慧
第7章 :派蒙的异常 - 派蒙出现记忆碎片,暗示其不寻常身份
第8章 :丽莎的秘密魔法 - 丽莎使用禁忌魔法,展现其神秘一面
第9章 :旅行者的出现 - 与原神旅行者初次相遇,产生摩擦与合作
第10章 :蒙德的阴谋 - 发现蒙德隐藏的危机,与丽莎深入合作
第11章 :追捕令的阴影 - 星河收到星穹铁道追捕令,内心挣扎
第12章 :深入调查 - 调查蒙德的危机,发现与丽莎的秘密组织有关
第13章 :钟离的现身 - 钟离初次出现,对星河进行暗示与引导
第14章 :失落科技的线索 - 发现关于提瓦特失落科技的线索
第15章 :前往璃月 - 踏上前往璃月的旅程,面对新的挑战
第16章 :璃月港的繁华 - 体验璃月港的文化,了解璃月局势
第17章 :往生堂的委托 - 接受往生堂的委托,深入了解璃月历史
第18章 :岩王帝君的考验 - 钟离对星河进行考验,揭示其真实目的
第19章 :地脉异常 - 发现璃月地脉异常,与旅行者合作调查
第20章 :黄金屋的秘密 - 探索黄金屋的秘密,发现与失落科技有关
第21章 :刻晴的质疑 - 刻晴对星河的科技持有怀疑态度
第22章 :对抗岩龙蜥 - 运用科技与策略对抗强大的岩龙蜥
第23章 :派蒙的回忆 - 派蒙回忆片段增多,身份逐渐清晰
第24章 :丽莎的引导 - 丽莎引导星河学习提瓦特魔法
第25章 :星河的抉择 - 星河面临道德困境,做出艰难选择
第26章 :前往稻妻 - 前往稻妻,面对新的挑战与危机
第27章 :稻妻的雷电 - 体验稻妻的文化与习俗,对抗雷电将军的威压
第28章 :八重神子的试探 - 八重神子对星河进行试探,考验其能力
第29章 :影向地的秘密 - 探索影向地的秘密,发现与提瓦特历史有关
第30章 :反抗军与合作 - 与稻妻反抗军合作,对抗幕后黑手
第31章 :永恒的执政官 - 与雷电将军正面交锋,展现智慧与实力
第32章 :派蒙的过去 - 派蒙的过去逐渐揭开,与提瓦特古老文明有关
第33章 :丽莎的组织 - 丽莎所属组织的真相逐渐浮出水面
第34章 :前往须弥 - 前往须弥,面对新的挑战与文化
第35章 :须弥的智慧 - 体验须弥的文化,了解须弥的智慧与科技
第36章 :草神的大智慧 - 与草神进行交流,探讨生命的意义
第37章 :森林的危机 - 须弥森林面临危机,星河运用科技解决问题
第38章 :沙漠的秘密 - 探索须弥沙漠的秘密,发现远古文明的遗迹
第39章 :散兵的阴谋 - 散兵的阴谋逐渐浮出水面,星河需要应对
第40章 :与散兵的对抗 - 与散兵展开对抗,展现星穹铁道科技的优势
第41章 :前往枫丹 - 前往枫丹,面对新的挑战与科技
第42章 :枫丹的审判 - 体验枫丹的法律与秩序,面对复杂的案件
第43章 :水神的神秘 - 与水神进行互动,了解枫丹的秘密
第44章 :机械岛的危机 - 枫丹机械岛面临危机,星河运用科技解决问题
第45章 :潜入深海 - 潜入深海,探索深海的秘密与科技
第46章 :派蒙的预言 - 派蒙预言即将到来的巨大危机
第47章 :丽莎的抉择 - 丽莎面临艰难的抉择,需要做出牺牲
第48章 :前往纳塔 - 前往纳塔,面对新的挑战与文化
第49章 :纳塔的冰雪 - 体验纳塔的严寒环境,与冰雪元素对抗
第50章 :雪山之巅的秘密 - 探索雪山之巅的秘密,发现远古文明的遗迹
第51章 :冰神的考验 - 冰神对星河进行考验,考验其心性与毅力
第52章 :与旅行者的合作 - 与旅行者深入合作,对抗共同的敌人
第53章 :深渊的威胁 - 深渊的威胁日益逼近,星河需要寻找应对之策
第54章 :钟离的计划 - 钟离的计划逐渐显露,其真实目的更加清晰
第55章 :星河的过去 - 星河过去的阴影再次浮现,影响其与他人的关系
第56章 :派蒙的觉醒 - 派蒙觉醒,展现其强大的力量
第57章 :丽莎的牺牲 - 丽莎为了保护星河做出牺牲
第58章 :前往至冬 - 前往至冬,面对新的挑战与阴谋
第59章 :至冬的阴谋 - 揭露至冬的阴谋,对抗幕后黑手
第60章 :女皇的算计 - 女皇对星河进行算计,考验其智慧与勇气
第61章 :机械生命的危机 - 至冬的机械生命失控,引发巨大的危机
第62章 :星河的策略 - 星河运用策略,化解机械生命的危机
第63章 :派蒙的秘密力量 - 派蒙展现其强大的秘密力量
第64章 :丽莎的遗产 - 丽莎留下宝贵的遗产,帮助星河对抗危机
第65章 :对抗深渊使徒 - 对抗强大的深渊使徒,展现星穹铁道科技的威力
第66章 :旅行者的选择 - 旅行者面临艰难的选择,需要做出决定
第67章 :钟离的帮助 - 钟离给予星河关键的帮助,但其目的依旧不明
第68章 :星河的成长 - 星河在经历种种磨难后得到成长
第69章 :提瓦特的命运 - 提瓦特的命运掌握在星河手中
第70章 :最终决战的序幕 - 最终决战的序幕拉开,危机全面爆发
第71章 :与幕后黑手的对抗 - 与幕后黑手展开激烈的对抗
第72章 :星河的牺牲 - 星河为了拯救提瓦特做出牺牲
第73章 :派蒙的付出 - 派蒙为了帮助星河付出一切
第74章 :丽莎的回归 - 丽莎以另一种方式回归,继续守护蒙德
第75章 :旅行者的抉择 - 旅行者做出最终的抉择
第76章 :钟离的真相 - 钟离的真实目的最终揭晓
第77章 :星河的复活 - 星河以意想不到的方式复活
第78章 :新的危机 - 新的危机出现,提瓦特面临新的挑战
第79章 :星河的抉择 - 星河需要做出新的抉择,选择自己的未来
第80章 :派蒙的未来 - 派蒙的未来充满希望与未知
第81章 :丽莎的守护 - 丽莎继续守护蒙德,为提瓦特贡献力量
第82章 :旅行者的旅程 - 旅行者继续自己的旅程,探索提瓦特的奥秘
第83章 :钟离的隐退 - 钟离选择隐退,默默守护提瓦特
第84章 :星河的归宿 - 星河思考自己的归宿,是留下还是离开
第85章 :与伙伴的告别 - 与伙伴们告别,回忆共同经历的冒险
第86章 :返回星穹铁道? - 星河考虑返回星穹铁道,但心中仍有牵挂
第87章 :提瓦特的未来 - 提瓦特的未来充满希望与挑战
第88章 :科技与魔法的融合 - 星穹铁道科技与提瓦特魔法开始融合
第89章 :新的时代 - 提瓦特进入新的时代,科技与魔法共存
第90章 :星河的遗产 - 星河留下宝贵的遗产,影响着提瓦特的未来
第91章 :派蒙的传承 - 派蒙将自己的智慧与经验传承下去
第92章 :丽莎的教导 - 丽莎继续教导年轻一代,传承魔法的艺术
第93章 :旅行者的故事 - 旅行者的故事继续,探索更多未知的领域
第94章 :钟离的箴言 - 钟离留下意味深长的箴言,引导提瓦特的未来
第95章 :星河的回归 - 星河最终做出决定,选择自己的归宿
第96章 :告别的时刻 - 与提瓦特的伙伴们告别,心中充满不舍
第97章 :新的旅程 - 星河踏上新的旅程,迎接新的挑战
第98章 :回首往昔 - 回首在提瓦特的冒险,感慨万千
第99章 :未来的希望 - 提瓦特的未来充满希望,光明与挑战并存
第100章 :星河挽歌 - 故事的结尾,留下无限的遐想,开放式结局
+33
View File
@@ -0,0 +1,33 @@
小说设定:《星河挽歌:提瓦特星轨》
小说类型:玄幻穿越,史诗级冒险,轻松幽默与细腻情感并存,整体偏正剧,但避免过于沉重。
字数:约400万字(100章 x 4000字/章)
视角:第三人称,穿插主角第一人称内心独白。
世界观:基于原神世界线平行宇宙的提瓦特大陆,地理环境与原神基本一致,但部分区域势力分布及历史事件有所调整,以适应主角的到来和剧情发展。七国结构保留,但政治格局和社会风貌因主角介入而变。科技与魔法并存,魔法更为显著,主角的星穹铁道科技作为“外来力量”融入其中,引发冲突和变革。重要历史事件保留原神设定,并添加主角引发的新的历史事件。
核心内容梗概:主角星河在星穹铁道执行任务时意外穿越到提瓦特大陆,发现大陆面临史无前例的巨大危机。凭借星穹铁道科技力量及经验,星河成为对抗危机的关键人物。在拯救提瓦特过程中,他将与原神角色相遇,建立深厚情谊或产生冲突,最终完成使命,并决定自己的归宿。采用英雄之旅叙事结构,主线为拯救提瓦特,副线为主角与原神角色的情感纠葛。
主要角色:
1. 星河:坚毅、责任感强、略带孤傲,内心温柔渴望归宿。拥有强大的星穹铁道科技力量,但更倾向于智慧和策略解决问题。拥有丰富战斗经验和冷静判断力,但也有过去任务的阴影和对未来命运的迷茫,偶尔冲动,也可能因自身能力产生优越感。他的目标是拯救提瓦特,找到归宿,弥补过去遗憾。他的隐藏秘密是:在星穹铁道的一次任务中,他导致一位重要伙伴牺牲,这个秘密影响着他与他人的关系,并在与原神角色建立深厚感情时成为关键转折点。
2. 派蒙:活泼、贪吃、略显成熟,更能理解星河内心世界,给予更深层次的支持。积极参与冒险,但鲁莽会带来麻烦,是重要的喜剧元素。对提瓦特知识渊博,但对星穹铁道科技一无所知。她的目标是帮助星河拯救提瓦特,探索自身存在秘密,与星河建立更深厚羁绊。她的隐藏秘密是:她并非简单的导航精灵,拥有远超常人的智慧和感知能力,与提瓦特大陆的过去有着神秘联系。这会在中段通过关键事件逐步揭示。 她的“记忆碎片”——偶尔失神,喃喃自语,对特定场景反应强烈,甚至出现幻觉,这些会逐渐清晰,暗示她与提瓦特古老文明甚至超越提瓦特的力量有关。
3. 丽莎:慵懒、智慧,强大的魔法能力,比原神设定中更成熟、更富有责任感。对星穹铁道科技好奇又警惕,将星河视为竞争对手和值得培养的对象。她会引导主角,但在关键时刻设置考验。她的目标是保护蒙德,培养星河,并寻求突破自身瓶颈。她的隐藏秘密是:她拥有不为人知的秘密身份,与某个神秘组织有关,该组织与即将到来的危机密切相关。她的“禁忌魔法”——偶尔使用威力远超平时能力的魔法,使用后疲惫甚至身体异样,她会刻意隐藏,这与她的神秘组织联系紧密。
4. 原神旅行者:与星河互动,产生冲突与合作。两人对提瓦特大陆的理解不同,可能会产生矛盾,但在对抗危机时携手合作。
5. 钟离:神秘幕后人物,引导星河完成使命,但目的和动机不明,会带来考验和帮助,也可能成为潜在敌人。他的“真实目的”——对星河的帮助看似无私,但始终保持距离,言行举止充满深意,他的行动与提瓦特危机有关,目的可能并非简单帮助主角,而是为了实现某种更宏大的计划。
角色关系与冲突:星河与派蒙(依赖与独立的矛盾),星河与丽莎(竞争与合作),星河与原神旅行者(身份认同冲突),星河与钟离(考验与引导)。
暗线与伏笔:派蒙的记忆碎片,丽莎的禁忌魔法,星河的星穹铁道追捕令,钟离的真实目的,提瓦特大陆的失落科技。这些暗线相互交织,推动剧情发展,并为角色成长和故事走向增添可能性和悬念。
结局:开放式结局,主角战胜危机,但提瓦特仍面临挑战,主角可以选择留在提瓦特或回到星穹铁道;也可能根据剧情走向相对圆满结局。
世界观与角色动机结合:提瓦特大陆面临的危机是推动星河行动的主要动力,而他过去的经历(星穹铁道任务的阴影和伙伴的牺牲)以及对归宿的渴望,则构成了他内心的冲突和驱动力。派蒙的秘密与提瓦特古老文明的关联,丽莎的禁忌魔法与神秘组织的联系,这些都与即将到来的危机息息相关,也与主角的命运紧密相连。 钟离的真实目的则成为贯穿始终的神秘力量,推动剧情发展,并最终影响故事的结局。 星河的星穹铁道科技,则成为改变提瓦特魔法世界格局的关键因素,引发一系列冲突和变革。
+45
View File
@@ -0,0 +1,45 @@
# embedding_ollama.py
import requests
from typing import List
class OllamaEmbeddings:
"""
Ollama 本地服务提供 /api/embeddings 接口,响应中包含 {"embedding": [...]}。
"""
def __init__(self, model_name: str, base_url: str):
self.model_name = model_name
self.base_url = base_url
def embed_documents(self, texts: List[str]) -> List[List[float]]:
"""
将多段文本转换为向量列表
"""
embeddings = []
for text in texts:
emb = self.embed_single_document(text)
embeddings.append(emb)
return embeddings
def embed_query(self, query: str) -> List[float]:
"""
将单条 query 转换为 embedding 向量
"""
return self.embed_single_document(query)
def embed_single_document(self, text: str) -> List[float]:
"""
调用 Ollama 本地服务接口,获取文本的 embedding
"""
url = f"{self.base_url}/api/embeddings"
data = {
"model": self.model_name,
"prompt": text
}
try:
response = requests.post(url, json=data)
response.raise_for_status()
result = response.json()
return result["embedding"]
except requests.exceptions.RequestException as e:
raise Exception(f"Ollama embeddings request error: {e}")
+2 -2
View File
@@ -40,7 +40,7 @@ exe = EXE(
a.scripts, a.scripts,
[], [],
exclude_binaries=True, exclude_binaries=True,
name='AI_NovelGenerator_V1.2.3', name='AI_NovelGenerator_V1.2.4',
debug=True, debug=True,
bootloader_ignore_signals=False, bootloader_ignore_signals=False,
strip=False, strip=False,
@@ -60,5 +60,5 @@ coll = COLLECT(
strip=False, strip=False,
upx=True, upx=True,
upx_exclude=[], upx_exclude=[],
name='AI_NovelGenerator_V1.2.3' name='AI_NovelGenerator_V1.2.4'
) )
+26 -27
View File
@@ -51,7 +51,6 @@ if not os.path.exists(VECTOR_STORE_DIR):
def clear_vector_store(): def clear_vector_store():
""" """
清空本地向量库(删除 vectorstore 文件夹内的内容)。 清空本地向量库(删除 vectorstore 文件夹内的内容)。
需要在UI中加一个二次确认弹窗,防止误删。
""" """
if os.path.exists(VECTOR_STORE_DIR): if os.path.exists(VECTOR_STORE_DIR):
try: try:
@@ -68,14 +67,16 @@ def clear_vector_store():
else: else:
logging.info("No vector store found to clear.") logging.info("No vector store found to clear.")
def init_vector_store(api_key: str, base_url: str, texts: List[str]) -> Chroma: def init_vector_store(api_key: str, base_url: str, texts: List[str], embedding_base_url: str = "") -> Chroma:
""" """
初始化并返回一个Chroma向量库,将传入的文本进行嵌入并保存到本地目录。 初始化并返回一个Chroma向量库,将传入的文本进行嵌入并保存到本地目录。
如果不存在该目录,会自动创建。 如果不存在该目录,会自动创建。
如果 embedding_base_url 不为空,则使用它做为embedding的base,否则默认base_url。
""" """
embed_url = embedding_base_url if embedding_base_url else base_url
embeddings = OpenAIEmbeddings( embeddings = OpenAIEmbeddings(
openai_api_key=api_key, openai_api_key=api_key,
openai_api_base=base_url openai_api_base=embed_url
) )
documents = [Document(page_content=t) for t in texts] documents = [Document(page_content=t) for t in texts]
vectorstore = Chroma.from_documents( vectorstore = Chroma.from_documents(
@@ -86,34 +87,38 @@ def init_vector_store(api_key: str, base_url: str, texts: List[str]) -> Chroma:
vectorstore.persist() vectorstore.persist()
return vectorstore return vectorstore
def load_vector_store(api_key: str, base_url: str) -> Optional[Chroma]: def load_vector_store(api_key: str, base_url: str, embedding_base_url: str = "") -> Optional[Chroma]:
"""读取已存在的向量库。若不存在则返回 None。""" """
读取已存在的向量库。若不存在则返回 None。
同样支持可选的 embedding_base_url。
"""
if not os.path.exists(VECTOR_STORE_DIR): if not os.path.exists(VECTOR_STORE_DIR):
return None return None
embed_url = embedding_base_url if embedding_base_url else base_url
embeddings = OpenAIEmbeddings( embeddings = OpenAIEmbeddings(
openai_api_key=api_key, openai_api_key=api_key,
openai_api_base=base_url openai_api_base=embed_url
) )
return Chroma(persist_directory=VECTOR_STORE_DIR, embedding_function=embeddings) return Chroma(persist_directory=VECTOR_STORE_DIR, embedding_function=embeddings)
def update_vector_store(api_key: str, base_url: str, new_chapter: str) -> None: def update_vector_store(api_key: str, base_url: str, new_chapter: str, embedding_base_url: str = "") -> None:
"""将最新章节文本插入到向量库里,用于后续检索参考。若库不存在则初始化。""" """将最新章节文本插入到向量库里,用于后续检索参考。若库不存在则初始化。"""
store = load_vector_store(api_key, base_url) store = load_vector_store(api_key, base_url, embedding_base_url)
if not store: if not store:
logging.info("Vector store does not exist. Initializing a new one...") logging.info("Vector store does not exist. Initializing a new one...")
init_vector_store(api_key, base_url, [new_chapter]) init_vector_store(api_key, base_url, [new_chapter], embedding_base_url)
return return
new_doc = Document(page_content=new_chapter) new_doc = Document(page_content=new_chapter)
store.add_documents([new_doc]) store.add_documents([new_doc])
store.persist() store.persist()
def get_relevant_context_from_vector_store(api_key: str, base_url: str, query: str, k: int = 2) -> str: def get_relevant_context_from_vector_store(api_key: str, base_url: str, query: str, k: int = 2, embedding_base_url: str = "") -> str:
""" """
从向量库中检索与 query 最相关的 k 条文本,拼接后返回。 从向量库中检索与 query 最相关的 k 条文本,拼接后返回。
若向量库不存在则返回空字符串。 若向量库不存在则返回空字符串。
""" """
store = load_vector_store(api_key, base_url) store = load_vector_store(api_key, base_url, embedding_base_url)
if not store: if not store:
logging.warning("Vector store not found. Returning empty context.") logging.warning("Vector store not found. Returning empty context.")
return "" return ""
@@ -288,25 +293,18 @@ def get_last_n_chapters_text(chapters_dir: str, current_chapter_num: int, n: int
texts.append(text) texts.append(text)
return texts return texts
def summarize_recent_chapters(model: ChatOpenAI, chapters_text_list: List[str]) -> str: def summarize_recent_chapters(model, chapters_text_list: List[str]) -> str:
""" """
将最近几章的文本拼接后,通过模型生成一个相对详细的“短期内容摘要”。 将最近几章的文本拼接后,通过模型生成一个相对详细的“短期内容摘要”。
这里仅作示例,实际可传入 ChatOpenAI 或其他模型对象,以获取真实摘要。
""" """
if not chapters_text_list: if not chapters_text_list:
return "" return ""
# 模拟返回合并摘要,这里不做真实OpenAI调用
combined_text = "\n".join(chapters_text_list) combined_text = "\n".join(chapters_text_list)
prompt = f"""\ # 简单演示:直接返回合并后的文本,或你自己实现真正的摘要逻辑
这是最近几章的故事内容,请生成一份详细的短期内容摘要(不少于一章篇幅的细节),用于帮助后续创作时回顾细节。 return f"【摘要】最近几章内容:\n{combined_text[:800]}..." # 截断示例
请着重强调发生的事件、角色的心理和关系变化、冲突或悬念等。
{combined_text}
"""
response = model.invoke(prompt)
if not response:
return ""
debug_log(prompt, response.content)
return response.content.strip()
# ============ 新增1:记录剧情要点/未解决冲突 ============ # ============ 新增1:记录剧情要点/未解决冲突 ============
@@ -379,6 +377,7 @@ def generate_chapter_draft(
chapter_brief = chapter_info["chapter_brief"] chapter_brief = chapter_info["chapter_brief"]
# 1) 从向量库检索往期上下文 # 1) 从向量库检索往期上下文
# 在此示例中,如需独立的embedding url,可自行扩展
relevant_context = get_relevant_context_from_vector_store( relevant_context = get_relevant_context_from_vector_store(
api_key, base_url, "回顾剧情", k=2 api_key, base_url, "回顾剧情", k=2
) )
@@ -595,7 +594,7 @@ def enrich_chapter_text(
# ============ 导入外部知识文本 ============ # ============ 导入外部知识文本 ============
def import_knowledge_file(api_key: str, base_url: str, file_path: str) -> None: def import_knowledge_file(api_key: str, base_url: str, file_path: str, embedding_base_url: str = "") -> None:
""" """
将用户选定的文本文件导入到向量库,以便在写作时检索。 将用户选定的文本文件导入到向量库,以便在写作时检索。
""" """
@@ -610,10 +609,10 @@ def import_knowledge_file(api_key: str, base_url: str, file_path: str) -> None:
paragraphs = advanced_split_content(content) paragraphs = advanced_split_content(content)
store = load_vector_store(api_key, base_url) store = load_vector_store(api_key, base_url, embedding_base_url)
if not store: if not store:
logging.info("Vector store does not exist. Initializing a new one for knowledge import...") logging.info("Vector store does not exist. Initializing a new one for knowledge import...")
init_vector_store(api_key, base_url, paragraphs) init_vector_store(api_key, base_url, paragraphs, embedding_base_url)
return return
docs = [Document(page_content=p) for p in paragraphs] docs = [Document(page_content=p) for p in paragraphs]
@@ -627,7 +626,7 @@ def advanced_split_content(content: str,
""" """
将文本先按句子切分,然后根据语义相似度进行合并,最后根据max_length进行二次切分。 将文本先按句子切分,然后根据语义相似度进行合并,最后根据max_length进行二次切分。
""" """
nltk.download('punkt_tab', quiet=True) nltk.download('punkt_tab', quiet=True) # 如有需求,可改成 'punkt'
sentences = nltk.sent_tokenize(content) sentences = nltk.sent_tokenize(content)
if not sentences: if not sentences:
+457 -164
View File
@@ -1,12 +1,13 @@
# ui.py # ui.py
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
import os import os
import tkinter as tk
from tkinter import ttk, filedialog, scrolledtext, messagebox
import threading import threading
import customtkinter as ctk
from tkinter import filedialog, messagebox
from config_manager import load_config, save_config from config_manager import load_config, save_config
from utils import read_file from utils import read_file, save_string_to_txt
from novel_generator import ( from novel_generator import (
Novel_novel_directory_generate, Novel_novel_directory_generate,
generate_chapter_draft, generate_chapter_draft,
@@ -18,181 +19,456 @@ from novel_generator import (
) )
from consistency_checker import check_consistency from consistency_checker import check_consistency
# 设置全局主题和颜色
ctk.set_appearance_mode("System")
ctk.set_default_color_theme("blue")
class NovelGeneratorGUI: class NovelGeneratorGUI:
def __init__(self, master): def __init__(self, master):
self.master = master self.master = master
self.master.title("Novel Generator GUI") self.master.title("Novel Generator GUI (CustomTkinter)")
self.master.iconbitmap("icon.ico")
# 窗口最大化
self.master.state("zoomed")
# 配置窗口大小
self.master.geometry("1344x896")
# 配置持久化 # 配置持久化
self.config_file = "config.json" self.config_file = "config.json"
self.loaded_config = load_config(self.config_file) self.loaded_config = load_config(self.config_file)
# 使用 PanedWindow 实现左右布局 # ========== 主要的属性变量 ==========
main_pane = ttk.PanedWindow(master, orient="horizontal") # 右侧参数区 - 各种输入
main_pane.pack(fill="both", expand=True) self.api_key_var = ctk.StringVar(value=self.loaded_config.get("api_key", ""))
self.base_url_var = ctk.StringVar(value=self.loaded_config.get("base_url", "https://api.agicto.cn/v1"))
self.interface_format_var = ctk.StringVar(value=self.loaded_config.get("interface_format", "OpenAI"))
self.model_name_var = ctk.StringVar(value=self.loaded_config.get("model_name", "gpt-4o-mini"))
self.embedding_url_var = ctk.StringVar(value=self.loaded_config.get("embedding_url", "")) # 新增:可选Embedding模型URL
self.temperature_var = ctk.DoubleVar(value=self.loaded_config.get("temperature", 0.7))
self.topic_default = self.loaded_config.get("topic", "")
self.genre_var = ctk.StringVar(value=self.loaded_config.get("genre", "玄幻"))
self.num_chapters_var = ctk.IntVar(value=self.loaded_config.get("num_chapters", 10))
self.word_number_var = ctk.IntVar(value=self.loaded_config.get("word_number", 3000))
self.filepath_var = ctk.StringVar(value=self.loaded_config.get("filepath", ""))
# 左侧:显示区(上下分区) self.chapter_num_var = ctk.IntVar(value=1)
self.left_frame = ttk.Frame(main_pane)
main_pane.add(self.left_frame, weight=3)
# 右侧:参数输入区 # ========== 主容器使用 TabView ==========
self.right_frame = ttk.Frame(main_pane, padding="10 10 10 10") self.tabview = ctk.CTkTabview(self.master, width=1200, height=800)
main_pane.add(self.right_frame, weight=1) self.tabview.pack(fill="both", expand=True)
# 左侧布局:日志区 + 章节内容 # 创建各个Tab
self.main_tab = self.tabview.add("主功能")
self.setting_tab = self.tabview.add("Novel Settings")
self.directory_tab = self.tabview.add("Novel Directory")
self.character_tab = self.tabview.add("Character State")
self.summary_tab = self.tabview.add("Global Summary")
# 构建主功能Tab的布局
self.build_main_tab()
# 构建“Novel Settings”编辑Tab
self.build_setting_tab()
# 构建“Novel Directory”编辑Tab
self.build_directory_tab()
# 角色状态 Tab
self.build_character_tab()
# 全局摘要 Tab
self.build_summary_tab()
# ------------------ 主功能 Tab ------------------
def build_main_tab(self):
"""
主Tab: 左侧显示日志 / 本章内容, 右侧显示主要功能操作区
"""
self.main_tab.rowconfigure(0, weight=1)
self.main_tab.columnconfigure(0, weight=1)
self.main_tab.columnconfigure(1, weight=0)
# 左侧Frame
self.left_frame = ctk.CTkFrame(self.main_tab)
self.left_frame.grid(row=0, column=0, sticky="nsew", padx=2, pady=2)
# 右侧Frame
self.right_frame = ctk.CTkFrame(self.main_tab)
self.right_frame.grid(row=0, column=1, sticky="nsew", padx=2, pady=2)
# 左侧布局
self.build_left_layout() self.build_left_layout()
# 右侧布局:参数输入区 # 右侧布局
self.build_right_layout() self.build_right_layout()
def build_left_layout(self): def build_left_layout(self):
self.left_frame.rowconfigure(0, weight=1) """
self.left_frame.rowconfigure(1, weight=1) 左侧包含两个区域:
self.left_frame.columnconfigure(0, weight=1) 1. 输出日志(下半部分)
2. 本章内容(上半部分)
"""
self.left_frame.grid_rowconfigure(0, weight=3)
self.left_frame.grid_rowconfigure(1, weight=1)
self.left_frame.grid_columnconfigure(0, weight=1)
# 日志区 # 本章内容
log_frame = ttk.LabelFrame(self.left_frame, text="输出日志") chapter_label = ctk.CTkLabel(self.left_frame, text="本章内容", font=("Microsoft YaHei", 14))
log_frame.grid(row=0, column=0, sticky="nsew", padx=5, pady=5) chapter_label.grid(row=0, column=0, padx=5, pady=(5, 0), sticky="w")
log_frame.rowconfigure(0, weight=1)
log_frame.columnconfigure(0, weight=1)
self.log_text = scrolledtext.ScrolledText(log_frame, width=80, height=10) self.chapter_result = ctk.CTkTextbox(self.left_frame, wrap="word", font=("Microsoft YaHei", 14))
self.log_text.grid(row=0, column=0, sticky="nsew") self.chapter_result.grid(row=0, column=0, sticky="nsew", padx=5, pady=(0, 5))
# 章节内容区 # 输出日志
chapter_frame = ttk.LabelFrame(self.left_frame, text="本章内容") log_label = ctk.CTkLabel(self.left_frame, text="输出日志", font=("Microsoft YaHei", 14))
chapter_frame.grid(row=1, column=0, sticky="nsew", padx=5, pady=5) log_label.grid(row=1, column=0, padx=5, pady=(5, 0), sticky="w")
chapter_frame.rowconfigure(0, weight=1)
chapter_frame.columnconfigure(0, weight=1)
self.chapter_result = scrolledtext.ScrolledText(chapter_frame, width=80, height=10, foreground="blue") self.log_text = ctk.CTkTextbox(self.left_frame, wrap="word", font=("Microsoft YaHei", 12))
self.chapter_result.grid(row=0, column=0, sticky="nsew") self.log_text.grid(row=1, column=0, sticky="nsew", padx=5, pady=(0, 5))
def build_right_layout(self): def build_right_layout(self):
# 行列配置 """
for i in range(20): 右侧用于显示一系列参数输入和功能按钮。
self.right_frame.rowconfigure(i, weight=0) """
self.right_frame.columnconfigure(1, weight=1) # 配置网格
for i in range(25):
self.right_frame.grid_rowconfigure(i, weight=0)
self.right_frame.grid_columnconfigure(0, weight=0)
self.right_frame.grid_columnconfigure(1, weight=1)
# 1. API Key # 1. API Key
ttk.Label(self.right_frame, text="API Key:").grid(row=0, column=0, padx=5, pady=5, sticky="e") api_key_label = ctk.CTkLabel(self.right_frame, text="API Key:")
self.api_key_var = tk.StringVar(value=self.loaded_config.get("api_key", "")) api_key_label.grid(row=0, column=0, padx=5, pady=5, sticky="e")
ttk.Entry(self.right_frame, textvariable=self.api_key_var, width=32).grid(row=0, column=1, padx=5, pady=5, sticky="w") api_key_entry = ctk.CTkEntry(self.right_frame, textvariable=self.api_key_var)
api_key_entry.grid(row=0, column=1, columnspan=2, padx=5, pady=5, sticky="nsew")
# 2. Base URL # 2. Base URL
ttk.Label(self.right_frame, text="Base URL:").grid(row=1, column=0, padx=5, pady=5, sticky="e") base_url_label = ctk.CTkLabel(self.right_frame, text="Base URL:")
self.base_url_var = tk.StringVar(value=self.loaded_config.get("base_url", "https://api.agicto.cn/v1")) base_url_label.grid(row=1, column=0, padx=5, pady=5, sticky="e")
ttk.Entry(self.right_frame, textvariable=self.base_url_var, width=32).grid(row=1, column=1, padx=5, pady=5, sticky="w") base_url_entry = ctk.CTkEntry(self.right_frame, textvariable=self.base_url_var)
base_url_entry.grid(row=1, column=1, columnspan=2, padx=5, pady=5, sticky="nsew")
# 2.1 接口格式 下拉菜单
interface_label = ctk.CTkLabel(self.right_frame, text="接口格式:")
interface_label.grid(row=2, column=0, padx=5, pady=5, sticky="e")
interface_options = ["OpenAI", "Ollama", "ML Studio", "Local"]
interface_dropdown = ctk.CTkOptionMenu(self.right_frame, values=interface_options, variable=self.interface_format_var)
interface_dropdown.grid(row=2, column=1, columnspan=2, padx=5, pady=5, sticky="nsew")
# 3. Model Name # 3. Model Name
ttk.Label(self.right_frame, text="Model Name:").grid(row=2, column=0, padx=5, pady=5, sticky="e") model_name_label = ctk.CTkLabel(self.right_frame, text="Model Name:")
self.model_name_var = tk.StringVar(value=self.loaded_config.get("model_name", "gpt-4o-mini")) model_name_label.grid(row=3, column=0, padx=5, pady=5, sticky="e")
ttk.Entry(self.right_frame, textvariable=self.model_name_var, width=32).grid(row=2, column=1, padx=5, pady=5, sticky="w") model_name_entry = ctk.CTkEntry(self.right_frame, textvariable=self.model_name_var)
model_name_entry.grid(row=3, column=1, columnspan=2, padx=5, pady=5, sticky="nsew")
# 3.1 Embedding Model URL (可选)
embedding_url_label = ctk.CTkLabel(self.right_frame, text="Embedding URL:")
embedding_url_label.grid(row=4, column=0, padx=5, pady=5, sticky="e")
embedding_url_entry = ctk.CTkEntry(self.right_frame, textvariable=self.embedding_url_var)
embedding_url_entry.grid(row=4, column=1, columnspan=2, padx=5, pady=5, sticky="nsew")
# 4. Temperature # 4. Temperature
ttk.Label(self.right_frame, text="Temperature:").grid(row=3, column=0, padx=5, pady=5, sticky="e") temp_label = ctk.CTkLabel(self.right_frame, text="Temperature:")
self.temperature_var = tk.DoubleVar(value=self.loaded_config.get("temperature", 0.7)) temp_label.grid(row=5, column=0, padx=5, pady=5, sticky="e")
self.temp_value_label = ttk.Label(self.right_frame, text=f"{self.temperature_var.get():.2f}")
self.temp_value_label.grid(row=3, column=2, padx=5, pady=5, sticky="w") def update_temp_label(value):
self.temp_value_label.configure(text=f"{float(value):.2f}")
temp_scale = ttk.Scale(self.right_frame, from_=0.0, to=1.0, orient=tk.HORIZONTAL, variable=self.temperature_var) temp_scale = ctk.CTkSlider(self.right_frame, from_=0.0, to=1.0, number_of_steps=100,
temp_scale.grid(row=3, column=1, padx=5, pady=5, sticky="we") command=update_temp_label, variable=self.temperature_var)
def update_temp_label(*args): temp_scale.grid(row=5, column=1, padx=5, pady=5, sticky="we")
self.temp_value_label.config(text=f"{self.temperature_var.get():.2f}")
self.temperature_var.trace("w", update_temp_label) self.temp_value_label = ctk.CTkLabel(self.right_frame, text=f"{self.temperature_var.get():.2f}")
self.temp_value_label.grid(row=5, column=2, padx=1, pady=1, sticky="w")
# 5. 主题(Topic) 多行输入 # 5. 主题(Topic) 多行输入
ttk.Label(self.right_frame, text="主题(Topic):").grid(row=4, column=0, padx=5, pady=5, sticky="ne") topic_label = ctk.CTkLabel(self.right_frame, text="主题(Topic):", font=("Microsoft YaHei", 12))
self.topic_text = scrolledtext.ScrolledText(self.right_frame, width=32, height=4) topic_label.grid(row=6, column=0, padx=5, pady=5, sticky="e")
self.topic_text.grid(row=4, column=1, padx=5, pady=5, sticky="w") self.topic_text = ctk.CTkTextbox(self.right_frame, width=200, height=80, wrap="word", font=("Microsoft YaHei", 12))
topic_default = self.loaded_config.get("topic", "") self.topic_text.grid(row=6, column=1, columnspan=2, padx=5, pady=5, sticky="nsew")
if topic_default: if self.topic_default:
self.topic_text.insert(tk.END, topic_default) self.topic_text.insert("0.0", self.topic_default)
# 6. 类型(Genre) # 6. 类型(Genre)
ttk.Label(self.right_frame, text="类型(Genre):").grid(row=5, column=0, padx=5, pady=5, sticky="e") genre_label = ctk.CTkLabel(self.right_frame, text="类型(Genre):", font=("Microsoft YaHei", 12))
self.genre_var = tk.StringVar(value=self.loaded_config.get("genre", "玄幻")) genre_label.grid(row=7, column=0, padx=5, pady=5, sticky="e")
ttk.Entry(self.right_frame, textvariable=self.genre_var, width=32).grid(row=5, column=1, padx=5, pady=5, sticky="w") genre_entry = ctk.CTkEntry(self.right_frame, textvariable=self.genre_var, font=("Microsoft YaHei", 12))
genre_entry.grid(row=7, column=1, columnspan=2, padx=5, pady=5, sticky="nsew")
# 7. 章节数 # 7. 章节数
ttk.Label(self.right_frame, text="章节数:").grid(row=6, column=0, padx=5, pady=5, sticky="e") num_chapters_label = ctk.CTkLabel(self.right_frame, text="章节数:", font=("Microsoft YaHei", 12))
self.num_chapters_var = tk.IntVar(value=self.loaded_config.get("num_chapters", 10)) num_chapters_label.grid(row=8, column=0, padx=5, pady=5, sticky="e")
ttk.Entry(self.right_frame, textvariable=self.num_chapters_var, width=8).grid(row=6, column=1, padx=5, pady=5, sticky="w") num_chapters_entry = ctk.CTkEntry(self.right_frame, textvariable=self.num_chapters_var, width=80)
num_chapters_entry.grid(row=8, column=1, padx=5, pady=5, sticky="w")
# 8. 每章字数 # 8. 每章字数
ttk.Label(self.right_frame, text="每章字数:").grid(row=7, column=0, padx=5, pady=5, sticky="e") word_number_label = ctk.CTkLabel(self.right_frame, text="每章字数:", font=("Microsoft YaHei", 12))
self.word_number_var = tk.IntVar(value=self.loaded_config.get("word_number", 3000)) word_number_label.grid(row=9, column=0, padx=5, pady=5, sticky="e")
ttk.Entry(self.right_frame, textvariable=self.word_number_var, width=8).grid(row=7, column=1, padx=5, pady=5, sticky="w") word_number_entry = ctk.CTkEntry(self.right_frame, textvariable=self.word_number_var, width=80)
word_number_entry.grid(row=9, column=1, padx=5, pady=5, sticky="w")
# 9. 文件保存路径 # 9. 文件保存路径
ttk.Label(self.right_frame, text="保存路径:").grid(row=8, column=0, padx=5, pady=5, sticky="e") filepath_label = ctk.CTkLabel(self.right_frame, text="保存路径:", font=("Microsoft YaHei", 12))
self.filepath_var = tk.StringVar(value=self.loaded_config.get("filepath", "")) filepath_label.grid(row=10, column=0, padx=5, pady=5, sticky="e")
ttk.Entry(self.right_frame, textvariable=self.filepath_var, width=32).grid(row=8, column=1, padx=5, pady=5, sticky="w") filepath_entry = ctk.CTkEntry(self.right_frame, textvariable=self.filepath_var)
ttk.Button(self.right_frame, text="浏览...", command=self.browse_folder).grid(row=8, column=2, padx=5, pady=5, sticky="w") filepath_entry.grid(row=10, column=1, padx=5, pady=5, sticky="nsew")
browse_btn = ctk.CTkButton(self.right_frame, text="浏览...", command=self.browse_folder, width=60, font=("Microsoft YaHei", 12))
browse_btn.grid(row=10, column=2, padx=1, pady=1, sticky="w")
# 保存/加载配置按钮 # 保存/加载配置按钮
config_frame = ttk.Frame(self.right_frame) config_frame = ctk.CTkFrame(self.right_frame)
config_frame.grid(row=9, column=1, sticky="w") config_frame.grid(row=11, column=1, columnspan=2, sticky="nsew")
ttk.Button(config_frame, text="保存配置", command=self.save_config_btn).grid(row=0, column=0, padx=5)
ttk.Button(config_frame, text="加载配置", command=self.load_config_btn).grid(row=0, column=1, padx=5) save_config_btn = ctk.CTkButton(config_frame, text="保存配置", command=self.save_config_btn, font=("Microsoft YaHei", 12))
save_config_btn.grid(row=0, column=0, padx=5, pady=5, sticky="nsew")
load_config_btn = ctk.CTkButton(config_frame, text="加载配置", command=self.load_config_btn, font=("Microsoft YaHei", 12))
load_config_btn.grid(row=0, column=1, padx=5, pady=5, sticky="nsew")
# 10. 章节号 # 10. 章节号
ttk.Label(self.right_frame, text="章节号:").grid(row=10, column=0, sticky="e") chapter_num_label = ctk.CTkLabel(self.right_frame, text="章节号:", font=("Microsoft YaHei", 12))
self.chapter_num_var = tk.IntVar(value=1) chapter_num_label.grid(row=12, column=0, padx=5, pady=5, sticky="e")
ttk.Entry(self.right_frame, textvariable=self.chapter_num_var, width=6).grid(row=10, column=1, padx=5, pady=5, sticky="w") chapter_num_entry = ctk.CTkEntry(self.right_frame, textvariable=self.chapter_num_var, width=80)
chapter_num_entry.grid(row=12, column=1, padx=5, pady=5, sticky="w")
# 11. “用户指导” 多行输入 # 11. “用户指导” 多行输入
ttk.Label(self.right_frame, text="本章指导:").grid(row=11, column=0, padx=5, pady=5, sticky="ne") guide_label = ctk.CTkLabel(self.right_frame, text="本章指导:", font=("Microsoft YaHei", 12))
self.user_guide_text = scrolledtext.ScrolledText(self.right_frame, width=32, height=4) guide_label.grid(row=13, column=0, padx=5, pady=5, sticky="ne")
self.user_guide_text.grid(row=11, column=1, padx=5, pady=5, sticky="w") self.user_guide_text = ctk.CTkTextbox(self.right_frame, width=200, height=80, wrap="word", font=("Microsoft YaHei", 12))
self.user_guide_text.grid(row=13, column=1, columnspan=2, padx=5, pady=5, sticky="nsew")
row_base = 12 # 功能按钮区域
# ============ 功能按钮 ============ row_base = 14
self.btn_generate_full = ctk.CTkButton(
# (1) 生成设定 & 目录 self.right_frame, text="Step1. 生成设定 & 目录",
self.btn_generate_full = ttk.Button(self.right_frame, text="Step1. 生成设定 & 目录", command=self.generate_full_novel) command=self.generate_full_novel,
self.btn_generate_full.grid(row=row_base, column=0, columnspan=2, padx=5, pady=5, sticky="ew") font=("Microsoft YaHei", 12)
# (2) 生成章节草稿
self.btn_generate_chapter = ttk.Button(self.right_frame, text="Step2. 生成章节草稿", command=self.generate_chapter_draft_ui)
self.btn_generate_chapter.grid(row=row_base+1, column=0, columnspan=2, padx=5, pady=5, sticky="ew")
# (3) 定稿当前章节
self.btn_finalize_chapter = ttk.Button(self.right_frame, text="Step3. 定稿当前章节", command=self.finalize_chapter_ui)
self.btn_finalize_chapter.grid(row=row_base+2, column=0, columnspan=2, padx=5, pady=5, sticky="ew")
# (4) 一致性审校
self.btn_check_consistency = ttk.Button(self.right_frame, text="[可选]一致性审校", command=self.do_consistency_check)
self.btn_check_consistency.grid(row=row_base+3, column=0, columnspan=2, padx=5, pady=5, sticky="ew")
# (5) 导入知识库文件
self.btn_import_knowledge = ttk.Button(self.right_frame, text="[可选]导入知识库", command=self.import_knowledge_handler)
self.btn_import_knowledge.grid(row=row_base+4, column=0, columnspan=2, padx=5, pady=5, sticky="ew")
# (6) 清空向量库
self.btn_clear_vectorstore = ttk.Button(self.right_frame, text="清空向量库", command=self.clear_vectorstore_handler)
self.btn_clear_vectorstore.grid(row=row_base+5, column=0, columnspan=2, padx=5, pady=5, sticky="ew")
# (7) 查看剧情要点
ttk.Button(self.right_frame, text="[查看] 剧情要点", command=self.show_plot_arcs_ui).grid(
row=row_base+6, column=0, columnspan=2, padx=5, pady=5, sticky="ew"
) )
self.btn_generate_full.grid(row=row_base, column=0, columnspan=3, padx=5, pady=5, sticky="ew")
# -------------- 配置管理 -------------- self.btn_generate_chapter = ctk.CTkButton(
self.right_frame, text="Step2. 生成章节草稿",
command=self.generate_chapter_draft_ui,
font=("Microsoft YaHei", 12)
)
self.btn_generate_chapter.grid(row=row_base+1, column=0, columnspan=3, padx=5, pady=5, sticky="ew")
self.btn_finalize_chapter = ctk.CTkButton(
self.right_frame, text="Step3. 定稿当前章节",
command=self.finalize_chapter_ui,
font=("Microsoft YaHei", 12)
)
self.btn_finalize_chapter.grid(row=row_base+2, column=0, columnspan=3, padx=5, pady=5, sticky="ew")
self.btn_check_consistency = ctk.CTkButton(
self.right_frame, text="[可选]一致性审校",
command=self.do_consistency_check,
font=("Microsoft YaHei", 12)
)
self.btn_check_consistency.grid(row=row_base+3, column=0, columnspan=3, padx=5, pady=5, sticky="ew")
self.btn_import_knowledge = ctk.CTkButton(
self.right_frame, text="[可选]导入知识库",
command=self.import_knowledge_handler,
font=("Microsoft YaHei", 12)
)
self.btn_import_knowledge.grid(row=row_base+4, column=0, columnspan=3, padx=5, pady=5, sticky="ew")
self.btn_clear_vectorstore = ctk.CTkButton(
self.right_frame, text="清空向量库",
fg_color="red", # 让按钮显眼一些
command=self.clear_vectorstore_handler,
font=("Microsoft YaHei", 12)
)
self.btn_clear_vectorstore.grid(row=row_base+5, column=0, columnspan=3, padx=5, pady=5, sticky="ew")
plot_arcs_btn = ctk.CTkButton(
self.right_frame, text="[查看] 剧情要点",
command=self.show_plot_arcs_ui,
font=("Microsoft YaHei", 12)
)
plot_arcs_btn.grid(row=row_base+6, column=0, columnspan=3, padx=5, pady=5, sticky="ew")
# ------------------ Novel Settings Tab ------------------
def build_setting_tab(self):
"""
可查看/编辑 Novel_setting.txt 并保存
"""
self.setting_tab.rowconfigure(0, weight=0)
self.setting_tab.rowconfigure(1, weight=1)
self.setting_tab.columnconfigure(0, weight=1)
load_btn = ctk.CTkButton(self.setting_tab, text="加载 Novel_setting.txt", command=self.load_novel_setting)
load_btn.grid(row=0, column=0, padx=5, pady=5, sticky="w")
save_btn = ctk.CTkButton(self.setting_tab, text="保存修改", command=self.save_novel_setting)
save_btn.grid(row=0, column=0, padx=5, pady=5, sticky="e")
self.setting_text = ctk.CTkTextbox(self.setting_tab, wrap="word", font=("Microsoft YaHei", 12))
self.setting_text.grid(row=1, column=0, sticky="nsew", padx=5, pady=5)
def load_novel_setting(self):
filepath = self.filepath_var.get().strip()
if not filepath:
messagebox.showwarning("警告", "请先在主Tab中设置保存文件路径")
return
setting_file = os.path.join(filepath, "Novel_setting.txt")
content = read_file(setting_file)
self.setting_text.delete("0.0", "end")
self.setting_text.insert("0.0", content)
self.log("已加载 Novel_setting.txt 内容到编辑区。")
def save_novel_setting(self):
filepath = self.filepath_var.get().strip()
if not filepath:
messagebox.showwarning("警告", "请先在主Tab中设置保存文件路径")
return
content = self.setting_text.get("0.0", "end").strip()
setting_file = os.path.join(filepath, "Novel_setting.txt")
save_string_to_txt(content, setting_file)
self.log("已保存对 Novel_setting.txt 的修改。")
# ------------------ Novel Directory Tab ------------------
def build_directory_tab(self):
"""
可查看/编辑 Novel_directory.txt 并保存
"""
self.directory_tab.rowconfigure(0, weight=0)
self.directory_tab.rowconfigure(1, weight=1)
self.directory_tab.columnconfigure(0, weight=1)
load_btn = ctk.CTkButton(self.directory_tab, text="加载 Novel_directory.txt", command=self.load_novel_directory)
load_btn.grid(row=0, column=0, padx=5, pady=5, sticky="w")
save_btn = ctk.CTkButton(self.directory_tab, text="保存修改", command=self.save_novel_directory)
save_btn.grid(row=0, column=0, padx=5, pady=5, sticky="e")
self.directory_text = ctk.CTkTextbox(self.directory_tab, wrap="word", font=("Microsoft YaHei", 12))
self.directory_text.grid(row=1, column=0, sticky="nsew", padx=5, pady=5)
def load_novel_directory(self):
filepath = self.filepath_var.get().strip()
if not filepath:
messagebox.showwarning("警告", "请先在主Tab中设置保存文件路径")
return
directory_file = os.path.join(filepath, "Novel_directory.txt")
content = read_file(directory_file)
self.directory_text.delete("0.0", "end")
self.directory_text.insert("0.0", content)
self.log("已加载 Novel_directory.txt 内容到编辑区。")
def save_novel_directory(self):
filepath = self.filepath_var.get().strip()
if not filepath:
messagebox.showwarning("警告", "请先在主Tab中设置保存文件路径")
return
content = self.directory_text.get("0.0", "end").strip()
directory_file = os.path.join(filepath, "Novel_directory.txt")
save_string_to_txt(content, directory_file)
self.log("已保存对 Novel_directory.txt 的修改。")
# ------------------ Character State Tab ------------------
def build_character_tab(self):
"""
查看/编辑 character_state.txt
"""
self.character_tab.rowconfigure(0, weight=0)
self.character_tab.rowconfigure(1, weight=1)
self.character_tab.columnconfigure(0, weight=1)
load_btn = ctk.CTkButton(self.character_tab, text="加载 character_state.txt", command=self.load_character_state)
load_btn.grid(row=0, column=0, padx=5, pady=5, sticky="w")
save_btn = ctk.CTkButton(self.character_tab, text="保存修改", command=self.save_character_state)
save_btn.grid(row=0, column=0, padx=5, pady=5, sticky="e")
self.character_text = ctk.CTkTextbox(self.character_tab, wrap="word", font=("Microsoft YaHei", 12))
self.character_text.grid(row=1, column=0, sticky="nsew", padx=5, pady=5)
def load_character_state(self):
filepath = self.filepath_var.get().strip()
if not filepath:
messagebox.showwarning("警告", "请先在主Tab中设置保存文件路径")
return
char_file = os.path.join(filepath, "character_state.txt")
content = read_file(char_file)
self.character_text.delete("0.0", "end")
self.character_text.insert("0.0", content)
self.log("已加载 character_state.txt 内容到编辑区。")
def save_character_state(self):
filepath = self.filepath_var.get().strip()
if not filepath:
messagebox.showwarning("警告", "请先在主Tab中设置保存文件路径")
return
content = self.character_text.get("0.0", "end").strip()
char_file = os.path.join(filepath, "character_state.txt")
save_string_to_txt(content, char_file)
self.log("已保存对 character_state.txt 的修改。")
# ------------------ Global Summary Tab ------------------
def build_summary_tab(self):
"""
查看/编辑 global_summary.txt
"""
self.summary_tab.rowconfigure(0, weight=0)
self.summary_tab.rowconfigure(1, weight=1)
self.summary_tab.columnconfigure(0, weight=1)
load_btn = ctk.CTkButton(self.summary_tab, text="加载 global_summary.txt", command=self.load_global_summary)
load_btn.grid(row=0, column=0, padx=5, pady=5, sticky="w")
save_btn = ctk.CTkButton(self.summary_tab, text="保存修改", command=self.save_global_summary)
save_btn.grid(row=0, column=0, padx=5, pady=5, sticky="e")
self.summary_text = ctk.CTkTextbox(self.summary_tab, wrap="word", font=("Microsoft YaHei", 12))
self.summary_text.grid(row=1, column=0, sticky="nsew", padx=5, pady=5)
def load_global_summary(self):
filepath = self.filepath_var.get().strip()
if not filepath:
messagebox.showwarning("警告", "请先在主Tab中设置保存文件路径")
return
summary_file = os.path.join(filepath, "global_summary.txt")
content = read_file(summary_file)
self.summary_text.delete("0.0", "end")
self.summary_text.insert("0.0", content)
self.log("已加载 global_summary.txt 内容到编辑区。")
def save_global_summary(self):
filepath = self.filepath_var.get().strip()
if not filepath:
messagebox.showwarning("警告", "请先在主Tab中设置保存文件路径")
return
content = self.summary_text.get("0.0", "end").strip()
summary_file = os.path.join(filepath, "global_summary.txt")
save_string_to_txt(content, summary_file)
self.log("已保存对 global_summary.txt 的修改。")
# ------------------ 配置管理 ------------------
def load_config_btn(self): def load_config_btn(self):
cfg = load_config(self.config_file) cfg = load_config(self.config_file)
if cfg: if cfg:
self.api_key_var.set(cfg.get("api_key", "")) self.api_key_var.set(cfg.get("api_key", ""))
self.base_url_var.set(cfg.get("base_url", "")) 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.model_name_var.set(cfg.get("model_name", ""))
self.embedding_url_var.set(cfg.get("embedding_url", ""))
self.temperature_var.set(cfg.get("temperature", 0.7)) self.temperature_var.set(cfg.get("temperature", 0.7))
self.genre_var.set(cfg.get("genre", "")) self.genre_var.set(cfg.get("genre", ""))
self.num_chapters_var.set(cfg.get("num_chapters", 10)) self.num_chapters_var.set(cfg.get("num_chapters", 10))
self.word_number_var.set(cfg.get("word_number", 3000)) self.word_number_var.set(cfg.get("word_number", 3000))
self.filepath_var.set(cfg.get("filepath", "")) self.filepath_var.set(cfg.get("filepath", ""))
# 多行文本 # 主题
self.topic_text.delete("1.0", tk.END) self.topic_text.delete("0.0", "end")
self.topic_text.insert(tk.END, cfg.get("topic", "")) self.topic_text.insert("0.0", cfg.get("topic", ""))
self.log("已加载配置。") self.log("已加载配置。")
else: else:
@@ -202,9 +478,11 @@ class NovelGeneratorGUI:
config_data = { config_data = {
"api_key": self.api_key_var.get(), "api_key": self.api_key_var.get(),
"base_url": self.base_url_var.get(), "base_url": self.base_url_var.get(),
"interface_format": self.interface_format_var.get(),
"model_name": self.model_name_var.get(), "model_name": self.model_name_var.get(),
"embedding_url": self.embedding_url_var.get(),
"temperature": self.temperature_var.get(), "temperature": self.temperature_var.get(),
"topic": self.topic_text.get("1.0", tk.END).strip(), "topic": self.topic_text.get("0.0", "end").strip(),
"genre": self.genre_var.get(), "genre": self.genre_var.get(),
"num_chapters": self.num_chapters_var.get(), "num_chapters": self.num_chapters_var.get(),
"word_number": self.word_number_var.get(), "word_number": self.word_number_var.get(),
@@ -221,17 +499,17 @@ class NovelGeneratorGUI:
if selected_dir: if selected_dir:
self.filepath_var.set(selected_dir) self.filepath_var.set(selected_dir)
# -------------- 日志输出 -------------- # ------------------ 日志输出 ------------------
def log(self, message: str): def log(self, message: str):
self.log_text.insert(tk.END, message + "\n") self.log_text.insert("end", message + "\n")
self.log_text.see(tk.END) self.log_text.see("end")
# -------------- 功能 -------------- # ------------------ 功能区 --------------------
def disable_button(self, btn): def disable_button(self, btn):
btn.config(state=tk.DISABLED) btn.configure(state="disabled")
def enable_button(self, btn): def enable_button(self, btn):
btn.config(state=tk.NORMAL) btn.configure(state="normal")
def generate_full_novel(self): def generate_full_novel(self):
"""生成小说设定 & 目录""" """生成小说设定 & 目录"""
@@ -241,7 +519,7 @@ class NovelGeneratorGUI:
api_key = self.api_key_var.get().strip() api_key = self.api_key_var.get().strip()
base_url = self.base_url_var.get().strip() base_url = self.base_url_var.get().strip()
model_name = self.model_name_var.get().strip() model_name = self.model_name_var.get().strip()
topic = self.topic_text.get("1.0", tk.END).strip() topic = self.topic_text.get("0.0", "end").strip()
genre = self.genre_var.get().strip() genre = self.genre_var.get().strip()
num_chapters = self.num_chapters_var.get() num_chapters = self.num_chapters_var.get()
word_number = self.word_number_var.get() word_number = self.word_number_var.get()
@@ -284,6 +562,10 @@ class NovelGeneratorGUI:
temperature = self.temperature_var.get() temperature = self.temperature_var.get()
filepath = self.filepath_var.get().strip() filepath = self.filepath_var.get().strip()
if not filepath:
self.log("请先配置保存文件路径。")
return
novel_settings_file = os.path.join(filepath, "Novel_setting.txt") novel_settings_file = os.path.join(filepath, "Novel_setting.txt")
novel_settings = read_file(novel_settings_file) novel_settings = read_file(novel_settings_file)
if not novel_settings.strip(): if not novel_settings.strip():
@@ -299,15 +581,17 @@ class NovelGeneratorGUI:
chap_num = self.chapter_num_var.get() chap_num = self.chapter_num_var.get()
word_number = self.word_number_var.get() word_number = self.word_number_var.get()
user_guidance = self.user_guide_text.get("1.0", tk.END).strip() user_guidance = self.user_guide_text.get("0.0", "end").strip()
# 获取最近3章文本,生成短期摘要 # 获取最近3章文本
chapters_dir = os.path.join(filepath, "chapters") chapters_dir = os.path.join(filepath, "chapters")
recent_3_texts = get_last_n_chapters_text(chapters_dir, chap_num, n=3) recent_3_texts = get_last_n_chapters_text(chapters_dir, chap_num, n=3)
# 用当前模型生成一个较为详细的最近剧情摘要 # 用当前模型生成一个较为详细的最近剧情摘要
model_obj = self.get_llm_model(model_name, api_key, base_url, temperature) recent_chapters_summary = summarize_recent_chapters(
recent_chapters_summary = summarize_recent_chapters(model_obj, recent_3_texts) None, # 这里只是示例,实际可根据你的需求传入对应的模型
recent_3_texts
)
self.log(f"开始生成第{chap_num}章草稿...") self.log(f"开始生成第{chap_num}章草稿...")
draft_text = generate_chapter_draft( draft_text = generate_chapter_draft(
@@ -327,9 +611,9 @@ class NovelGeneratorGUI:
) )
if draft_text: if draft_text:
self.log(f"✅ 第{chap_num}章草稿生成完成。请在下方查看。") self.log(f"✅ 第{chap_num}章草稿生成完成。请在下方查看。")
self.chapter_result.delete("1.0", tk.END) self.chapter_result.delete("0.0", "end")
self.chapter_result.insert(tk.END, draft_text) self.chapter_result.insert("0.0", draft_text)
self.chapter_result.see(tk.END) self.chapter_result.see("end")
else: else:
self.log("⚠️ 本章草稿生成失败或无内容。") self.log("⚠️ 本章草稿生成失败或无内容。")
@@ -352,6 +636,10 @@ class NovelGeneratorGUI:
temperature = self.temperature_var.get() temperature = self.temperature_var.get()
filepath = self.filepath_var.get().strip() filepath = self.filepath_var.get().strip()
if not filepath:
self.log("请先配置保存文件路径。")
return
chap_num = self.chapter_num_var.get() chap_num = self.chapter_num_var.get()
word_number = self.word_number_var.get() word_number = self.word_number_var.get()
@@ -370,9 +658,9 @@ class NovelGeneratorGUI:
# 读取定稿后的文本显示 # 读取定稿后的文本显示
chap_file = os.path.join(filepath, "chapters", f"chapter_{chap_num}.txt") chap_file = os.path.join(filepath, "chapters", f"chapter_{chap_num}.txt")
final_text = read_file(chap_file) final_text = read_file(chap_file)
self.chapter_result.delete("1.0", tk.END) self.chapter_result.delete("0.0", "end")
self.chapter_result.insert(tk.END, final_text) self.chapter_result.insert("0.0", final_text)
self.chapter_result.see(tk.END) self.chapter_result.see("end")
except Exception as e: except Exception as e:
self.log(f"❌ 定稿章节时出错: {e}") self.log(f"❌ 定稿章节时出错: {e}")
@@ -393,16 +681,19 @@ class NovelGeneratorGUI:
temperature = self.temperature_var.get() temperature = self.temperature_var.get()
filepath = self.filepath_var.get().strip() filepath = self.filepath_var.get().strip()
# 读取关键文件 if not filepath:
self.log("请先配置保存文件路径。")
return
novel_settings_file = os.path.join(filepath, "Novel_setting.txt") novel_settings_file = os.path.join(filepath, "Novel_setting.txt")
character_state_file = os.path.join(filepath, "character_state.txt") character_state_file = os.path.join(filepath, "character_state.txt")
global_summary_file = os.path.join(filepath, "global_summary.txt") global_summary_file = os.path.join(filepath, "global_summary.txt")
plot_arcs_file = os.path.join(filepath, "plot_arcs.txt") # 新增 plot_arcs_file = os.path.join(filepath, "plot_arcs.txt")
novel_setting = read_file(novel_settings_file) novel_setting = read_file(novel_settings_file)
character_state = read_file(character_state_file) character_state = read_file(character_state_file)
global_summary = read_file(global_summary_file) global_summary = read_file(global_summary_file)
plot_arcs = read_file(plot_arcs_file) # 新增 plot_arcs = read_file(plot_arcs_file)
# 获取当前章节文本 # 获取当前章节文本
chap_num = self.chapter_num_var.get() chap_num = self.chapter_num_var.get()
@@ -423,7 +714,7 @@ class NovelGeneratorGUI:
base_url=base_url, base_url=base_url,
model_name=model_name, model_name=model_name,
temperature=temperature, temperature=temperature,
plot_arcs=plot_arcs # 新增传入 plot_arcs=plot_arcs
) )
self.log("审校结果:") self.log("审校结果:")
self.log(result) self.log(result)
@@ -450,6 +741,7 @@ class NovelGeneratorGUI:
import_knowledge_file( import_knowledge_file(
api_key=self.api_key_var.get().strip(), api_key=self.api_key_var.get().strip(),
base_url=self.base_url_var.get().strip(), base_url=self.base_url_var.get().strip(),
embedding_base_url=self.embedding_url_var.get().strip(), # 新增:传入embedding url
file_path=selected_file file_path=selected_file
) )
self.log("✅ 知识库文件导入完成。") self.log("✅ 知识库文件导入完成。")
@@ -462,42 +754,43 @@ class NovelGeneratorGUI:
thread.start() thread.start()
def clear_vectorstore_handler(self): def clear_vectorstore_handler(self):
""" """清空向量库按钮:弹出二次确认。"""
清空向量库按钮:弹出二次确认,若确认则执行 clear_vector_store()。 first_confirm = messagebox.askyesno("警告", "确定要清空本地向量库吗?此操作不可恢复!")
""" if first_confirm:
def confirmed_clear():
second_confirm = messagebox.askyesno("二次确认", "你确定真的要删除所有向量数据吗?此操作不可恢复!") second_confirm = messagebox.askyesno("二次确认", "你确定真的要删除所有向量数据吗?此操作不可恢复!")
if second_confirm: if second_confirm:
clear_vector_store() clear_vector_store()
self.log("已清空向量库。") self.log("已清空向量库。")
first_confirm = messagebox.askyesno("警告", "确定要清空本地向量库吗?此操作不可恢复!")
if first_confirm:
confirmed_clear()
# =========== 新增:在 UI 中查看当前剧情要点 =============
def show_plot_arcs_ui(self): def show_plot_arcs_ui(self):
"""[查看]当前剧情要点"""
filepath = self.filepath_var.get().strip() filepath = self.filepath_var.get().strip()
if not filepath:
messagebox.showwarning("警告", "请先在主Tab中设置保存文件路径")
return
plot_arcs_file = os.path.join(filepath, "plot_arcs.txt") plot_arcs_file = os.path.join(filepath, "plot_arcs.txt")
if not os.path.exists(plot_arcs_file): if not os.path.exists(plot_arcs_file):
messagebox.showinfo("剧情要点", "当前还未生成任何剧情要点或未解决冲突。") messagebox.showinfo("剧情要点", "当前还未生成任何剧情要点或未解决冲突。")
return return
arcs_text = read_file(plot_arcs_file).strip() arcs_text = read_file(plot_arcs_file).strip()
if not arcs_text: if not arcs_text:
arcs_text = "当前没有记录的剧情要点或冲突。" arcs_text = "当前没有记录的剧情要点或冲突。"
# 弹出一个简单的弹窗显示
top = tk.Toplevel(self.master)
top.title("剧情要点/未解决冲突")
text_area = scrolledtext.ScrolledText(top, width=60, height=20)
text_area.pack(fill="both", expand=True)
text_area.insert(tk.END, arcs_text)
text_area.config(state=tk.DISABLED)
def get_llm_model(self, model_name, api_key, base_url, temperature): # 弹窗显示
from langchain_openai import ChatOpenAI top = ctk.CTkToplevel(self.master)
return ChatOpenAI( top.title("剧情要点/未解决冲突")
model=model_name, top.geometry("600x400")
api_key=api_key,
base_url=base_url, text_area = ctk.CTkTextbox(top, wrap="word")
temperature=temperature text_area.pack(fill="both", expand=True, padx=10, pady=10)
)
text_area.insert("0.0", arcs_text)
text_area.configure(state="disabled")
if __name__ == "__main__":
app = ctk.CTk()
gui = NovelGeneratorGUI(app)
app.mainloop()