diff --git a/Src/Novel_directory.txt b/Src/Novel_directory.txt new file mode 100644 index 0000000..bdbd913 --- /dev/null +++ b/Src/Novel_directory.txt @@ -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章 ：星河挽歌 - 故事的结尾，留下无限的遐想，开放式结局 \ No newline at end of file diff --git a/Src/Novel_setting.txt b/Src/Novel_setting.txt new file mode 100644 index 0000000..5767994 --- /dev/null +++ b/Src/Novel_setting.txt @@ -0,0 +1,33 @@ + +小说设定：《星河挽歌：提瓦特星轨》 + +小说类型：玄幻穿越，史诗级冒险，轻松幽默与细腻情感并存，整体偏正剧，但避免过于沉重。 + +字数：约400万字（100章 x 4000字/章） + +视角：第三人称，穿插主角第一人称内心独白。 + +世界观：基于原神世界线平行宇宙的提瓦特大陆，地理环境与原神基本一致，但部分区域势力分布及历史事件有所调整，以适应主角的到来和剧情发展。七国结构保留，但政治格局和社会风貌因主角介入而变。科技与魔法并存，魔法更为显著，主角的星穹铁道科技作为“外来力量”融入其中，引发冲突和变革。重要历史事件保留原神设定，并添加主角引发的新的历史事件。 + +核心内容梗概：主角星河在星穹铁道执行任务时意外穿越到提瓦特大陆，发现大陆面临史无前例的巨大危机。凭借星穹铁道科技力量及经验，星河成为对抗危机的关键人物。在拯救提瓦特过程中，他将与原神角色相遇，建立深厚情谊或产生冲突，最终完成使命，并决定自己的归宿。采用英雄之旅叙事结构，主线为拯救提瓦特，副线为主角与原神角色的情感纠葛。 + +主要角色： + +1. 星河：坚毅、责任感强、略带孤傲，内心温柔渴望归宿。拥有强大的星穹铁道科技力量，但更倾向于智慧和策略解决问题。拥有丰富战斗经验和冷静判断力，但也有过去任务的阴影和对未来命运的迷茫，偶尔冲动，也可能因自身能力产生优越感。他的目标是拯救提瓦特，找到归宿，弥补过去遗憾。他的隐藏秘密是：在星穹铁道的一次任务中，他导致一位重要伙伴牺牲，这个秘密影响着他与他人的关系，并在与原神角色建立深厚感情时成为关键转折点。 + +2. 派蒙：活泼、贪吃、略显成熟，更能理解星河内心世界，给予更深层次的支持。积极参与冒险，但鲁莽会带来麻烦，是重要的喜剧元素。对提瓦特知识渊博，但对星穹铁道科技一无所知。她的目标是帮助星河拯救提瓦特，探索自身存在秘密，与星河建立更深厚羁绊。她的隐藏秘密是：她并非简单的导航精灵，拥有远超常人的智慧和感知能力，与提瓦特大陆的过去有着神秘联系。这会在中段通过关键事件逐步揭示。 她的“记忆碎片”——偶尔失神，喃喃自语，对特定场景反应强烈，甚至出现幻觉，这些会逐渐清晰，暗示她与提瓦特古老文明甚至超越提瓦特的力量有关。 + +3. 丽莎：慵懒、智慧，强大的魔法能力，比原神设定中更成熟、更富有责任感。对星穹铁道科技好奇又警惕，将星河视为竞争对手和值得培养的对象。她会引导主角，但在关键时刻设置考验。她的目标是保护蒙德，培养星河，并寻求突破自身瓶颈。她的隐藏秘密是：她拥有不为人知的秘密身份，与某个神秘组织有关，该组织与即将到来的危机密切相关。她的“禁忌魔法”——偶尔使用威力远超平时能力的魔法，使用后疲惫甚至身体异样，她会刻意隐藏，这与她的神秘组织联系紧密。 + +4. 原神旅行者：与星河互动，产生冲突与合作。两人对提瓦特大陆的理解不同，可能会产生矛盾，但在对抗危机时携手合作。 + +5. 钟离：神秘幕后人物，引导星河完成使命，但目的和动机不明，会带来考验和帮助，也可能成为潜在敌人。他的“真实目的”——对星河的帮助看似无私，但始终保持距离，言行举止充满深意，他的行动与提瓦特危机有关，目的可能并非简单帮助主角，而是为了实现某种更宏大的计划。 + +角色关系与冲突：星河与派蒙（依赖与独立的矛盾），星河与丽莎（竞争与合作），星河与原神旅行者（身份认同冲突），星河与钟离（考验与引导）。 + +暗线与伏笔：派蒙的记忆碎片，丽莎的禁忌魔法，星河的星穹铁道追捕令，钟离的真实目的，提瓦特大陆的失落科技。这些暗线相互交织，推动剧情发展，并为角色成长和故事走向增添可能性和悬念。 + +结局：开放式结局，主角战胜危机，但提瓦特仍面临挑战，主角可以选择留在提瓦特或回到星穹铁道；也可能根据剧情走向相对圆满结局。 + + +世界观与角色动机结合：提瓦特大陆面临的危机是推动星河行动的主要动力，而他过去的经历（星穹铁道任务的阴影和伙伴的牺牲）以及对归宿的渴望，则构成了他内心的冲突和驱动力。派蒙的秘密与提瓦特古老文明的关联，丽莎的禁忌魔法与神秘组织的联系，这些都与即将到来的危机息息相关，也与主角的命运紧密相连。 钟离的真实目的则成为贯穿始终的神秘力量，推动剧情发展，并最终影响故事的结局。 星河的星穹铁道科技，则成为改变提瓦特魔法世界格局的关键因素，引发一系列冲突和变革。 \ No newline at end of file diff --git a/embedding_ollama.py b/embedding_ollama.py new file mode 100644 index 0000000..4bbd8e7 --- /dev/null +++ b/embedding_ollama.py @@ -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}") diff --git a/main.spec b/main.spec index 93ae127..cf52c4c 100644 --- a/main.spec +++ b/main.spec @@ -40,7 +40,7 @@ exe = EXE( a.scripts, [], exclude_binaries=True, - name='AI_NovelGenerator_V1.2.3', + name='AI_NovelGenerator_V1.2.4', debug=True, bootloader_ignore_signals=False, strip=False, @@ -60,5 +60,5 @@ coll = COLLECT( strip=False, upx=True, upx_exclude=[], - name='AI_NovelGenerator_V1.2.3' + name='AI_NovelGenerator_V1.2.4' ) diff --git a/novel_generator.py b/novel_generator.py index a5d62bf..66807a4 100644 --- a/novel_generator.py +++ b/novel_generator.py @@ -51,7 +51,6 @@ if not os.path.exists(VECTOR_STORE_DIR): def clear_vector_store(): """ 清空本地向量库（删除 vectorstore 文件夹内的内容）。 - 需要在UI中加一个二次确认弹窗，防止误删。 """ if os.path.exists(VECTOR_STORE_DIR): try: @@ -68,14 +67,16 @@ def clear_vector_store(): else: 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向量库，将传入的文本进行嵌入并保存到本地目录。 如果不存在该目录，会自动创建。 + 如果 embedding_base_url 不为空，则使用它做为embedding的base，否则默认base_url。 """ + embed_url = embedding_base_url if embedding_base_url else base_url embeddings = OpenAIEmbeddings( openai_api_key=api_key, - openai_api_base=base_url + openai_api_base=embed_url ) documents = [Document(page_content=t) for t in texts] vectorstore = Chroma.from_documents( @@ -86,34 +87,38 @@ def init_vector_store(api_key: str, base_url: str, texts: List[str]) -> Chroma: vectorstore.persist() return vectorstore -def load_vector_store(api_key: str, base_url: str) -> Optional[Chroma]: - """读取已存在的向量库。若不存在则返回 None。""" +def load_vector_store(api_key: str, base_url: str, embedding_base_url: str = "") -> Optional[Chroma]: + """ + 读取已存在的向量库。若不存在则返回 None。 + 同样支持可选的 embedding_base_url。 + """ if not os.path.exists(VECTOR_STORE_DIR): return None + embed_url = embedding_base_url if embedding_base_url else base_url embeddings = OpenAIEmbeddings( 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) -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: 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 new_doc = Document(page_content=new_chapter) store.add_documents([new_doc]) 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 条文本，拼接后返回。 若向量库不存在则返回空字符串。 """ - store = load_vector_store(api_key, base_url) + store = load_vector_store(api_key, base_url, embedding_base_url) if not store: logging.warning("Vector store not found. Returning empty context.") return "" @@ -288,25 +293,18 @@ def get_last_n_chapters_text(chapters_dir: str, current_chapter_num: int, n: int texts.append(text) 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: return "" + # 模拟返回合并摘要，这里不做真实OpenAI调用 combined_text = "\n".join(chapters_text_list) - prompt = f"""\ -这是最近几章的故事内容，请生成一份详细的短期内容摘要（不少于一章篇幅的细节），用于帮助后续创作时回顾细节。 -请着重强调发生的事件、角色的心理和关系变化、冲突或悬念等。 - -{combined_text} -""" - response = model.invoke(prompt) - if not response: - return "" - debug_log(prompt, response.content) - return response.content.strip() + # 简单演示：直接返回合并后的文本，或你自己实现真正的摘要逻辑 + return f"【摘要】最近几章内容:\n{combined_text[:800]}..." # 截断示例 # ============ 新增1：记录剧情要点/未解决冲突 ============ @@ -379,6 +377,7 @@ def generate_chapter_draft( chapter_brief = chapter_info["chapter_brief"] # 1) 从向量库检索往期上下文 + # 在此示例中，如需独立的embedding url，可自行扩展 relevant_context = get_relevant_context_from_vector_store( 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) - store = load_vector_store(api_key, base_url) + store = load_vector_store(api_key, base_url, embedding_base_url) if not store: 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 docs = [Document(page_content=p) for p in paragraphs] @@ -627,7 +626,7 @@ def advanced_split_content(content: str, """ 将文本先按句子切分，然后根据语义相似度进行合并，最后根据max_length进行二次切分。 """ - nltk.download('punkt_tab', quiet=True) + nltk.download('punkt_tab', quiet=True) # 如有需求，可改成 'punkt' sentences = nltk.sent_tokenize(content) if not sentences: diff --git a/ui.py b/ui.py index d5d79ca..4bfa638 100644 --- a/ui.py +++ b/ui.py @@ -1,12 +1,13 @@ # ui.py # -*- coding: utf-8 -*- + import os -import tkinter as tk -from tkinter import ttk, filedialog, scrolledtext, messagebox import threading +import customtkinter as ctk +from tkinter import filedialog, messagebox from config_manager import load_config, save_config -from utils import read_file +from utils import read_file, save_string_to_txt from novel_generator import ( Novel_novel_directory_generate, generate_chapter_draft, @@ -18,181 +19,456 @@ from novel_generator import ( ) from consistency_checker import check_consistency +# 设置全局主题和颜色 +ctk.set_appearance_mode("System") +ctk.set_default_color_theme("blue") + class NovelGeneratorGUI: def __init__(self, master): self.master = master - self.master.title("Novel Generator GUI") + self.master.title("Novel Generator GUI (CustomTkinter)") + self.master.iconbitmap("icon.ico") + # 窗口最大化 + self.master.state("zoomed") + # 配置窗口大小 + self.master.geometry("1344x896") # 配置持久化 self.config_file = "config.json" self.loaded_config = load_config(self.config_file) - # 使用 PanedWindow 实现左右布局 - main_pane = ttk.PanedWindow(master, orient="horizontal") - main_pane.pack(fill="both", expand=True) + # ========== 主要的属性变量 ========== + # 右侧参数区 - 各种输入 + self.api_key_var = ctk.StringVar(value=self.loaded_config.get("api_key", "")) + self.base_url_var = ctk.StringVar(value=self.loaded_config.get("base_url", "https://api.agicto.cn/v1")) + self.interface_format_var = ctk.StringVar(value=self.loaded_config.get("interface_format", "OpenAI")) + self.model_name_var = ctk.StringVar(value=self.loaded_config.get("model_name", "gpt-4o-mini")) + self.embedding_url_var = ctk.StringVar(value=self.loaded_config.get("embedding_url", "")) # 新增：可选Embedding模型URL + + self.temperature_var = ctk.DoubleVar(value=self.loaded_config.get("temperature", 0.7)) + self.topic_default = self.loaded_config.get("topic", "") + self.genre_var = ctk.StringVar(value=self.loaded_config.get("genre", "玄幻")) + self.num_chapters_var = ctk.IntVar(value=self.loaded_config.get("num_chapters", 10)) + self.word_number_var = ctk.IntVar(value=self.loaded_config.get("word_number", 3000)) + self.filepath_var = ctk.StringVar(value=self.loaded_config.get("filepath", "")) - # 左侧：显示区(上下分区) - self.left_frame = ttk.Frame(main_pane) - main_pane.add(self.left_frame, weight=3) + self.chapter_num_var = ctk.IntVar(value=1) - # 右侧：参数输入区 - self.right_frame = ttk.Frame(main_pane, padding="10 10 10 10") - main_pane.add(self.right_frame, weight=1) + # ========== 主容器使用 TabView ========== + self.tabview = ctk.CTkTabview(self.master, width=1200, height=800) + self.tabview.pack(fill="both", expand=True) - # 左侧布局：日志区 + 章节内容 + # 创建各个Tab + self.main_tab = self.tabview.add("主功能") + self.setting_tab = self.tabview.add("Novel Settings") + self.directory_tab = self.tabview.add("Novel Directory") + self.character_tab = self.tabview.add("Character State") + self.summary_tab = self.tabview.add("Global Summary") + + # 构建主功能Tab的布局 + self.build_main_tab() + + # 构建“Novel Settings”编辑Tab + self.build_setting_tab() + + # 构建“Novel Directory”编辑Tab + self.build_directory_tab() + + # 角色状态 Tab + self.build_character_tab() + + # 全局摘要 Tab + self.build_summary_tab() + + # ------------------ 主功能 Tab ------------------ + def build_main_tab(self): + """ + 主Tab: 左侧显示日志 / 本章内容, 右侧显示主要功能操作区 + """ + self.main_tab.rowconfigure(0, weight=1) + self.main_tab.columnconfigure(0, weight=1) + self.main_tab.columnconfigure(1, weight=0) + + # 左侧Frame + self.left_frame = ctk.CTkFrame(self.main_tab) + self.left_frame.grid(row=0, column=0, sticky="nsew", padx=2, pady=2) + # 右侧Frame + self.right_frame = ctk.CTkFrame(self.main_tab) + self.right_frame.grid(row=0, column=1, sticky="nsew", padx=2, pady=2) + + # 左侧布局 self.build_left_layout() - # 右侧布局：参数输入区 + # 右侧布局 self.build_right_layout() def build_left_layout(self): - self.left_frame.rowconfigure(0, weight=1) - self.left_frame.rowconfigure(1, weight=1) - self.left_frame.columnconfigure(0, weight=1) + """ + 左侧包含两个区域： + 1. 输出日志（下半部分） + 2. 本章内容（上半部分） + """ + self.left_frame.grid_rowconfigure(0, weight=3) + self.left_frame.grid_rowconfigure(1, weight=1) + self.left_frame.grid_columnconfigure(0, weight=1) - # 日志区 - log_frame = ttk.LabelFrame(self.left_frame, text="输出日志") - log_frame.grid(row=0, column=0, sticky="nsew", padx=5, pady=5) - log_frame.rowconfigure(0, weight=1) - log_frame.columnconfigure(0, weight=1) + # 本章内容 + chapter_label = ctk.CTkLabel(self.left_frame, text="本章内容", font=("Microsoft YaHei", 14)) + chapter_label.grid(row=0, column=0, padx=5, pady=(5, 0), sticky="w") - self.log_text = scrolledtext.ScrolledText(log_frame, width=80, height=10) - self.log_text.grid(row=0, column=0, sticky="nsew") + self.chapter_result = ctk.CTkTextbox(self.left_frame, wrap="word", font=("Microsoft YaHei", 14)) + self.chapter_result.grid(row=0, column=0, sticky="nsew", padx=5, pady=(0, 5)) - # 章节内容区 - chapter_frame = ttk.LabelFrame(self.left_frame, text="本章内容") - chapter_frame.grid(row=1, column=0, sticky="nsew", padx=5, pady=5) - chapter_frame.rowconfigure(0, weight=1) - chapter_frame.columnconfigure(0, weight=1) + # 输出日志 + log_label = ctk.CTkLabel(self.left_frame, text="输出日志", font=("Microsoft YaHei", 14)) + log_label.grid(row=1, column=0, padx=5, pady=(5, 0), sticky="w") - self.chapter_result = scrolledtext.ScrolledText(chapter_frame, width=80, height=10, foreground="blue") - self.chapter_result.grid(row=0, column=0, sticky="nsew") + self.log_text = ctk.CTkTextbox(self.left_frame, wrap="word", font=("Microsoft YaHei", 12)) + self.log_text.grid(row=1, column=0, sticky="nsew", padx=5, pady=(0, 5)) def build_right_layout(self): - # 行列配置 - for i in range(20): - self.right_frame.rowconfigure(i, weight=0) - self.right_frame.columnconfigure(1, weight=1) + """ + 右侧用于显示一系列参数输入和功能按钮。 + """ + # 配置网格 + for i in range(25): + self.right_frame.grid_rowconfigure(i, weight=0) + self.right_frame.grid_columnconfigure(0, weight=0) + self.right_frame.grid_columnconfigure(1, weight=1) # 1. API Key - ttk.Label(self.right_frame, text="API Key:").grid(row=0, column=0, padx=5, pady=5, sticky="e") - self.api_key_var = tk.StringVar(value=self.loaded_config.get("api_key", "")) - ttk.Entry(self.right_frame, textvariable=self.api_key_var, width=32).grid(row=0, column=1, padx=5, pady=5, sticky="w") + api_key_label = ctk.CTkLabel(self.right_frame, text="API Key:") + api_key_label.grid(row=0, column=0, padx=5, pady=5, sticky="e") + api_key_entry = ctk.CTkEntry(self.right_frame, textvariable=self.api_key_var) + api_key_entry.grid(row=0, column=1, columnspan=2, padx=5, pady=5, sticky="nsew") # 2. Base URL - ttk.Label(self.right_frame, text="Base URL:").grid(row=1, column=0, padx=5, pady=5, sticky="e") - self.base_url_var = tk.StringVar(value=self.loaded_config.get("base_url", "https://api.agicto.cn/v1")) - ttk.Entry(self.right_frame, textvariable=self.base_url_var, width=32).grid(row=1, column=1, padx=5, pady=5, sticky="w") + base_url_label = ctk.CTkLabel(self.right_frame, text="Base URL:") + base_url_label.grid(row=1, column=0, padx=5, pady=5, sticky="e") + base_url_entry = ctk.CTkEntry(self.right_frame, textvariable=self.base_url_var) + base_url_entry.grid(row=1, column=1, columnspan=2, padx=5, pady=5, sticky="nsew") + + # 2.1 接口格式 下拉菜单 + interface_label = ctk.CTkLabel(self.right_frame, text="接口格式:") + interface_label.grid(row=2, column=0, padx=5, pady=5, sticky="e") + interface_options = ["OpenAI", "Ollama", "ML Studio", "Local"] + interface_dropdown = ctk.CTkOptionMenu(self.right_frame, values=interface_options, variable=self.interface_format_var) + interface_dropdown.grid(row=2, column=1, columnspan=2, padx=5, pady=5, sticky="nsew") # 3. Model Name - ttk.Label(self.right_frame, text="Model Name:").grid(row=2, column=0, padx=5, pady=5, sticky="e") - self.model_name_var = tk.StringVar(value=self.loaded_config.get("model_name", "gpt-4o-mini")) - ttk.Entry(self.right_frame, textvariable=self.model_name_var, width=32).grid(row=2, column=1, padx=5, pady=5, sticky="w") + model_name_label = ctk.CTkLabel(self.right_frame, text="Model Name:") + model_name_label.grid(row=3, column=0, padx=5, pady=5, sticky="e") + model_name_entry = ctk.CTkEntry(self.right_frame, textvariable=self.model_name_var) + model_name_entry.grid(row=3, column=1, columnspan=2, padx=5, pady=5, sticky="nsew") + + # 3.1 Embedding Model URL (可选) + embedding_url_label = ctk.CTkLabel(self.right_frame, text="Embedding URL:") + embedding_url_label.grid(row=4, column=0, padx=5, pady=5, sticky="e") + embedding_url_entry = ctk.CTkEntry(self.right_frame, textvariable=self.embedding_url_var) + embedding_url_entry.grid(row=4, column=1, columnspan=2, padx=5, pady=5, sticky="nsew") # 4. Temperature - ttk.Label(self.right_frame, text="Temperature:").grid(row=3, column=0, padx=5, pady=5, sticky="e") - self.temperature_var = tk.DoubleVar(value=self.loaded_config.get("temperature", 0.7)) - self.temp_value_label = ttk.Label(self.right_frame, text=f"{self.temperature_var.get():.2f}") - self.temp_value_label.grid(row=3, column=2, padx=5, pady=5, sticky="w") - - temp_scale = ttk.Scale(self.right_frame, from_=0.0, to=1.0, orient=tk.HORIZONTAL, variable=self.temperature_var) - temp_scale.grid(row=3, column=1, padx=5, pady=5, sticky="we") - def update_temp_label(*args): - self.temp_value_label.config(text=f"{self.temperature_var.get():.2f}") - self.temperature_var.trace("w", update_temp_label) + temp_label = ctk.CTkLabel(self.right_frame, text="Temperature:") + temp_label.grid(row=5, column=0, padx=5, pady=5, sticky="e") + + def update_temp_label(value): + self.temp_value_label.configure(text=f"{float(value):.2f}") + temp_scale = ctk.CTkSlider(self.right_frame, from_=0.0, to=1.0, number_of_steps=100, + command=update_temp_label, variable=self.temperature_var) + temp_scale.grid(row=5, column=1, padx=5, pady=5, sticky="we") + + self.temp_value_label = ctk.CTkLabel(self.right_frame, text=f"{self.temperature_var.get():.2f}") + self.temp_value_label.grid(row=5, column=2, padx=1, pady=1, sticky="w") # 5. 主题(Topic) 多行输入 - ttk.Label(self.right_frame, text="主题(Topic):").grid(row=4, column=0, padx=5, pady=5, sticky="ne") - self.topic_text = scrolledtext.ScrolledText(self.right_frame, width=32, height=4) - self.topic_text.grid(row=4, column=1, padx=5, pady=5, sticky="w") - topic_default = self.loaded_config.get("topic", "") - if topic_default: - self.topic_text.insert(tk.END, topic_default) + topic_label = ctk.CTkLabel(self.right_frame, text="主题(Topic):", font=("Microsoft YaHei", 12)) + topic_label.grid(row=6, column=0, padx=5, pady=5, sticky="e") + self.topic_text = ctk.CTkTextbox(self.right_frame, width=200, height=80, wrap="word", font=("Microsoft YaHei", 12)) + self.topic_text.grid(row=6, column=1, columnspan=2, padx=5, pady=5, sticky="nsew") + if self.topic_default: + self.topic_text.insert("0.0", self.topic_default) # 6. 类型(Genre) - ttk.Label(self.right_frame, text="类型(Genre):").grid(row=5, column=0, padx=5, pady=5, sticky="e") - self.genre_var = tk.StringVar(value=self.loaded_config.get("genre", "玄幻")) - ttk.Entry(self.right_frame, textvariable=self.genre_var, width=32).grid(row=5, column=1, padx=5, pady=5, sticky="w") + genre_label = ctk.CTkLabel(self.right_frame, text="类型(Genre):", font=("Microsoft YaHei", 12)) + genre_label.grid(row=7, column=0, padx=5, pady=5, sticky="e") + genre_entry = ctk.CTkEntry(self.right_frame, textvariable=self.genre_var, font=("Microsoft YaHei", 12)) + genre_entry.grid(row=7, column=1, columnspan=2, padx=5, pady=5, sticky="nsew") # 7. 章节数 - ttk.Label(self.right_frame, text="章节数:").grid(row=6, column=0, padx=5, pady=5, sticky="e") - self.num_chapters_var = tk.IntVar(value=self.loaded_config.get("num_chapters", 10)) - ttk.Entry(self.right_frame, textvariable=self.num_chapters_var, width=8).grid(row=6, column=1, padx=5, pady=5, sticky="w") + num_chapters_label = ctk.CTkLabel(self.right_frame, text="章节数:", font=("Microsoft YaHei", 12)) + num_chapters_label.grid(row=8, column=0, padx=5, pady=5, sticky="e") + num_chapters_entry = ctk.CTkEntry(self.right_frame, textvariable=self.num_chapters_var, width=80) + num_chapters_entry.grid(row=8, column=1, padx=5, pady=5, sticky="w") # 8. 每章字数 - ttk.Label(self.right_frame, text="每章字数:").grid(row=7, column=0, padx=5, pady=5, sticky="e") - self.word_number_var = tk.IntVar(value=self.loaded_config.get("word_number", 3000)) - ttk.Entry(self.right_frame, textvariable=self.word_number_var, width=8).grid(row=7, column=1, padx=5, pady=5, sticky="w") + word_number_label = ctk.CTkLabel(self.right_frame, text="每章字数:", font=("Microsoft YaHei", 12)) + word_number_label.grid(row=9, column=0, padx=5, pady=5, sticky="e") + word_number_entry = ctk.CTkEntry(self.right_frame, textvariable=self.word_number_var, width=80) + word_number_entry.grid(row=9, column=1, padx=5, pady=5, sticky="w") # 9. 文件保存路径 - ttk.Label(self.right_frame, text="保存路径:").grid(row=8, column=0, padx=5, pady=5, sticky="e") - self.filepath_var = tk.StringVar(value=self.loaded_config.get("filepath", "")) - ttk.Entry(self.right_frame, textvariable=self.filepath_var, width=32).grid(row=8, column=1, padx=5, pady=5, sticky="w") - ttk.Button(self.right_frame, text="浏览...", command=self.browse_folder).grid(row=8, column=2, padx=5, pady=5, sticky="w") + filepath_label = ctk.CTkLabel(self.right_frame, text="保存路径:", font=("Microsoft YaHei", 12)) + filepath_label.grid(row=10, column=0, padx=5, pady=5, sticky="e") + filepath_entry = ctk.CTkEntry(self.right_frame, textvariable=self.filepath_var) + filepath_entry.grid(row=10, column=1, padx=5, pady=5, sticky="nsew") + browse_btn = ctk.CTkButton(self.right_frame, text="浏览...", command=self.browse_folder, width=60, font=("Microsoft YaHei", 12)) + browse_btn.grid(row=10, column=2, padx=1, pady=1, sticky="w") # 保存/加载配置按钮 - config_frame = ttk.Frame(self.right_frame) - config_frame.grid(row=9, column=1, sticky="w") - ttk.Button(config_frame, text="保存配置", command=self.save_config_btn).grid(row=0, column=0, padx=5) - ttk.Button(config_frame, text="加载配置", command=self.load_config_btn).grid(row=0, column=1, padx=5) + config_frame = ctk.CTkFrame(self.right_frame) + config_frame.grid(row=11, column=1, columnspan=2, sticky="nsew") + + save_config_btn = ctk.CTkButton(config_frame, text="保存配置", command=self.save_config_btn, font=("Microsoft YaHei", 12)) + save_config_btn.grid(row=0, column=0, padx=5, pady=5, sticky="nsew") + + load_config_btn = ctk.CTkButton(config_frame, text="加载配置", command=self.load_config_btn, font=("Microsoft YaHei", 12)) + load_config_btn.grid(row=0, column=1, padx=5, pady=5, sticky="nsew") # 10. 章节号 - ttk.Label(self.right_frame, text="章节号:").grid(row=10, column=0, sticky="e") - self.chapter_num_var = tk.IntVar(value=1) - ttk.Entry(self.right_frame, textvariable=self.chapter_num_var, width=6).grid(row=10, column=1, padx=5, pady=5, sticky="w") + chapter_num_label = ctk.CTkLabel(self.right_frame, text="章节号:", font=("Microsoft YaHei", 12)) + chapter_num_label.grid(row=12, column=0, padx=5, pady=5, sticky="e") + chapter_num_entry = ctk.CTkEntry(self.right_frame, textvariable=self.chapter_num_var, width=80) + chapter_num_entry.grid(row=12, column=1, padx=5, pady=5, sticky="w") # 11. “用户指导” 多行输入 - ttk.Label(self.right_frame, text="本章指导:").grid(row=11, column=0, padx=5, pady=5, sticky="ne") - self.user_guide_text = scrolledtext.ScrolledText(self.right_frame, width=32, height=4) - self.user_guide_text.grid(row=11, column=1, padx=5, pady=5, sticky="w") + guide_label = ctk.CTkLabel(self.right_frame, text="本章指导:", font=("Microsoft YaHei", 12)) + guide_label.grid(row=13, column=0, padx=5, pady=5, sticky="ne") + self.user_guide_text = ctk.CTkTextbox(self.right_frame, width=200, height=80, wrap="word", font=("Microsoft YaHei", 12)) + self.user_guide_text.grid(row=13, column=1, columnspan=2, padx=5, pady=5, sticky="nsew") - row_base = 12 - # ============ 功能按钮 ============ - - # (1) 生成设定 & 目录 - self.btn_generate_full = ttk.Button(self.right_frame, text="Step1. 生成设定 & 目录", command=self.generate_full_novel) - self.btn_generate_full.grid(row=row_base, column=0, columnspan=2, padx=5, pady=5, sticky="ew") - - # (2) 生成章节草稿 - self.btn_generate_chapter = ttk.Button(self.right_frame, text="Step2. 生成章节草稿", command=self.generate_chapter_draft_ui) - self.btn_generate_chapter.grid(row=row_base+1, column=0, columnspan=2, padx=5, pady=5, sticky="ew") - - # (3) 定稿当前章节 - self.btn_finalize_chapter = ttk.Button(self.right_frame, text="Step3. 定稿当前章节", command=self.finalize_chapter_ui) - self.btn_finalize_chapter.grid(row=row_base+2, column=0, columnspan=2, padx=5, pady=5, sticky="ew") - - # (4) 一致性审校 - self.btn_check_consistency = ttk.Button(self.right_frame, text="[可选]一致性审校", command=self.do_consistency_check) - self.btn_check_consistency.grid(row=row_base+3, column=0, columnspan=2, padx=5, pady=5, sticky="ew") - - # (5) 导入知识库文件 - self.btn_import_knowledge = ttk.Button(self.right_frame, text="[可选]导入知识库", command=self.import_knowledge_handler) - self.btn_import_knowledge.grid(row=row_base+4, column=0, columnspan=2, padx=5, pady=5, sticky="ew") - - # (6) 清空向量库 - self.btn_clear_vectorstore = ttk.Button(self.right_frame, text="清空向量库", command=self.clear_vectorstore_handler) - self.btn_clear_vectorstore.grid(row=row_base+5, column=0, columnspan=2, padx=5, pady=5, sticky="ew") - - # (7) 查看剧情要点 - ttk.Button(self.right_frame, text="[查看] 剧情要点", command=self.show_plot_arcs_ui).grid( - row=row_base+6, column=0, columnspan=2, padx=5, pady=5, sticky="ew" + # 功能按钮区域 + row_base = 14 + self.btn_generate_full = ctk.CTkButton( + self.right_frame, text="Step1. 生成设定 & 目录", + command=self.generate_full_novel, + font=("Microsoft YaHei", 12) ) + self.btn_generate_full.grid(row=row_base, column=0, columnspan=3, padx=5, pady=5, sticky="ew") - # -------------- 配置管理 -------------- + self.btn_generate_chapter = ctk.CTkButton( + self.right_frame, text="Step2. 生成章节草稿", + command=self.generate_chapter_draft_ui, + font=("Microsoft YaHei", 12) + ) + self.btn_generate_chapter.grid(row=row_base+1, column=0, columnspan=3, padx=5, pady=5, sticky="ew") + + self.btn_finalize_chapter = ctk.CTkButton( + self.right_frame, text="Step3. 定稿当前章节", + command=self.finalize_chapter_ui, + font=("Microsoft YaHei", 12) + ) + self.btn_finalize_chapter.grid(row=row_base+2, column=0, columnspan=3, padx=5, pady=5, sticky="ew") + + self.btn_check_consistency = ctk.CTkButton( + self.right_frame, text="[可选]一致性审校", + command=self.do_consistency_check, + font=("Microsoft YaHei", 12) + ) + self.btn_check_consistency.grid(row=row_base+3, column=0, columnspan=3, padx=5, pady=5, sticky="ew") + + self.btn_import_knowledge = ctk.CTkButton( + self.right_frame, text="[可选]导入知识库", + command=self.import_knowledge_handler, + font=("Microsoft YaHei", 12) + ) + self.btn_import_knowledge.grid(row=row_base+4, column=0, columnspan=3, padx=5, pady=5, sticky="ew") + + self.btn_clear_vectorstore = ctk.CTkButton( + self.right_frame, text="清空向量库", + fg_color="red", # 让按钮显眼一些 + command=self.clear_vectorstore_handler, + font=("Microsoft YaHei", 12) + ) + self.btn_clear_vectorstore.grid(row=row_base+5, column=0, columnspan=3, padx=5, pady=5, sticky="ew") + + plot_arcs_btn = ctk.CTkButton( + self.right_frame, text="[查看] 剧情要点", + command=self.show_plot_arcs_ui, + font=("Microsoft YaHei", 12) + ) + plot_arcs_btn.grid(row=row_base+6, column=0, columnspan=3, padx=5, pady=5, sticky="ew") + + # ------------------ Novel Settings Tab ------------------ + def build_setting_tab(self): + """ + 可查看/编辑 Novel_setting.txt 并保存 + """ + self.setting_tab.rowconfigure(0, weight=0) + self.setting_tab.rowconfigure(1, weight=1) + self.setting_tab.columnconfigure(0, weight=1) + + load_btn = ctk.CTkButton(self.setting_tab, text="加载 Novel_setting.txt", command=self.load_novel_setting) + load_btn.grid(row=0, column=0, padx=5, pady=5, sticky="w") + + save_btn = ctk.CTkButton(self.setting_tab, text="保存修改", command=self.save_novel_setting) + save_btn.grid(row=0, column=0, padx=5, pady=5, sticky="e") + + self.setting_text = ctk.CTkTextbox(self.setting_tab, wrap="word", font=("Microsoft YaHei", 12)) + self.setting_text.grid(row=1, column=0, sticky="nsew", padx=5, pady=5) + + def load_novel_setting(self): + filepath = self.filepath_var.get().strip() + if not filepath: + messagebox.showwarning("警告", "请先在主Tab中设置保存文件路径") + return + + setting_file = os.path.join(filepath, "Novel_setting.txt") + content = read_file(setting_file) + self.setting_text.delete("0.0", "end") + self.setting_text.insert("0.0", content) + self.log("已加载 Novel_setting.txt 内容到编辑区。") + + def save_novel_setting(self): + filepath = self.filepath_var.get().strip() + if not filepath: + messagebox.showwarning("警告", "请先在主Tab中设置保存文件路径") + return + + content = self.setting_text.get("0.0", "end").strip() + setting_file = os.path.join(filepath, "Novel_setting.txt") + save_string_to_txt(content, setting_file) + self.log("已保存对 Novel_setting.txt 的修改。") + + # ------------------ Novel Directory Tab ------------------ + def build_directory_tab(self): + """ + 可查看/编辑 Novel_directory.txt 并保存 + """ + self.directory_tab.rowconfigure(0, weight=0) + self.directory_tab.rowconfigure(1, weight=1) + self.directory_tab.columnconfigure(0, weight=1) + + load_btn = ctk.CTkButton(self.directory_tab, text="加载 Novel_directory.txt", command=self.load_novel_directory) + load_btn.grid(row=0, column=0, padx=5, pady=5, sticky="w") + + save_btn = ctk.CTkButton(self.directory_tab, text="保存修改", command=self.save_novel_directory) + save_btn.grid(row=0, column=0, padx=5, pady=5, sticky="e") + + self.directory_text = ctk.CTkTextbox(self.directory_tab, wrap="word", font=("Microsoft YaHei", 12)) + self.directory_text.grid(row=1, column=0, sticky="nsew", padx=5, pady=5) + + def load_novel_directory(self): + filepath = self.filepath_var.get().strip() + if not filepath: + messagebox.showwarning("警告", "请先在主Tab中设置保存文件路径") + return + + directory_file = os.path.join(filepath, "Novel_directory.txt") + content = read_file(directory_file) + self.directory_text.delete("0.0", "end") + self.directory_text.insert("0.0", content) + self.log("已加载 Novel_directory.txt 内容到编辑区。") + + def save_novel_directory(self): + filepath = self.filepath_var.get().strip() + if not filepath: + messagebox.showwarning("警告", "请先在主Tab中设置保存文件路径") + return + + content = self.directory_text.get("0.0", "end").strip() + directory_file = os.path.join(filepath, "Novel_directory.txt") + save_string_to_txt(content, directory_file) + self.log("已保存对 Novel_directory.txt 的修改。") + + # ------------------ Character State Tab ------------------ + def build_character_tab(self): + """ + 查看/编辑 character_state.txt + """ + self.character_tab.rowconfigure(0, weight=0) + self.character_tab.rowconfigure(1, weight=1) + self.character_tab.columnconfigure(0, weight=1) + + load_btn = ctk.CTkButton(self.character_tab, text="加载 character_state.txt", command=self.load_character_state) + load_btn.grid(row=0, column=0, padx=5, pady=5, sticky="w") + + save_btn = ctk.CTkButton(self.character_tab, text="保存修改", command=self.save_character_state) + save_btn.grid(row=0, column=0, padx=5, pady=5, sticky="e") + + self.character_text = ctk.CTkTextbox(self.character_tab, wrap="word", font=("Microsoft YaHei", 12)) + self.character_text.grid(row=1, column=0, sticky="nsew", padx=5, pady=5) + + def load_character_state(self): + filepath = self.filepath_var.get().strip() + if not filepath: + messagebox.showwarning("警告", "请先在主Tab中设置保存文件路径") + return + + char_file = os.path.join(filepath, "character_state.txt") + content = read_file(char_file) + self.character_text.delete("0.0", "end") + self.character_text.insert("0.0", content) + self.log("已加载 character_state.txt 内容到编辑区。") + + def save_character_state(self): + filepath = self.filepath_var.get().strip() + if not filepath: + messagebox.showwarning("警告", "请先在主Tab中设置保存文件路径") + return + + content = self.character_text.get("0.0", "end").strip() + char_file = os.path.join(filepath, "character_state.txt") + save_string_to_txt(content, char_file) + self.log("已保存对 character_state.txt 的修改。") + + # ------------------ Global Summary Tab ------------------ + def build_summary_tab(self): + """ + 查看/编辑 global_summary.txt + """ + self.summary_tab.rowconfigure(0, weight=0) + self.summary_tab.rowconfigure(1, weight=1) + self.summary_tab.columnconfigure(0, weight=1) + + load_btn = ctk.CTkButton(self.summary_tab, text="加载 global_summary.txt", command=self.load_global_summary) + load_btn.grid(row=0, column=0, padx=5, pady=5, sticky="w") + + save_btn = ctk.CTkButton(self.summary_tab, text="保存修改", command=self.save_global_summary) + save_btn.grid(row=0, column=0, padx=5, pady=5, sticky="e") + + self.summary_text = ctk.CTkTextbox(self.summary_tab, wrap="word", font=("Microsoft YaHei", 12)) + self.summary_text.grid(row=1, column=0, sticky="nsew", padx=5, pady=5) + + def load_global_summary(self): + filepath = self.filepath_var.get().strip() + if not filepath: + messagebox.showwarning("警告", "请先在主Tab中设置保存文件路径") + return + + summary_file = os.path.join(filepath, "global_summary.txt") + content = read_file(summary_file) + self.summary_text.delete("0.0", "end") + self.summary_text.insert("0.0", content) + self.log("已加载 global_summary.txt 内容到编辑区。") + + def save_global_summary(self): + filepath = self.filepath_var.get().strip() + if not filepath: + messagebox.showwarning("警告", "请先在主Tab中设置保存文件路径") + return + + content = self.summary_text.get("0.0", "end").strip() + summary_file = os.path.join(filepath, "global_summary.txt") + save_string_to_txt(content, summary_file) + self.log("已保存对 global_summary.txt 的修改。") + + # ------------------ 配置管理 ------------------ def load_config_btn(self): cfg = load_config(self.config_file) if cfg: self.api_key_var.set(cfg.get("api_key", "")) self.base_url_var.set(cfg.get("base_url", "")) + self.interface_format_var.set(cfg.get("interface_format", "OpenAI")) self.model_name_var.set(cfg.get("model_name", "")) + self.embedding_url_var.set(cfg.get("embedding_url", "")) self.temperature_var.set(cfg.get("temperature", 0.7)) self.genre_var.set(cfg.get("genre", "")) self.num_chapters_var.set(cfg.get("num_chapters", 10)) self.word_number_var.set(cfg.get("word_number", 3000)) self.filepath_var.set(cfg.get("filepath", "")) - # 多行文本 - self.topic_text.delete("1.0", tk.END) - self.topic_text.insert(tk.END, cfg.get("topic", "")) + # 主题 + self.topic_text.delete("0.0", "end") + self.topic_text.insert("0.0", cfg.get("topic", "")) self.log("已加载配置。") else: @@ -202,9 +478,11 @@ class NovelGeneratorGUI: config_data = { "api_key": self.api_key_var.get(), "base_url": self.base_url_var.get(), + "interface_format": self.interface_format_var.get(), "model_name": self.model_name_var.get(), + "embedding_url": self.embedding_url_var.get(), "temperature": self.temperature_var.get(), - "topic": self.topic_text.get("1.0", tk.END).strip(), + "topic": self.topic_text.get("0.0", "end").strip(), "genre": self.genre_var.get(), "num_chapters": self.num_chapters_var.get(), "word_number": self.word_number_var.get(), @@ -221,17 +499,17 @@ class NovelGeneratorGUI: if selected_dir: self.filepath_var.set(selected_dir) - # -------------- 日志输出 -------------- + # ------------------ 日志输出 ------------------ def log(self, message: str): - self.log_text.insert(tk.END, message + "\n") - self.log_text.see(tk.END) + self.log_text.insert("end", message + "\n") + self.log_text.see("end") - # -------------- 功能 -------------- + # ------------------ 功能区 -------------------- def disable_button(self, btn): - btn.config(state=tk.DISABLED) + btn.configure(state="disabled") def enable_button(self, btn): - btn.config(state=tk.NORMAL) + btn.configure(state="normal") def generate_full_novel(self): """生成小说设定 & 目录""" @@ -241,7 +519,7 @@ class NovelGeneratorGUI: api_key = self.api_key_var.get().strip() base_url = self.base_url_var.get().strip() model_name = self.model_name_var.get().strip() - topic = self.topic_text.get("1.0", tk.END).strip() + topic = self.topic_text.get("0.0", "end").strip() genre = self.genre_var.get().strip() num_chapters = self.num_chapters_var.get() word_number = self.word_number_var.get() @@ -284,6 +562,10 @@ class NovelGeneratorGUI: temperature = self.temperature_var.get() filepath = self.filepath_var.get().strip() + if not filepath: + self.log("请先配置保存文件路径。") + return + novel_settings_file = os.path.join(filepath, "Novel_setting.txt") novel_settings = read_file(novel_settings_file) if not novel_settings.strip(): @@ -299,15 +581,17 @@ class NovelGeneratorGUI: chap_num = self.chapter_num_var.get() word_number = self.word_number_var.get() - user_guidance = self.user_guide_text.get("1.0", tk.END).strip() + user_guidance = self.user_guide_text.get("0.0", "end").strip() - # 获取最近3章文本，生成短期摘要 + # 获取最近3章文本 chapters_dir = os.path.join(filepath, "chapters") recent_3_texts = get_last_n_chapters_text(chapters_dir, chap_num, n=3) # 用当前模型生成一个较为详细的最近剧情摘要 - model_obj = self.get_llm_model(model_name, api_key, base_url, temperature) - recent_chapters_summary = summarize_recent_chapters(model_obj, recent_3_texts) + recent_chapters_summary = summarize_recent_chapters( + None, # 这里只是示例，实际可根据你的需求传入对应的模型 + recent_3_texts + ) self.log(f"开始生成第{chap_num}章草稿...") draft_text = generate_chapter_draft( @@ -327,9 +611,9 @@ class NovelGeneratorGUI: ) if draft_text: self.log(f"✅ 第{chap_num}章草稿生成完成。请在下方查看。") - self.chapter_result.delete("1.0", tk.END) - self.chapter_result.insert(tk.END, draft_text) - self.chapter_result.see(tk.END) + self.chapter_result.delete("0.0", "end") + self.chapter_result.insert("0.0", draft_text) + self.chapter_result.see("end") else: self.log("⚠️ 本章草稿生成失败或无内容。") @@ -352,6 +636,10 @@ class NovelGeneratorGUI: temperature = self.temperature_var.get() filepath = self.filepath_var.get().strip() + if not filepath: + self.log("请先配置保存文件路径。") + return + chap_num = self.chapter_num_var.get() word_number = self.word_number_var.get() @@ -370,9 +658,9 @@ class NovelGeneratorGUI: # 读取定稿后的文本显示 chap_file = os.path.join(filepath, "chapters", f"chapter_{chap_num}.txt") final_text = read_file(chap_file) - self.chapter_result.delete("1.0", tk.END) - self.chapter_result.insert(tk.END, final_text) - self.chapter_result.see(tk.END) + self.chapter_result.delete("0.0", "end") + self.chapter_result.insert("0.0", final_text) + self.chapter_result.see("end") except Exception as e: self.log(f"❌ 定稿章节时出错: {e}") @@ -393,16 +681,19 @@ class NovelGeneratorGUI: temperature = self.temperature_var.get() filepath = self.filepath_var.get().strip() - # 读取关键文件 + if not filepath: + self.log("请先配置保存文件路径。") + return + novel_settings_file = os.path.join(filepath, "Novel_setting.txt") character_state_file = os.path.join(filepath, "character_state.txt") global_summary_file = os.path.join(filepath, "global_summary.txt") - plot_arcs_file = os.path.join(filepath, "plot_arcs.txt") # 新增 + plot_arcs_file = os.path.join(filepath, "plot_arcs.txt") novel_setting = read_file(novel_settings_file) character_state = read_file(character_state_file) global_summary = read_file(global_summary_file) - plot_arcs = read_file(plot_arcs_file) # 新增 + plot_arcs = read_file(plot_arcs_file) # 获取当前章节文本 chap_num = self.chapter_num_var.get() @@ -423,7 +714,7 @@ class NovelGeneratorGUI: base_url=base_url, model_name=model_name, temperature=temperature, - plot_arcs=plot_arcs # 新增传入 + plot_arcs=plot_arcs ) self.log("审校结果：") self.log(result) @@ -450,6 +741,7 @@ class NovelGeneratorGUI: import_knowledge_file( api_key=self.api_key_var.get().strip(), base_url=self.base_url_var.get().strip(), + embedding_base_url=self.embedding_url_var.get().strip(), # 新增：传入embedding url file_path=selected_file ) self.log("✅ 知识库文件导入完成。") @@ -462,42 +754,43 @@ class NovelGeneratorGUI: thread.start() def clear_vectorstore_handler(self): - """ - 清空向量库按钮：弹出二次确认，若确认则执行 clear_vector_store()。 - """ - def confirmed_clear(): + """清空向量库按钮：弹出二次确认。""" + first_confirm = messagebox.askyesno("警告", "确定要清空本地向量库吗？此操作不可恢复！") + if first_confirm: second_confirm = messagebox.askyesno("二次确认", "你确定真的要删除所有向量数据吗？此操作不可恢复！") if second_confirm: clear_vector_store() self.log("已清空向量库。") - first_confirm = messagebox.askyesno("警告", "确定要清空本地向量库吗？此操作不可恢复！") - if first_confirm: - confirmed_clear() - - # =========== 新增：在 UI 中查看当前剧情要点 ============= def show_plot_arcs_ui(self): + """[查看]当前剧情要点""" filepath = self.filepath_var.get().strip() + if not filepath: + messagebox.showwarning("警告", "请先在主Tab中设置保存文件路径") + return + plot_arcs_file = os.path.join(filepath, "plot_arcs.txt") if not os.path.exists(plot_arcs_file): messagebox.showinfo("剧情要点", "当前还未生成任何剧情要点或未解决冲突。") return + arcs_text = read_file(plot_arcs_file).strip() if not arcs_text: arcs_text = "当前没有记录的剧情要点或冲突。" - # 弹出一个简单的弹窗显示 - top = tk.Toplevel(self.master) - top.title("剧情要点/未解决冲突") - text_area = scrolledtext.ScrolledText(top, width=60, height=20) - text_area.pack(fill="both", expand=True) - text_area.insert(tk.END, arcs_text) - text_area.config(state=tk.DISABLED) - def get_llm_model(self, model_name, api_key, base_url, temperature): - from langchain_openai import ChatOpenAI - return ChatOpenAI( - model=model_name, - api_key=api_key, - base_url=base_url, - temperature=temperature - ) + # 弹窗显示 + top = ctk.CTkToplevel(self.master) + top.title("剧情要点/未解决冲突") + top.geometry("600x400") + + text_area = ctk.CTkTextbox(top, wrap="word") + text_area.pack(fill="both", expand=True, padx=10, pady=10) + + text_area.insert("0.0", arcs_text) + text_area.configure(state="disabled") + + +if __name__ == "__main__": + app = ctk.CTk() + gui = NovelGeneratorGUI(app) + app.mainloop()