Update app.py

app.py

@@ -14,26 +14,28 @@ from loguru import logger
 ssl._create_default_https_context = ssl._create_unverified_context
 import nltk
 
-nltk.download('cmudict')
+# 檢查是否已下載資源，若未下載則進行下載
+nltk_data_path = os.path.expanduser('~/nltk_data')
+if not os.path.exists(os.path.join(nltk_data_path, 'corpora/cmudict.zip')):
+    nltk.download('cmudict', download_dir=nltk_data_path)
 
-# 下載 NLTK 所需的資料集
-nltk.download('averaged_perceptron_tagger')
+if not os.path.exists(os.path.join(nltk_data_path, 'taggers/averaged_perceptron_tagger.zip')):
+    nltk.download('averaged_perceptron_tagger', download_dir=nltk_data_path)
 
 from parrots import TextToSpeech
 
+# 設定裝置與模式
 device = "cuda" if torch.cuda.is_available() else "cpu"
 logger.info(f"device: {device}")
 half = True if device == "cuda" else False
 
+# 初始化語音合成模型
 m = TextToSpeech(speaker_model_path="DeepLearning101/GPT-SoVITS_TWMAN", speaker_name="TWMAN", device=device, half=half)
-m.predict(text="台灣南波萬。Taiwan Number One.", text_language="auto", output_path="output_audio.wav")
-assert os.path.exists("output_audio.wav"), "output_audio.wav not found"
-
 
+# 用於檢查和生成語音的音訊檔案
 def get_text_hash(text: str):
     return hashlib.md5(text.encode('utf-8')).hexdigest()
 
-
 def do_tts_wav_predict(text: str, output_path: str = None):
     if output_path is None:
         output_path = f"output_audio_{get_text_hash(text)}.wav"
@@ -41,30 +43,35 @@ def do_tts_wav_predict(text: str, output_path: str = None):
         m.predict(text, text_language="auto", output_path=output_path)
     return output_path
 
-
+# 建立 Gradio WebUI
 with gr.Blocks(title="TTS WebUI") as app:
-    gr.Markdown(value="""
-    # <center>線上語音合成；speaker:TWMAN\n
-
-     ### <center>語音處理：https://www.twman.org/AI/ASR\n
-     ### <center>那些語音處理 (Speech Processing) 踩的坑 https://blog.twman.org/2021/04/ASR.html\n
-     ### <center>parrots專案：https://github.com/shibing624/parrots\n
-     ### <center>使用模型：https://github.com/RVC-Boss/GPT-SoVITS\n
-     ### <center>使用本模型請嚴格遵守法規！ 發布二創作品請標註本專案作者及連結、作品使用GPT-SoVITS AI生成！ \n
-     ### <center>⚠️在線端不穩定且生成速度較慢，建議使用parrots本地推理！ \n
-                """)
-
+    gr.Markdown("""
+    # 線上語音合成 (TWMAN)
+    #### 請嚴格遵守法規，發布二創作品請標註本專案作者及連結，並標註生成工具 GPT-SoVITS AI！
+    ⚠️ 注意：在線生成可能較慢，建議在本地進行推理。 
+    
+    更多相關內容：
+    - [語音處理技術](https://www.twman.org/AI/ASR)
+    - [語音處理常見問題](https://blog.twman.org/2021/04/ASR.html)
+    - [Parrots專案](https://github.com/shibing624/parrots)
+    - [模型使用說明](https://github.com/RVC-Boss/GPT-SoVITS)
+    """)
+
+    # 設定語音合成輸入與按鈕
     with gr.Group():
-        gr.Markdown(value="*請在這裡輸入要進行語音合成的文字")
+        gr.Markdown("*請在下方輸入要進行語音合成的文字*")
         with gr.Row():
-            text = gr.Textbox(label="想語音合成的文字(100字以内)或使用預設", value="床前明月光，疑是地上霜。舉頭望明月，低頭思故鄉。", placeholder="請輸入您想要的文字", lines=3)
+            text = gr.Textbox(label="想語音合成的文字 (100字以内)", value="床前明月光，疑是地上霜。舉頭望明月，低頭思故鄉。", placeholder="請輸入您想要的文字", lines=3)
             inference_button = gr.Button("語音合成", variant="primary")
             output = gr.Audio(label="合成的語音")
+
+        # 設定按鈕點擊事件
         inference_button.click(
             do_tts_wav_predict,
             [text],
             [output],
         )
 
+# 啟動 Gradio 應用
 app.queue(max_size=10)
-app.launch(inbrowser=True)
+app.launch(share=True, inbrowser=True)