Update app.py

app.py

@@ -19,7 +19,7 @@ import warnings
 
 warnings.filterwarnings("ignore", category=UserWarning, module="gradio.blocks")
 
-
+from datetime import datetime
 import re
 import torch
 import utils
@@ -29,6 +29,9 @@ import numpy as np
 from tools.sentence import extrac, is_japanese, is_chinese
 import sys, os
 import math
+
+from tools.translate import translate
+
 net_g = None
 
 cara_list = ["ひまり","たえ","彩","日菜","美咲","ましろ","燐子","香子","珠緒","たえ"]
@@ -53,7 +56,6 @@ if sys.platform == "darwin" and torch.backends.mps.is_available():
     device = "mps"
     os.environ["PYTORCH_ENABLE_MPS_FALLBACK"] = "1"
 else:
-    #device = "cuda"
     device = "cpu"
 
 def generate_audio(
@@ -73,7 +75,8 @@ def generate_audio(
                 language = "JP"
             elif is_chinese(text):
                 language = "ZH"
-        print(text+":"+language)
+        current_time = datetime.now()
+        print(str(current_time)+':'+str(speaker)+ text+":"+language)
         audio = infer(
             text,
             sdp_ratio=sdp_ratio,
@@ -152,9 +155,19 @@ if __name__ == "__main__":
         for filename in filenames:
             modelPaths.append(os.path.join(dirpath, filename))
     with gr.Blocks() as app:
-        gr.Markdown(
-             f"少歌邦邦全员TTS,使用本模型请严格遵守法律法规!\现已支持日语bert推理<a href='https://huggingface.co/spaces/Mahiruoshi/MyGO_VIts-bert/'>上一版本模型</a>及V1.0版本模型\n 发布二创作品请注明项目和本模型作者<a href='https://space.bilibili.com/19874615/'>B站@Mahiroshi</a>及项目链接\n从 <a href='https://nijigaku.top/2023/10/03/BangDreamTTS/'>我的博客站点</a> 查看使用说明</a>"
-        )
+        gr.Markdown(value="""
+            少歌邦邦全员在线语音合成（Bert-Vits2）\n
+            作者：B站@Mahiroshi https://space.bilibili.com/19874615\n
+            声音归属：BangDream及少歌手游\n
+            Bert-VITS2项目：https://github.com/Stardust-minus/Bert-VITS2\n
+            使用参考: https://nijigaku.top/2023/10/03/BangDreamTTS\n
+            数据集制作: https://huggingface.co/spaces/Mahiruoshi/BangDream-Bert-VITS2/tree/main/%E7%88%AC%E8%99%AB
+            服务器启动示例: https://huggingface.co/spaces/Mahiruoshi/BangDream-Bert-VITS2/blob/main/server.py\n
+            使用本模型请严格遵守法律法规!禁止生成任何有损声优或者企划的内容!!!!!\n
+            このモデルを使用する際は法律法規を厳守してください！声優や企画に損害を与える内容の生成は禁止です!!!!!\n
+            Please strictly follow the laws in your country and regulations when using this model! It is prohibited to generate any content that is harmful to others!!!!!\n
+            发布二创作品请标注本项目作者及链接、作品使用Bert-VITS2 AI生成！\n                
+            """)
         for band in BandList:
             with gr.TabItem(band):
                 for name in BandList[band]:
@@ -170,6 +183,7 @@ if __name__ == "__main__":
                                 length_scale = gr.Slider(
                                         minimum=0.1, maximum=2, value=1, step=0.01, label="语速调节"
                                     )
+                                LongSentence = gr.Checkbox(value=True, label="自动拆分句子")
                                 with gr.Accordion(label="切换模型", open=False):
                                     modelstrs = gr.Dropdown(label = "模型", choices = modelPaths, value = modelPaths[0], type = "value")
                                     btnMod = gr.Button("载入模型")
@@ -183,6 +197,9 @@ if __name__ == "__main__":
                                 )                                
                                 btn = gr.Button("点击生成", variant="primary")
                                 audio_output = gr.Audio(label="Output Audio")
+                                btntran = gr.Button("快速中翻日")
+                                translateResult = gr.TextArea("从这复制翻译后的文本")
+                                btntran.click(translate, inputs=[text], outputs = [translateResult])
                                 with gr.Accordion(label="其它参数设定", open=False):
                                     sdp_ratio = gr.Slider(
                                     minimum=0, maximum=1, value=0.2, step=0.01, label="SDP/DP混合比"
@@ -193,7 +210,6 @@ if __name__ == "__main__":
                                     noise_scale_w = gr.Slider(
                                         minimum=0.1, maximum=2, value=0.8, step=0.01, label="音素长度"
                                     )
-                                    LongSentence = gr.Checkbox(value=True, label="Generate LongSentence")
                                     language = gr.Dropdown(
                                     choices=languages, value=languages[0], label="选择语言(默认自动)"
                                     )