Update app.py

app.py

@@ -273,7 +273,7 @@ def extract_text_from_pdf(file_path):
         content = [page.extract_text() for page in reader.pages]
     return '\n'.join(content)
 
-def extract_text_from_game(data):
+def extract_text_from_game2(data):
     current_content = []
 
     def _extract(data, current_data=None):
@@ -283,26 +283,17 @@ def extract_text_from_game(data):
             current_data = {}
 
         if isinstance(data, dict):
-            if 'windowDisplayName' in data:
-                current_data['windowDisplayName'] = data['windowDisplayName']
-            if 'body' in data:
-                current_data['body'] = data['body'].replace('\n', '')
-            if 'voiceId' in data:
-                current_data['voiceId'] = data['voiceId']
+            if 'name' in data and 'body' in data:
+                current_name = data['name']
+                current_body = data['body'].replace('\n', '')
+                current_content.append(f"{current_name}|{current_body}")
 
-                valid_data = all(current_data.get(k) for k in ['windowDisplayName', 'body', 'voiceId'])
-                valid_displayname = "・" not in current_data.get('windowDisplayName', "")
-                valid_body = bool(re.sub(r'[^\w]', '', current_data.get('body', "")))
-
-                if valid_data and valid_displayname and valid_body:
-                    current_content.append(f"{current_data['windowDisplayName']}|{current_data['body']}")
-
-            for key in data:
-                _extract(data[key], dict(current_data)) 
+            for key, value in data.items():
+                _extract(value, dict(current_data))
 
         elif isinstance(data, list):
             for item in data:
-                _extract(item, dict(current_data)) 
+                _extract(item, dict(current_data))
 
     _extract(data)
     return '\n'.join(current_content)
@@ -320,7 +311,7 @@ def extract_text_from_file(inputFile):
     elif file_extension == ".asset":
         with open(inputFile, 'r', encoding='utf-8') as f:
             content =  json.load(f)
-        return extract_text_from_game(content)
+        return extract_text_from_game2(content) if extract_text_from_game(content) != '' else extract_text_from_game(content)
     else:
         raise ValueError(f"Unsupported file format: {file_extension}")
 
@@ -396,7 +387,7 @@ if __name__ == "__main__":
     ]
     with gr.Blocks() as app:
         gr.Markdown(
-            '# Bang Dream全员TTS,使用本模型请严格遵守法律法规！\n发布二创作品请标注本项目作者及链接、作品使用Bert-VITS2 AI生成！'
+            '# Bang Dream全员TTS,使用本模型请严格遵守法律法规！\n发布二创作品请标注本项目作者及链接、作品使用Bert-VITS2 AI生成！\n <a href='https://nijigaku.top/2023/10/03/BangDreamTTS/'>使用以及本地部署指南</a> '
         )
         for band in BandList:
             with gr.TabItem(band):
@@ -453,7 +444,7 @@ if __name__ == "__main__":
                 with gr.Row():
                     with gr.Column():
                         gr.Markdown(
-                                        f"从 <a href='filelists'>filelists文件夹</a> 下载示例\n游戏脚本见<a href='https://bestdori.com/tool/explorer/asset/cn/scenario'>bestdori</a>"
+                                        f"从 <a href='https://huggingface.co/spaces/Mahiruoshi/BangDream-Bert-VITS2/filelists'>filelists文件夹</a> 查看示例\n游戏脚本见<a href='https://bestdori.com/tool/explorer/asset/cn/scenario'>bestdori</a>"
                                     )
                         inputFile = gr.inputs.File(label="上传游戏脚本(日文)、中文脚本(需设置角色对应关系)、自制文、(需设置角色对应关系")
                         groupSize = gr.Slider(