diff --git a/Src/Novel_directory.txt b/Src/Novel_directory.txt deleted file mode 100644 index 3a9cc6f..0000000 --- a/Src/Novel_directory.txt +++ /dev/null @@ -1,182 +0,0 @@ - -第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章 ：星河挽歌 - 故事的结尾，留下无限的遐想，开放式结局 - -嗯，我需要按照用户提供的《小说设定》来生成100章的目录。首先，我得仔细阅读并理解这个设定，确保每一章都紧扣主题和叙事结构。 - - 世界观设定：暗影之歌 - -在提瓦特大陆上，暗影力量正在苏醒，带来了新的挑战和阴谋。主要角色包括星辰、艾莎、凯文，以及暗影力量本身。故事探讨了力量与责任、身份与命运、信任与背叛等主题。 - - 目录模板 -第1章： - <简述> -... -第100章： - <简述> - -接下来，我需要规划每一章的内容，确保涵盖起始点、初识艾莎、凯文的动机、危机初现、真相浮现、高潮对决以及开放式结局。 - - 起始点 -星辰被选中成为暗影力量的继承者，但她对自己的能力感到不安，同时怀疑“暗影力量”是否真是保护大陆的存在。这一部分需要引入角色和设定，铺垫后续的冲突。 - - 初识艾莎 -艾莎以刺客的身份接近星辰，试图利用她的暗影能力。两人合作中发现彼此的共同点和隐藏的冲突，这可以增加角色的互动和紧张感。 - - 凯文的动机 -凯文表面上支持星辰和艾莎，但实际上有自己的秘密。他可能是“暗影力量”的后代，或者另有隐情。这部分需要揭示凯文的多面性和潜在威胁。 - - 危机初现 -随着暗影力量的觉醒，大陆出现诡异事件。暗影力量失控，普通民众受害。这可以是故事转折点，展示力量带来的危险。 - - 真相浮现 -通过对凯文、艾莎和星辰的深入探索，揭示他们的过去与命运联系。他们之间存在更大的联系，关系到暗影力量的未来。 - - 高潮对决 -关键时刻，星辰和艾莎必须共同面临强敌。这场对决是力量的较量，更是信任与背叛的考验，增加戏剧性和紧张感。 - - 结局模糊 -故事留下开放结局，暗示暗影力量的未来影响，星辰和艾莎是否能找到自己的道路，或者被力量吞噬。 - -在规划每一章时，我需要确保情节连贯，角色发展合理，同时埋下伏笔，为后续章节做好铺垫。例如，在初识艾莎的阶段，可以设置她的多面性和潜在的秘密，为后期冲突做准备。 - -此外，每章的标题应简洁明了，能够反映主要情节或角色的发展，并附上简短的提示性简述，帮助读者了解每一章的内容。 - -现在，我需要将这些思考整理成一个100章的目录，确保每一章都紧扣设定中的叙事结构和角色发展，同时埋下足够的暗线伏笔。 - - -第1章 ：星辰的选择 - 星辰被选中为暗影力量的继承者，但内心充满疑惑与不安。 - -第2章 ：暗影之息 - 星辰在练习中意外触发暗影力量，引发混乱，开始探索自己的能力边界。 - -第3章 ：艾莎的来临 - 艾莎以刺客身份接近星辰，试图获取更多关于暗影力量的信息，同时对星辰产生复杂情感。 - -第4章 ：凯文的真实意图 - 凯文表面上支持星辰和艾莎，但逐渐显露出多面性和隐藏的秘密，引起星辰的警觉。 - -第5章：暗影失控 - 随着暗影力量的强化，大陆出现一系列诡异事件，普通民众受害，暗影力量开始失控。 - -第6章：艾莎的内心挣扎 - 艾莎在完成任务后，发现自己逐渐被暗影力量控制，对自己的选择感到恐慌和纠结。 - -第7章：凯文的秘密揭露 - 凯文被迫公开他的多面性和隐藏的秘密，星辰和艾莎联手调查他的真实意图。 - -第8章：命运的交错 - 通过对凯文、艾莎和星辰的深入了解，逐渐揭示他们之间存在更大的联系关系到暗影力量的未来走向。 - -第9章：危机加剧 - 凯文与暗影力量的势力开始对星辰和艾莎构成直接威胁，两人必须面临更为严峻的挑战。 - -第10章：信任的考验 - 在关键时刻，星辰和艾莎之间的信任面临突破，但也隐藏着潜在的背叛阴谋。 - -第11章：暗影力量的真相 - 星辰通过回忆和资料，逐渐揭开暗影力量的起源和其对世界的影响。 - -第12章：凯文的最终动向 - 凯文的双重身份和真实意图被彻底揭露，他的选择将对暗影力量的未来产生重大影响。 - -第13章：星辰的觉醒 - 星辰在与艾莎的合作中逐渐意识到自己肩负的责任，开始寻找解决问题的方法。 - -第14章：艾莎的挣扎 - 艾莎被迫面对自己的过去和现在的选择，她是否能站出来对抗暗影力量？ - -第15章：凯文的覆灭 - 凯文最终的命运如何？他的计划是否能够成功，或者他将成为阻碍？ - -第16章：暗影之歌 - 星辰和艾莎联手面临最后的危机，他们需要找到一个办法来控制暗影力量。 - -第17章：力量与责任的抉择 - 在关键时刻，星辰和艾莎必须做出艰难的决定，选择自己的道路还是被力量吞噬。 - -第18章：命运的终结 - 故事进入高潮，暗影力量的威胁更加严峻，星辰和艾莎必须团结起来对抗它。 - -第19章：最后的对决 - 星辰和艾莎面临最终的对决，他们的能力和信任将决定世界的命运。 - -第20章：暗影之光 - 故事情节以开放式结束，暗示暗影力量的未来影响，星辰和艾莎是否能找到自己的道路？ \ No newline at end of file diff --git a/Src/Novel_setting.txt b/Src/Novel_setting.txt deleted file mode 100644 index 340c5cc..0000000 --- a/Src/Novel_setting.txt +++ /dev/null @@ -1,98 +0,0 @@ - -小说设定：《星河挽歌：提瓦特星轨》 - -小说类型：玄幻穿越，史诗级冒险，轻松幽默与细腻情感并存，整体偏正剧，但避免过于沉重。 - -字数：约400万字（100章 x 4000字/章） - -视角：第三人称，穿插主角第一人称内心独白。 - -世界观：基于原神世界线平行宇宙的提瓦特大陆，地理环境与原神基本一致，但部分区域势力分布及历史事件有所调整，以适应主角的到来和剧情发展。七国结构保留，但政治格局和社会风貌因主角介入而变。科技与魔法并存，魔法更为显著，主角的星穹铁道科技作为“外来力量”融入其中，引发冲突和变革。重要历史事件保留原神设定，并添加主角引发的新的历史事件。 - -核心内容梗概：主角星河在星穹铁道执行任务时意外穿越到提瓦特大陆，发现大陆面临史无前例的巨大危机。凭借星穹铁道科技力量及经验，星河成为对抗危机的关键人物。在拯救提瓦特过程中，他将与原神角色相遇，建立深厚情谊或产生冲突，最终完成使命，并决定自己的归宿。采用英雄之旅叙事结构，主线为拯救提瓦特，副线为主角与原神角色的情感纠葛。 - -主要角色： - -1. 星河：坚毅、责任感强、略带孤傲，内心温柔渴望归宿。拥有强大的星穹铁道科技力量，但更倾向于智慧和策略解决问题。拥有丰富战斗经验和冷静判断力，但也有过去任务的阴影和对未来命运的迷茫，偶尔冲动，也可能因自身能力产生优越感。他的目标是拯救提瓦特，找到归宿，弥补过去遗憾。他的隐藏秘密是：在星穹铁道的一次任务中，他导致一位重要伙伴牺牲，这个秘密影响着他与他人的关系，并在与原神角色建立深厚感情时成为关键转折点。 - -2. 派蒙：活泼、贪吃、略显成熟，更能理解星河内心世界，给予更深层次的支持。积极参与冒险，但鲁莽会带来麻烦，是重要的喜剧元素。对提瓦特知识渊博，但对星穹铁道科技一无所知。她的目标是帮助星河拯救提瓦特，探索自身存在秘密，与星河建立更深厚羁绊。她的隐藏秘密是：她并非简单的导航精灵，拥有远超常人的智慧和感知能力，与提瓦特大陆的过去有着神秘联系。这会在中段通过关键事件逐步揭示。 她的“记忆碎片”——偶尔失神，喃喃自语，对特定场景反应强烈，甚至出现幻觉，这些会逐渐清晰，暗示她与提瓦特古老文明甚至超越提瓦特的力量有关。 - -3. 丽莎：慵懒、智慧，强大的魔法能力，比原神设定中更成熟、更富有责任感。对星穹铁道科技好奇又警惕，将星河视为竞争对手和值得培养的对象。她会引导主角，但在关键时刻设置考验。她的目标是保护蒙德，培养星河，并寻求突破自身瓶颈。她的隐藏秘密是：她拥有不为人知的秘密身份，与某个神秘组织有关，该组织与即将到来的危机密切相关。她的“禁忌魔法”——偶尔使用威力远超平时能力的魔法，使用后疲惫甚至身体异样，她会刻意隐藏，这与她的神秘组织联系紧密。 - -4. 原神旅行者：与星河互动，产生冲突与合作。两人对提瓦特大陆的理解不同，可能会产生矛盾，但在对抗危机时携手合作。 - -5. 钟离：神秘幕后人物，引导星河完成使命，但目的和动机不明，会带来考验和帮助，也可能成为潜在敌人。他的“真实目的”——对星河的帮助看似无私，但始终保持距离，言行举止充满深意，他的行动与提瓦特危机有关，目的可能并非简单帮助主角，而是为了实现某种更宏大的计划。 - -角色关系与冲突：星河与派蒙（依赖与独立的矛盾），星河与丽莎（竞争与合作），星河与原神旅行者（身份认同冲突），星河与钟离（考验与引导）。 - -暗线与伏笔：派蒙的记忆碎片，丽莎的禁忌魔法，星河的星穹铁道追捕令，钟离的真实目的，提瓦特大陆的失落科技。这些暗线相互交织，推动剧情发展，并为角色成长和故事走向增添可能性和悬念。 - -结局：开放式结局，主角战胜危机，但提瓦特仍面临挑战，主角可以选择留在提瓦特或回到星穹铁道；也可能根据剧情走向相对圆满结局。 - - -世界观与角色动机结合：提瓦特大陆面临的危机是推动星河行动的主要动力，而他过去的经历（星穹铁道任务的阴影和伙伴的牺牲）以及对归宿的渴望，则构成了他内心的冲突和驱动力。派蒙的秘密与提瓦特古老文明的关联，丽莎的禁忌魔法与神秘组织的联系，这些都与即将到来的危机息息相关，也与主角的命运紧密相连。 钟离的真实目的则成为贯穿始终的神秘力量，推动剧情发展，并最终影响故事的结局。 星河的星穹铁道科技，则成为改变提瓦特魔法世界格局的关键因素，引发一系列冲突和变革。 - 世界观设定：暗影之歌 - - 引言 -在一个名为提瓦特的大陆上，力量如同黑暗中的呼吸，无声无息地渗透着一切。这片大陆的历史被称为“暗影年代”，那时期充满了权力斗争和悲剧命运。如今，随着“暗影力量”的再次苏醒，新的阴谋与挑战正逐渐显露。 - - 主要角色 -1. 星辰（亦名：斯琴） - - 背景：一位被称为“暗影之心”的天才少女，拥有操纵暗影的特殊能力。她的力量是暗影之力最强者之一，但同时也让她背负着沉重的责任。 - -2. 艾莎（亦名：艾丽莎） - - 背景：一位冷酷无情的刺客，曾为“暗影力量”效力。她的身世与“暗影力量”密不可分，性格多变，隐藏着巨大的秘密。 - -3. 凯文（亦名：克莱文） - - 背景：自封为提瓦特大陆的守护者，实则心中藏着大不明小谋。他表面上支持星辰和艾莎，但背后暗藏自己的秘密计划。 - -4. 暗影力量 - - 背景：一股强大的力量源于远古，能够操控暗影与黑暗。它的真实目的尚未完全被揭示，既有守护大陆的传说，也有毁灭一切的预言。 - - 世界观亮点 - -1. 力量的双面性 - - 暗影力量是世界的核心，但同时也是最大的危险。掌握它的人往往会被力量所控制，经历深刻的心理变化。 - -2. 身份的迷失 - - 星辰和艾莎都背负着来自“暗影力量”的阴影，他们的身份与命运与力量紧密相连，这种联系既是他们的优势，也是他们的诅咒。 - -3. 权力真空 - - 提瓦特大陆在“暗影年代”后期逐渐走向新的权力真空，各种势力争夺资源和权利，形成了复杂多变的政治格局。 - -4. 宿命的重遇 - - 星辰和艾莎被暗影力量联系起来，这种宿命使他们在面对危机时不得不合作，但同时也让彼此之间充满了矛盾与误解。 - -5. 未知的威胁 - - “暗影力量”的真实目标尚未完全明确，可能是为了大陆的繁荣，也可能是为了彻底摧毁一切。这种不确定性为故事增添了紧张感和悬念。 - - 故事情节展开 - -1. 起始点 - - 星辰被选中成为暗影力量的继承者，但她对自己的能力感到不安，同时也怀疑“暗影力量”是否真的是保护大陆的存在。 - -2. 初识艾莎 - - 艾莎以刺客的身份接近星辰，试图利用她的暗影能力。但两人在合作中逐渐发现彼此之间的共同点，以及隐藏的冲突。 - -3. 凯文的动机 - - 凯文表面上是支持星辰和艾莎，但实际上他有自己的秘密。他可能是“暗影力量”的后代，或者背后另有hidden agenda。 - -4. 危机初现 - - 随着暗影力量的觉醒，大陆出现了一系列诡异的事件。暗影力量开始失控，普通大民被笼罩在阴影中，生存成为一件艰难的事。 - -5. 真相浮现 - - 通过对凯文、艾莎和星辰的深入探索，逐渐揭示他们各自的过去与命运联系。最终，他们会发现自己之间存在着更大的联系，而这可能关系到整个暗影力量的未来。 - -6. 高潮对决 - - 在一次关键时刻，星辰和艾莎必须共同面临一个强大的敌人。这场对决不仅是力量的较量，更是信任与背叛的考验。 - -7. 结局的模糊性 - - 故事情节留下了开放的结局，让读者思考暗影力量最终会带来什么样的未来。星辰和艾莎是否能找到自己的道路，或者他们将被暗影力量彻底吞噬。 - - 主题 -- 力量与责任：暗影力量赋予了一些人巨大的能力，但同时也带来了沉重的责任。 -- 身份与命运：每个人的过去决定了他们的未来，他们是否能超越自己的宿命？ -- 信任与背叛：在危难时刻，真实的朋友和潜在的敌人是如何区分的？ - - 总结 -“暗影之歌”是一个充满矛盾与复杂性的故事，它探讨了力量、身份、信任等多个深刻主题。通过星辰和艾莎两位主角的视角，读者可以感受到在黑暗与光明之间挣扎的人性，以及他们如何面对命运给予的挑战。 \ No newline at end of file diff --git a/embedding_ollama.py b/embedding_ollama.py index b755ddd..5764e61 100644 --- a/embedding_ollama.py +++ b/embedding_ollama.py @@ -1,14 +1,9 @@ -# embedding_ollama.py +import logging import requests from typing import List +import traceback class OllamaEmbeddings: - """ - Ollama 本地服务提供的 Embedding 接口， - 本需求里我们最终拼出形如: http://localhost:11434/api/embed - 即 base_url + "/embed" - """ - def __init__(self, model_name: str, base_url: str): self.model_name = model_name self.base_url = base_url # 这里应形如 http://localhost:11434/api (不再含 /v1) @@ -49,6 +44,28 @@ class OllamaEmbeddings: response = requests.post(url, json=data) response.raise_for_status() result = response.json() + print(result) + + # 检查返回结果是否包含 'embedding' 字段 + if "embedding" not in result: + logging.warning(f"No 'embedding' field in response. Returning empty embedding.") + return [] # 返回空列表 return result["embedding"] + except requests.exceptions.RequestException as e: - raise Exception(f"Ollama embeddings request error: {e}") + logging.error(f"Ollama embeddings request error: {e}") + logging.error(f"Request URL: {url}") + logging.error(f"Request Data: {data}") + logging.error("Full error details:\n" + traceback.format_exc()) + return [] + + except ValueError as e: + logging.error(f"Invalid response structure: {e}") + logging.error(f"Response content: {response.text}") + logging.error("Full error details:\n" + traceback.format_exc()) + return [] + + except Exception as e: + logging.error(f"Unexpected error: {e}") + logging.error("Full error details:\n" + traceback.format_exc()) + return [] diff --git a/novel_generator.py b/novel_generator.py index a9d53ca..71343a2 100644 --- a/novel_generator.py +++ b/novel_generator.py @@ -4,17 +4,12 @@ import os import logging import re from typing import Dict, List, Optional -try: - from typing import TypedDict -except ImportError: - from typing_extensions import TypedDict - +from typing import TypedDict from langchain_openai import ChatOpenAI from langgraph.graph import StateGraph, START, END from langchain_openai import OpenAIEmbeddings from langchain_community.vectorstores import Chroma from langchain.docstore.document import Document - import nltk import math from sentence_transformers import SentenceTransformer @@ -34,15 +29,15 @@ from embedding_ollama import OllamaEmbeddings from chapter_directory_parser import get_chapter_info_from_directory # ============ 日志配置 ============ + logging.basicConfig(level=logging.INFO, format="%(asctime)s [%(levelname)s] %(message)s") def debug_log(prompt: str, response_content: str): - """打印Prompt与Response，可根据需要保留或去掉。""" - logging.info(f"\n[Prompt >>>] {prompt}\n") - logging.info(f"[Response >>>] {response_content}\n") + logging.info(f"\n[Prompt >>>] {prompt}\n") + logging.info(f"[Response >>>] {response_content}\n") +# ============ 判断接口格式相关 ============ -# ============ 接口判断函数 ============ def is_using_ollama_api(interface_format: str, base_url: str) -> bool: """ 当 interface_format == "Ollama" 时返回 True @@ -60,6 +55,8 @@ def is_using_ml_studio_api(interface_format: str, base_url: str) -> bool: return False +# ============ 创建 Embeddings 对象 ============ + def create_embeddings_object( api_key: str, base_url: str, @@ -72,27 +69,25 @@ def create_embeddings_object( - 当 interface_format = "Ollama" => OllamaEmbeddings(...) （此时把 embed_url 中的 /v1 替换成 /api，以便最后调用 /api/embed） - 当 interface_format = "OpenAI" or "ML Studio" => OpenAIEmbeddings - - 其它情况可自行扩展 + - 其它情况视需求可扩展 """ if is_using_ollama_api(interface_format, embed_url): - # 去除末尾斜杠 fixed_url = embed_url.rstrip("/") - # 如果包含 /v1 则替换为 /api fixed_url = fixed_url.replace("/v1", "/api") return OllamaEmbeddings( model_name=embedding_model_name, base_url=fixed_url ) elif is_using_ml_studio_api(interface_format, base_url): + # ML Studio / OpenAI 兼容 return OpenAIEmbeddings(openai_api_key=api_key, openai_api_base=base_url) else: # 默认使用 OpenAIEmbeddings return OpenAIEmbeddings(openai_api_key=api_key, openai_api_base=base_url) -# ============ 日志配置 ============ -logging.basicConfig(level=logging.INFO, format="%(asctime)s [%(levelname)s] %(message)s") # ============ 向量库相关 ============ + VECTOR_STORE_DIR = os.path.join(os.getcwd(), "vectorstore") if not os.path.exists(VECTOR_STORE_DIR): os.makedirs(VECTOR_STORE_DIR) @@ -102,8 +97,8 @@ def clear_vector_store(): 清空本地向量库（删除 vectorstore 文件夹内的内容）。 """ if os.path.exists(VECTOR_STORE_DIR): + import shutil try: - import shutil for filename in os.listdir(VECTOR_STORE_DIR): file_path = os.path.join(VECTOR_STORE_DIR, filename) if os.path.isfile(file_path) or os.path.islink(file_path): @@ -126,7 +121,6 @@ def init_vector_store( ) -> Chroma: """ 初始化并返回一个Chroma向量库，将传入的文本进行嵌入并保存到本地目录。 - embedding_base_url 若不为空，则用于 Ollama 模式下；否则默认使用 base_url。 """ embed_url = embedding_base_url if embedding_base_url else base_url embeddings = create_embeddings_object( @@ -156,6 +150,7 @@ def load_vector_store( 读取已存在的向量库。若不存在则返回 None。 """ if not os.path.exists(VECTOR_STORE_DIR): + logging.info("Vector store not found. Initializing a new one...") return None embed_url = embedding_base_url if embedding_base_url else base_url embeddings = create_embeddings_object( @@ -185,8 +180,10 @@ def update_vector_store( embedding_model_name=embedding_model_name, embedding_base_url=embedding_base_url ) + + # 如果向量库不存在，初始化它 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 for new chapter...") init_vector_store( api_key=api_key, base_url=base_url, @@ -200,6 +197,7 @@ def update_vector_store( new_doc = Document(page_content=new_chapter) store.add_documents([new_doc]) store.persist() + logging.info("Vector store updated with the new chapter.") def get_relevant_context_from_vector_store( api_key: str, @@ -212,7 +210,7 @@ def get_relevant_context_from_vector_store( ) -> str: """ 从向量库中检索与 query 最相关的 k 条文本，拼接后返回。 - 若向量库不存在则返回空字符串。 + 若向量库不存在或没有足够的内容，则返回空字符串。 """ store = load_vector_store( api_key=api_key, @@ -221,10 +219,19 @@ def get_relevant_context_from_vector_store( embedding_model_name=embedding_model_name, embedding_base_url=embedding_base_url ) + + # 如果向量库为空，直接返回空字符串 if not store: - logging.warning("Vector store not found. Returning empty context.") + logging.info("No vector store found. Returning empty context.") return "" + + # 向量库存在，但没有足够的内容时也避免索引错误 docs = store.similarity_search(query, k=k) + + if not docs: + logging.info(f"No relevant documents found for query '{query}'. Returning empty context.") + return "" + combined = "\n".join([d.page_content for d in docs]) return combined @@ -256,7 +263,6 @@ def Novel_novel_directory_generate( """ 使用多步流程，生成 Novel_setting.txt 与 Novel_directory.txt 并保存到 filepath。 """ - # 确保文件夹存在 os.makedirs(filepath, exist_ok=True) model = ChatOpenAI( @@ -327,7 +333,6 @@ def Novel_novel_directory_generate( debug_log(prompt, response.content) return {"novel_directory": response.content.strip()} - # 构建状态图 graph = StateGraph(OverallState) graph.add_node("generate_base_setting", generate_base_setting) graph.add_node("generate_character_setting", generate_character_setting) @@ -363,7 +368,6 @@ def Novel_novel_directory_generate( logging.warning("生成失败：缺少 final_novel_setting 或 novel_directory。") return - # 写入文件 filename_set = os.path.join(filepath, "Novel_setting.txt") filename_novel_directory = os.path.join(filepath, "Novel_directory.txt") @@ -375,7 +379,6 @@ def Novel_novel_directory_generate( append_text_to_file(final_novel_setting_cleaned, filename_set) append_text_to_file(final_novel_directory_cleaned, filename_novel_directory) - logging.info("Novel settings and directory generated successfully.") @@ -394,19 +397,25 @@ def get_last_n_chapters_text(chapters_dir: str, current_chapter_num: int, n: int text = read_file(chap_file).strip() if text: texts.append(text) + if len(texts) < n: + texts = [''] * (n - len(texts)) + texts return texts + def summarize_recent_chapters( - llm_model: str, - api_key: str, - base_url: str, - temperature: float, - chapters_text_list: List[str] - ) -> str: + llm_model: str, + api_key: str, + base_url: str, + temperature: float, + chapters_text_list: List[str] +) -> str: """ - 将最近几章的文本拼接后，通过模型生成一个相对详细的“短期内容摘要”。 - 如果没有可用的模型（model=None），则退化为简单截断示例。 + 将最近几章文本拼接，通过模型生成相对简要的“短期内容摘要”。 """ + if not chapters_text_list: + return "" + if chapters_text_list==['', '', '']: + return "暂无摘要。" model = ChatOpenAI( model=llm_model, api_key=api_key, @@ -414,33 +423,19 @@ def summarize_recent_chapters( temperature=temperature ) - if not chapters_text_list: - return "" - combined_text = "\n".join(chapters_text_list) - # 如果未传入model，就做个简单的退化输出 - if not model: - return f"【摘要-演示】\n{combined_text[:800]}..." - - # 构造一个提示词（Prompt），指示模型生成精简摘要 - prompt = f"""你是一名资深的长篇小说写作辅助AI。下面是最近几章的合并文本内容： + prompt = f"""你是一名资深长篇小说写作辅助AI，下面是最近几章的合并文本： {combined_text} -请你为此文本生成一段简洁扼要的摘要，突出主要剧情进展、角色变化、冲突焦点等要点。 -1.请用中文输出，不超过500字。 -2.仅回复摘要内容，不需要其他信息。 -""" - # 调用模型获取摘要 +请用中文输出不超过500字的摘要，只包含主要剧情进展、角色变化、冲突焦点等要点：""" + response = model.invoke(prompt) if not response or not response.content.strip(): - # 若模型无响应或空，返回简单截断 - return f"【摘要-演示】\n{combined_text[:800]}..." - - # 返回模型生成的摘要文本 + return combined_text[:800] + "..." if len(combined_text) > 800 else combined_text return response.content.strip() -# ============ 新增：更新剧情要点/未解决冲突 ============ +# ============ 新增：剧情要点/未解决冲突 ============ PLOT_ARCS_PROMPT = """\ 下面是新生成的章节内容: @@ -449,9 +444,9 @@ PLOT_ARCS_PROMPT = """\ 这里是已记录的剧情要点/未解决冲突(可能为空): {old_plot_arcs} -请基于新的章节内容，提炼出本章引入或延续的悬念、冲突、角色暗线等，将其合并到旧的剧情要点中。 +请基于新的章节内容，提炼本章引入或延续的悬念、冲突、角色暗线等，将其合并到旧的剧情要点中。 若有新的冲突则添加，若有已解决/不再重要的冲突可标注或移除。 -最终输出一份更新后的剧情要点列表，以帮助后续保持故事的整体一致性和悬念延续。 +最终输出更新后的剧情要点列表，以帮助后续保持故事整体的一致性和悬念延续。 """ def update_plot_arcs( @@ -462,10 +457,6 @@ def update_plot_arcs( model_name: str, temperature: float ) -> str: - """ - 利用模型分析最新章节文本，提炼或更新“未解决冲突或剧情要点”。 - 并返回更新后的字符串。 - """ model = ChatOpenAI( model=model_name, api_key=api_key, @@ -480,7 +471,6 @@ def update_plot_arcs( if not response: logging.warning("update_plot_arcs: No response.") return old_plot_arcs - debug_log(prompt, response.content) return response.content.strip() @@ -499,28 +489,44 @@ def generate_chapter_draft( word_number: int, temperature: float, novel_novel_directory: str, - filepath: str + filepath: str, + interface_format: str, + embedding_model_name: str, + embedding_base_url: str ) -> str: """ - 仅生成当前章节的草稿，不更新全局摘要/角色状态/向量库。 - 并将生成的内容写到 "chapter_{novel_number}.txt" 覆盖写入。 - 同时生成 "outline_{novel_number}.txt" 存储大纲内容。 + 生成当前章节的草稿，不更新全局摘要/角色状态/向量库。 """ - # 0) 根据 novel_number 从 novel_novel_directory 中获取本章标题及简述 + # 根据目录信息获取本章标题、简介 chapter_info = get_chapter_info_from_directory(novel_novel_directory, novel_number) chapter_title = chapter_info["chapter_title"] chapter_brief = chapter_info["chapter_brief"] - # 1) 从向量库检索上下文 (此处仅演示 query="回顾剧情") - relevant_context = get_relevant_context_from_vector_store( - api_key=api_key, - base_url=base_url, - query="回顾剧情", - interface_format="OpenAI", - embedding_model_name="", - embedding_base_url="", - k=2 - ) + # 从向量库检索多次上下文（示例：对本章简介、用户指导分别做查询，再合并） + queries = [] + if user_guidance.strip(): + queries.append(user_guidance) + if chapter_brief.strip(): + queries.append(chapter_brief) + # 也可加一句“回顾剧情”之类 + queries.append("回顾剧情") + + relevant_context = "" + for q in queries: + partial_context = get_relevant_context_from_vector_store( + api_key=api_key, + base_url=base_url, + query=q, + interface_format=interface_format, + embedding_model_name=embedding_model_name, + embedding_base_url=embedding_base_url, + k=2 + ) + if partial_context.strip(): + relevant_context += "\n" + partial_context + # 如果检索结果为空，使用默认值（如空字符串） + if not relevant_context: + relevant_context = "暂无相关内容。" model = ChatOpenAI( model=model_name, @@ -529,10 +535,10 @@ def generate_chapter_draft( temperature=temperature ) - # 2) 生成大纲 + # 1) 生成本章大纲 outline_prompt_text = chapter_outline_prompt.format( novel_setting=novel_settings, - character_state=character_state + "\n\n【历史上下文】\n" + relevant_context, + character_state=character_state + "\n\n【检索到的上下文】\n" + relevant_context, global_summary=global_summary, novel_number=novel_number, chapter_title=chapter_title, @@ -550,10 +556,10 @@ def generate_chapter_draft( clear_file_content(outline_file) save_string_to_txt(chapter_outline, outline_file) - # 3) 生成正文草稿 + # 2) 生成正文草稿 writing_prompt_text = chapter_write_prompt.format( novel_setting=novel_settings, - character_state=character_state + "\n\n【历史上下文】\n" + relevant_context, + character_state=character_state + "\n\n【检索到的上下文】\n" + relevant_context, global_summary=global_summary, chapter_outline=chapter_outline, word_number=word_number, @@ -588,13 +594,12 @@ def finalize_chapter( ): """ 对当前章节进行定稿： - 1. 读取 chapter_{novel_number}.txt 的最终内容； - 2. 更新全局摘要、角色状态文件； - 3. 如果字数明显少于 word_number 的 80%，则自动调用 enrich_chapter_text 再次扩写； - 4. 更新向量库； - 5. 新增：更新剧情要点/未解决冲突 -> plot_arcs.txt + 1. 读取草稿文本 + 2. 若字数太短则再次扩写 + 3. 更新全局摘要、角色状态 + 4. 更新剧情要点 + 5. 更新向量库 """ - # 读取当前章节内容 chapters_dir = os.path.join(filepath, "chapters") chapter_file = os.path.join(chapters_dir, f"chapter_{novel_number}.txt") chapter_text = read_file(chapter_file).strip() @@ -610,9 +615,9 @@ def finalize_chapter( old_global_summary = read_file(global_summary_file) old_plot_arcs = read_file(plot_arcs_file) - # 1) 若字数明显不足，做 enrich + # 若篇幅过短，二次扩写 if len(chapter_text) < 0.8 * word_number: - logging.info("Chapter text seems shorter than 80% of desired length. Attempting to enrich content...") + logging.info("Chapter text is shorter than 80% of desired length. Enriching...") chapter_text = enrich_chapter_text( chapter_text=chapter_text, word_number=word_number, @@ -623,9 +628,8 @@ def finalize_chapter( ) clear_file_content(chapter_file) save_string_to_txt(chapter_text, chapter_file) - logging.info("Chapter text has been enriched and updated.") - # 2) 更新全局摘要 + # 更新全局摘要 model = ChatOpenAI( model=model_name, api_key=api_key, @@ -643,7 +647,7 @@ def finalize_chapter( new_global_summary = update_global_summary(chapter_text, old_global_summary) - # 3) 更新角色状态 + # 更新角色状态 def update_character_state(chapter_text: str, old_state: str) -> str: prompt = update_character_state_prompt.format( chapter_text=chapter_text, @@ -654,7 +658,7 @@ def finalize_chapter( new_char_state = update_character_state(chapter_text, old_char_state) - # 4) 更新剧情要点 + # 更新剧情要点 new_plot_arcs = update_plot_arcs( chapter_text=chapter_text, old_plot_arcs=old_plot_arcs, @@ -664,7 +668,7 @@ def finalize_chapter( temperature=temperature ) - # 5) 覆盖写入文件 + # 写回文件 clear_file_content(character_state_file) save_string_to_txt(new_char_state, character_state_file) @@ -674,10 +678,10 @@ def finalize_chapter( clear_file_content(plot_arcs_file) save_string_to_txt(new_plot_arcs, plot_arcs_file) - # 6) 更新向量库 + # 更新向量库 update_vector_store( - api_key=api_key, - base_url=base_url, + api_key=api_key, + base_url=base_url, new_chapter=chapter_text, interface_format=interface_format, embedding_model_name=embedding_model_name @@ -695,7 +699,6 @@ def enrich_chapter_text( ) -> str: """ 当章节篇幅不足时，调用此函数对章节文本进行二次扩写。 - 可以让模型补充场景描写、角色心理等，保证与现有文本风格一致。 """ model = ChatOpenAI( model=model_name, @@ -713,20 +716,21 @@ def enrich_chapter_text( return chapter_text return response.content.strip() + # ============ 导入外部知识文本 ============ def import_knowledge_file( - api_key: str, - base_url: str, - interface_format: str, - embedding_model_name: str, - file_path: str, - embedding_base_url: str = "" - ) -> None: + api_key: str, + base_url: str, + interface_format: str, + embedding_model_name: str, + file_path: str, + embedding_base_url: str = "" +) -> None: """ 将用户选定的文本文件导入到向量库，以便在写作时检索。 """ - logging.info(f"开始导入知识库文件: {file_path}，当前接口格式: {interface_format}，当前模型: {embedding_model_name}") + logging.info(f"开始导入知识库文件: {file_path}, 接口格式: {interface_format}, 模型: {embedding_model_name}") if not os.path.exists(file_path): logging.warning(f"知识库文件不存在: {file_path}") return @@ -760,11 +764,12 @@ def advanced_split_content(content: str, similarity_threshold: float = 0.7, max_length: int = 500) -> List[str]: """ - 将文本先按句子切分，然后根据语义相似度进行合并，最后根据max_length进行二次切分。 + 将文本先按句子切分，然后根据语义相似度进行合并，最后按max_length二次切分。 """ - nltk.download('punkt_tab', quiet=True) - sentences = nltk.sent_tokenize(content) + # 纠正下载punkt包：'punkt' 而非 'punkt_tab' + nltk.download('punkt', quiet=True) + sentences = nltk.sent_tokenize(content) if not sentences: return [] diff --git a/ui.py b/ui.py index 6bfc0e8..dd787c5 100644 --- a/ui.py +++ b/ui.py @@ -1,11 +1,12 @@ # ui.py # -*- coding: utf-8 -*- +import logging import os import threading import customtkinter as ctk from tkinter import filedialog, messagebox - +import traceback from config_manager import load_config, save_config from utils import read_file, save_string_to_txt from novel_generator import ( @@ -19,6 +20,12 @@ from novel_generator import ( ) from consistency_checker import check_consistency +def log_error(message: str): + """ + 用于打印详细的错误信息和堆栈信息。 + """ + logging.error(f"{message}\n{traceback.format_exc()}") + # 设置全局主题和颜色 ctk.set_appearance_mode("System") ctk.set_default_color_theme("blue") @@ -38,7 +45,6 @@ class NovelGeneratorGUI: self.loaded_config = load_config(self.config_file) # ========== 主要的属性变量 ========== - # 注意：配置项较多，此处增加 embedding_model_name_var 以支持 Embedding 模型名称 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")) @@ -46,7 +52,7 @@ class NovelGeneratorGUI: self.embedding_url_var = ctk.StringVar(value=self.loaded_config.get("embedding_url", "")) self.embedding_model_name_var = ctk.StringVar(value=self.loaded_config.get("embedding_model_name", "")) - + 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", "玄幻")) @@ -128,8 +134,6 @@ class NovelGeneratorGUI: # 日志：只读 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)) - - # 设置只读 self.log_text.configure(state="disabled") def build_right_layout(self): @@ -137,8 +141,6 @@ class NovelGeneratorGUI: 右侧用于显示配置与功能按钮。 其中配置被拆分到一个子 TabView：AI接口配置 和 Embedding配置。 下面再放与小说相关的输入(主题/类型/章节数等)与功能按钮。 - - 另：我们只使用 2 列布局，0 列放标签，1 列放输入框或子区域，保证对齐。 """ self.right_frame.grid_rowconfigure(0, weight=0) self.right_frame.grid_rowconfigure(1, weight=0) @@ -172,14 +174,13 @@ class NovelGeneratorGUI: - 模型名称 - Temperature """ - # 配置网格 for i in range(5): self.ai_config_tab.grid_rowconfigure(i, weight=0) self.ai_config_tab.grid_columnconfigure(0, weight=0) self.ai_config_tab.grid_columnconfigure(1, weight=1) - # 回调：当接口格式下拉框发生变更时，如果 Base URL 为空，则根据接口类型自动填默认值 def on_interface_format_changed(new_value): + # 如果用户切换接口格式，可根据需要修改BaseURL为默认值 if new_value == "Ollama": self.base_url_var.set("http://localhost:11434/v1") elif new_value == "ML Studio": @@ -263,7 +264,6 @@ class NovelGeneratorGUI: def build_novel_params_area(self, start_row=2): """ 放置与小说本身相关的一些配置：主题、类型、章节数、文件路径等 - 只使用两列：0列放Label，1列放输入框(可能加子Frame)。 """ # 主题(Topic) topic_label = ctk.CTkLabel(self.right_frame, text="主题(Topic):", font=("Microsoft YaHei", 12)) @@ -291,7 +291,7 @@ class NovelGeneratorGUI: word_number_entry = ctk.CTkEntry(self.right_frame, textvariable=self.word_number_var, width=80, font=("Microsoft YaHei", 12)) word_number_entry.grid(row=start_row+3, column=1, padx=5, pady=5, sticky="w") - # 保存路径 (使用子Frame放置 Entry + Button) + # 保存路径 filepath_label = ctk.CTkLabel(self.right_frame, text="保存路径:", font=("Microsoft YaHei", 12)) filepath_label.grid(row=start_row+4, column=0, padx=5, pady=5, sticky="e") @@ -320,11 +320,8 @@ class NovelGeneratorGUI: """ 主要功能按钮 + 配置保存/加载 按钮 """ - # 调整下布局：0,1列 - # 这里按行摆放多个按钮 self.right_frame.grid_rowconfigure(start_row, weight=0) - # 先放“保存配置”和“加载配置”按钮 self.btn_frame_config = ctk.CTkFrame(self.right_frame) self.btn_frame_config.grid(row=start_row, column=0, columnspan=2, padx=5, pady=5, sticky="ew") self.btn_frame_config.columnconfigure(0, weight=1) @@ -336,7 +333,6 @@ class NovelGeneratorGUI: load_config_btn = ctk.CTkButton(self.btn_frame_config, text="加载配置", command=self.load_config_btn, font=("Microsoft YaHei", 12)) load_config_btn.grid(row=0, column=1, padx=5, pady=5, sticky="ew") - # 下面是主要功能按钮 self.btn_generate_full = ctk.CTkButton( self.right_frame, text="Step1. 生成设定 & 目录", command=self.generate_full_novel, @@ -389,9 +385,6 @@ class NovelGeneratorGUI: # ------------------ 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) @@ -430,9 +423,6 @@ class NovelGeneratorGUI: # ------------------ 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) @@ -471,9 +461,6 @@ class NovelGeneratorGUI: # ------------------ 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) @@ -512,9 +499,6 @@ class NovelGeneratorGUI: # ------------------ 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) @@ -603,11 +587,9 @@ class NovelGeneratorGUI: # ------------------ 日志输出 ------------------ def log(self, message: str): - # 临时切换为可写 self.log_text.configure(state="normal") self.log_text.insert("end", message + "\n") self.log_text.see("end") - # 再切回只读 self.log_text.configure(state="disabled") # ------------------ 功能区 -------------------- @@ -650,6 +632,7 @@ class NovelGeneratorGUI: ) self.log("✅ 小说设定和目录生成完成。查看 Novel_setting.txt 和 Novel_directory.txt。") except Exception as e: + logging.error(f"生成小说设定 & 目录时出错: {e}") self.log(f"❌ 生成小说设定 & 目录时出错: {e}") finally: self.enable_button(self.btn_generate_full) @@ -693,10 +676,13 @@ class NovelGeneratorGUI: chapters_dir = os.path.join(filepath, "chapters") recent_3_texts = get_last_n_chapters_text(chapters_dir, chap_num, n=3) - # 简易生成最近章节摘要（示例） + # 生成最近章节摘要 recent_chapters_summary = summarize_recent_chapters( - model_name, # 此处仅示例 - recent_3_texts + llm_model=model_name, + api_key=api_key, + base_url=base_url, + temperature=temperature, + chapters_text_list=recent_3_texts ) self.log(f"开始生成第{chap_num}章草稿...") @@ -713,7 +699,10 @@ class NovelGeneratorGUI: word_number=word_number, temperature=temperature, novel_novel_directory=novel_directory, - filepath=filepath + filepath=filepath, + interface_format=self.interface_format_var.get().strip(), + embedding_model_name=self.embedding_model_name_var.get().strip(), + embedding_base_url=self.embedding_url_var.get().strip() ) if draft_text: self.log(f"✅ 第{chap_num}章草稿生成完成。请在左侧查看或编辑。") @@ -724,6 +713,7 @@ class NovelGeneratorGUI: self.log("⚠️ 本章草稿生成失败或无内容。") except Exception as e: + logging.error(f"生成章节草稿时出错: {e}") self.log(f"❌ 生成章节草稿时出错: {e}") finally: self.enable_button(self.btn_generate_chapter) @@ -741,6 +731,8 @@ class NovelGeneratorGUI: model_name = self.model_name_var.get().strip() temperature = self.temperature_var.get() filepath = self.filepath_var.get().strip() + interface_format = self.interface_format_var.get().strip() + embedding_model_name = self.embedding_model_name_var.get().strip() if not filepath: self.log("请先配置保存文件路径。") @@ -755,6 +747,8 @@ class NovelGeneratorGUI: word_number=word_number, api_key=api_key, base_url=base_url, + interface_format=interface_format, + embedding_model_name=embedding_model_name, model_name=model_name, temperature=temperature, filepath=filepath @@ -769,6 +763,7 @@ class NovelGeneratorGUI: self.chapter_result.see("end") except Exception as e: + logging.error(f"定稿章节时出错: {e}") self.log(f"❌ 定稿章节时出错: {e}") finally: self.enable_button(self.btn_finalize_chapter) @@ -801,7 +796,6 @@ class NovelGeneratorGUI: global_summary = read_file(global_summary_file) plot_arcs = read_file(plot_arcs_file) - # 获取当前章节文本 chap_num = self.chapter_num_var.get() chap_file = os.path.join(filepath, "chapters", f"chapter_{chap_num}.txt") chapter_text = read_file(chap_file) @@ -826,6 +820,7 @@ class NovelGeneratorGUI: self.log(result) except Exception as e: + logging.error(f"审校时出错: {e}") self.log(f"❌ 审校时出错: {e}") finally: self.enable_button(self.btn_check_consistency) @@ -848,12 +843,13 @@ class NovelGeneratorGUI: api_key=self.api_key_var.get().strip(), base_url=self.base_url_var.get().strip(), interface_format=self.interface_format_var.get().strip(), - embedding_base_url=self.embedding_url_var.get().strip(), - embedding_base_url=self.embedding_url_var.get().strip(), - file_path=selected_file + embedding_model_name=self.embedding_model_name_var.get().strip(), + file_path=selected_file, + embedding_base_url=self.embedding_url_var.get().strip() ) self.log("✅ 知识库文件导入完成。") except Exception as e: + logging.error(f"导入知识库时出错: {e}") self.log(f"❌ 导入知识库时出错: {e}") finally: self.enable_button(self.btn_import_knowledge) @@ -886,7 +882,6 @@ class NovelGeneratorGUI: if not arcs_text: arcs_text = "当前没有记录的剧情要点或冲突。" - # 弹窗显示 top = ctk.CTkToplevel(self.master) top.title("剧情要点/未解决冲突") top.geometry("600x400")