Upload 2 files

AIGN.py

@@ -165,19 +165,30 @@ class AIGN:
             temperature=0.66,
         )
 
+    def split_chapters(self, novel_content):
+        # 使用正则表达式匹配章节标题
+        chapter_pattern = re.compile(r'(?:##?|)?\s*第([一二三四五六七八九十百千万亿\d]+)章[:：]?\s*(.+)')
+        
+        # 将小说正文按章节标题分割
+        chapters = chapter_pattern.split(novel_content)
+        
+        # 移除第一个空字符串(如果存在)
+        if chapters[0] == '':
+            chapters = chapters[1:]
+        
+        # 将章节标题和内容组合成元组
+        chapter_tuples = []
+        for i in range(0, len(chapters), 3):
+            if i + 2 < len(chapters):
+                chapter_num = chapters[i]
+                chapter_title = chapters[i + 1]
+                chapter_content = chapters[i + 2]
+                chapter_tuples.append((f"第{chapter_num}章 {chapter_title}", chapter_content))
+        
+        return chapter_tuples
+
     def update_chapter_list(self):
-        self.chapter_list = []
-        chapter_content = ""
-        for line in self.novel_content.split('\n'):
-            if line.startswith('## 第'):
-                if chapter_content:
-                    self.chapter_list.append((chapter_title, chapter_content.strip()))
-                chapter_title = line.strip()
-                chapter_content = ""
-            else:
-                chapter_content += line + "\n"
-        if chapter_content:
-            self.chapter_list.append((chapter_title, chapter_content.strip()))
+        self.chapter_list = self.split_chapters(self.novel_content)
 
 
     def updateNovelContent(self):

app.py

@@ -41,7 +41,7 @@ def set_api_model(model):
 # 修改获取章节名的函数
 def get_chapter_name(content):
     # 查找最后一个章节标题,支持数字和汉字表示的章节
-    matches = re.findall(r'##\s*第([一二三四五六七八九十百千万亿\d]+)章[:：]?\s*(.+)', content)
+    matches = re.findall(r'(?:##?|)?\s*第([一二三四五六七八九十百千万亿\d]+)章[:：]?\s*(.+)', content)
     if matches:
         chapter_num, chapter_title = matches[-1]
         # 如果章节号是数字,直接使用;如果是汉字,转换为数字