记录一下,便于待会核对更改
This commit is contained in:
@@ -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章 :星河挽歌 - 故事的结尾,留下无限的遐想,开放式结局
|
||||||
@@ -0,0 +1,33 @@
|
|||||||
|
|
||||||
|
小说设定:《星河挽歌:提瓦特星轨》
|
||||||
|
|
||||||
|
小说类型:玄幻穿越,史诗级冒险,轻松幽默与细腻情感并存,整体偏正剧,但避免过于沉重。
|
||||||
|
|
||||||
|
字数:约400万字(100章 x 4000字/章)
|
||||||
|
|
||||||
|
视角:第三人称,穿插主角第一人称内心独白。
|
||||||
|
|
||||||
|
世界观:基于原神世界线平行宇宙的提瓦特大陆,地理环境与原神基本一致,但部分区域势力分布及历史事件有所调整,以适应主角的到来和剧情发展。七国结构保留,但政治格局和社会风貌因主角介入而变。科技与魔法并存,魔法更为显著,主角的星穹铁道科技作为“外来力量”融入其中,引发冲突和变革。重要历史事件保留原神设定,并添加主角引发的新的历史事件。
|
||||||
|
|
||||||
|
核心内容梗概:主角星河在星穹铁道执行任务时意外穿越到提瓦特大陆,发现大陆面临史无前例的巨大危机。凭借星穹铁道科技力量及经验,星河成为对抗危机的关键人物。在拯救提瓦特过程中,他将与原神角色相遇,建立深厚情谊或产生冲突,最终完成使命,并决定自己的归宿。采用英雄之旅叙事结构,主线为拯救提瓦特,副线为主角与原神角色的情感纠葛。
|
||||||
|
|
||||||
|
主要角色:
|
||||||
|
|
||||||
|
1. 星河:坚毅、责任感强、略带孤傲,内心温柔渴望归宿。拥有强大的星穹铁道科技力量,但更倾向于智慧和策略解决问题。拥有丰富战斗经验和冷静判断力,但也有过去任务的阴影和对未来命运的迷茫,偶尔冲动,也可能因自身能力产生优越感。他的目标是拯救提瓦特,找到归宿,弥补过去遗憾。他的隐藏秘密是:在星穹铁道的一次任务中,他导致一位重要伙伴牺牲,这个秘密影响着他与他人的关系,并在与原神角色建立深厚感情时成为关键转折点。
|
||||||
|
|
||||||
|
2. 派蒙:活泼、贪吃、略显成熟,更能理解星河内心世界,给予更深层次的支持。积极参与冒险,但鲁莽会带来麻烦,是重要的喜剧元素。对提瓦特知识渊博,但对星穹铁道科技一无所知。她的目标是帮助星河拯救提瓦特,探索自身存在秘密,与星河建立更深厚羁绊。她的隐藏秘密是:她并非简单的导航精灵,拥有远超常人的智慧和感知能力,与提瓦特大陆的过去有着神秘联系。这会在中段通过关键事件逐步揭示。 她的“记忆碎片”——偶尔失神,喃喃自语,对特定场景反应强烈,甚至出现幻觉,这些会逐渐清晰,暗示她与提瓦特古老文明甚至超越提瓦特的力量有关。
|
||||||
|
|
||||||
|
3. 丽莎:慵懒、智慧,强大的魔法能力,比原神设定中更成熟、更富有责任感。对星穹铁道科技好奇又警惕,将星河视为竞争对手和值得培养的对象。她会引导主角,但在关键时刻设置考验。她的目标是保护蒙德,培养星河,并寻求突破自身瓶颈。她的隐藏秘密是:她拥有不为人知的秘密身份,与某个神秘组织有关,该组织与即将到来的危机密切相关。她的“禁忌魔法”——偶尔使用威力远超平时能力的魔法,使用后疲惫甚至身体异样,她会刻意隐藏,这与她的神秘组织联系紧密。
|
||||||
|
|
||||||
|
4. 原神旅行者:与星河互动,产生冲突与合作。两人对提瓦特大陆的理解不同,可能会产生矛盾,但在对抗危机时携手合作。
|
||||||
|
|
||||||
|
5. 钟离:神秘幕后人物,引导星河完成使命,但目的和动机不明,会带来考验和帮助,也可能成为潜在敌人。他的“真实目的”——对星河的帮助看似无私,但始终保持距离,言行举止充满深意,他的行动与提瓦特危机有关,目的可能并非简单帮助主角,而是为了实现某种更宏大的计划。
|
||||||
|
|
||||||
|
角色关系与冲突:星河与派蒙(依赖与独立的矛盾),星河与丽莎(竞争与合作),星河与原神旅行者(身份认同冲突),星河与钟离(考验与引导)。
|
||||||
|
|
||||||
|
暗线与伏笔:派蒙的记忆碎片,丽莎的禁忌魔法,星河的星穹铁道追捕令,钟离的真实目的,提瓦特大陆的失落科技。这些暗线相互交织,推动剧情发展,并为角色成长和故事走向增添可能性和悬念。
|
||||||
|
|
||||||
|
结局:开放式结局,主角战胜危机,但提瓦特仍面临挑战,主角可以选择留在提瓦特或回到星穹铁道;也可能根据剧情走向相对圆满结局。
|
||||||
|
|
||||||
|
|
||||||
|
世界观与角色动机结合:提瓦特大陆面临的危机是推动星河行动的主要动力,而他过去的经历(星穹铁道任务的阴影和伙伴的牺牲)以及对归宿的渴望,则构成了他内心的冲突和驱动力。派蒙的秘密与提瓦特古老文明的关联,丽莎的禁忌魔法与神秘组织的联系,这些都与即将到来的危机息息相关,也与主角的命运紧密相连。 钟离的真实目的则成为贯穿始终的神秘力量,推动剧情发展,并最终影响故事的结局。 星河的星穹铁道科技,则成为改变提瓦特魔法世界格局的关键因素,引发一系列冲突和变革。
|
||||||
@@ -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}")
|
||||||
@@ -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
@@ -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:
|
||||||
|
|||||||
@@ -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.left_frame = ttk.Frame(main_pane)
|
self.topic_default = self.loaded_config.get("topic", "")
|
||||||
main_pane.add(self.left_frame, weight=3)
|
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.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_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")
|
|
||||||
|
|
||||||
temp_scale = ttk.Scale(self.right_frame, from_=0.0, to=1.0, orient=tk.HORIZONTAL, variable=self.temperature_var)
|
def update_temp_label(value):
|
||||||
temp_scale.grid(row=3, column=1, padx=5, pady=5, sticky="we")
|
self.temp_value_label.configure(text=f"{float(value):.2f}")
|
||||||
def update_temp_label(*args):
|
temp_scale = ctk.CTkSlider(self.right_frame, from_=0.0, to=1.0, number_of_steps=100,
|
||||||
self.temp_value_label.config(text=f"{self.temperature_var.get():.2f}")
|
command=update_temp_label, variable=self.temperature_var)
|
||||||
self.temperature_var.trace("w", update_temp_label)
|
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) 多行输入
|
# 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()
|
||||||
|
|||||||
Reference in New Issue
Block a user