Merge pull request #5 from YILING0013/dev

Dev

Author: YILING0013

.gitignore

@@ -1,8 +1,8 @@
 /Novel_Src
-/vectorstore
-/.conda
+/.venv
 /build
 /dist
 /.vscode
 /__pycache__
+/vectorstore
 config.json

chapter_directory_parser.py

@@ -0,0 +1,50 @@
+# chapter_directory_parser.py
+# -*- coding: utf-8 -*-
+import re
+
+def get_chapter_info_from_directory(novel_directory_content: str, chapter_number: int):
+    """
+    从给定的 novel_directory_content 文本中，解析 “第X章” 行，并提取本章的标题和可能的简述。
+    返回一个 dict: {
+       "chapter_title": <字符串>,
+       "chapter_brief": <字符串> (若没有则为空)
+    }
+    注意：目录文本示例格式:
+        第1章 ：潮起
+        第2章 ：阴影浮现 - 主要角色冲突爆发
+        ...
+    也可能没有简述，只有一个简单标题。
+    """
+
+    # 将文本逐行拆分
+    lines = novel_directory_content.splitlines()
+
+    # 章节匹配：形如 “第5章 ：xxx” or “第5章: xxx” or “第5章 xxx”
+    pattern = re.compile(r'^第\s*(\d+)\s*章\s*[:：]?\s*(.*)$')
+
+    for line in lines:
+        match = pattern.match(line.strip())
+        if match:
+            chap_num = int(match.group(1))
+            if chap_num == chapter_number:
+                # group(2) 可能是标题及简述的混合
+                full_title = match.group(2).strip()
+                # 这里假设用 '-' 进一步区分“标题 - 简述”，也可能用户没写“ - ”
+                if ' - ' in full_title:
+                    # 根据你的目录格式自由处理
+                    parts = full_title.split(' - ', 1)
+                    return {
+                        "chapter_title": parts[0].strip(),
+                        "chapter_brief": parts[1].strip()
+                    }
+                else:
+                    return {
+                        "chapter_title": full_title,
+                        "chapter_brief": ""
+                    }
+
+    # 如果没有匹配到，返回默认
+    return {
+        "chapter_title": f"第{chapter_number}章",
+        "chapter_brief": ""
+    }

main.py

@@ -5,7 +5,7 @@
 
 def main():
     root = tk.Tk()
-    root.title("Novel Generator - Innovative Flow")
+    root.title("Novel Generator")
     app = NovelGeneratorGUI(root)
     root.mainloop()
 

main.spec

@@ -1,11 +1,13 @@
 # -*- mode: python ; coding: utf-8 -*-
-
+from PyInstaller.utils.hooks import collect_submodules
 
 a = Analysis(
     ['main.py'],
     pathex=[],
     binaries=[],
-    datas=[],
+    datas=[
+        ('vectorstore', 'vectorstore')
+    ],
     hiddenimports=['typing_extensions', 
                     'langchain-openai',
                     'langgraph', 
@@ -17,7 +19,15 @@ a = Analysis(
                     'langchain-community',
                     'pydantic',
                     'pydantic.deprecated.decorator',
-                    'chromadb.utils.embedding_functions.onnx_mini_lm_l6_v2'
+                    *collect_submodules('chromadb'),
+                    'chromadb.utils.embedding_functions.onnx_mini_lm_l6_v2',
+                    'chromadb.telemetry.product.posthog',
+                    'chromadb.api.segment',
+                    'chromadb.db.impl',
+                    'chromadb.db.impl.sqlite',
+                    'chromadb.migrations',
+                    'chromadb.migrations.embeddings_queue'
+                    
                 ],
     hookspath=[],
     hooksconfig={},