From 0e08f12bed31c170c189fad3e1713257a42ea439 Mon Sep 17 00:00:00 2001
From: starylan <starylan@outlook.com>
Date: Sat, 18 May 2024 23:10:14 +0800
Subject: [PATCH] Filter unsupported languages

---
 GPT_SoVITS/prepare_datasets/1-get-text.py | 11 +++++++----
 webui.py                                  |  7 ++++---
 2 files changed, 11 insertions(+), 7 deletions(-)

diff --git a/GPT_SoVITS/prepare_datasets/1-get-text.py b/GPT_SoVITS/prepare_datasets/1-get-text.py
index b2413826..2305b377 100644
--- a/GPT_SoVITS/prepare_datasets/1-get-text.py
+++ b/GPT_SoVITS/prepare_datasets/1-get-text.py
@@ -117,12 +117,15 @@ if os.path.exists(txt_path) == False:
         try:
             wav_name, spk_name, language, text = line.split("|")
             # todo.append([name,text,"zh"])
-            todo.append(
-                [wav_name, text, language_v1_to_language_v2.get(language, language)]
-            )
+            if language in language_v1_to_language_v2.keys():
+                todo.append(
+                    [wav_name, text, language_v1_to_language_v2.get(language, language)]
+                )
+            else:
+                print(f"\033[33m[Waring] The {language = } of {wav_name} is not supported for training.\033[0m")
         except:
             print(line, traceback.format_exc())
-
+    assert len(todo) > 0, f"\033[31m[Error] Part {i_part}: No data loaded from {inp_text}.\033[0m"
     process(todo, res)
     opt = []
     for name, phones, word2ph, norm_text in res:
diff --git a/webui.py b/webui.py
index e1c36e1e..17032be1 100644
--- a/webui.py
+++ b/webui.py
@@ -411,9 +411,10 @@ def open1a(inp_text,inp_wav_dir,exp_name,gpu_numbers,bert_pretrained_dir):
         opt = []
         for i_part in range(all_parts):
             txt_path = "%s/2-name2text-%s.txt" % (opt_dir, i_part)
-            with open(txt_path, "r", encoding="utf8") as f:
-                opt += f.read().strip("\n").split("\n")
-            os.remove(txt_path)
+            if os.path.exists(txt_path):
+                with open(txt_path, "r", encoding="utf8") as f:
+                    opt += f.read().strip("\n").split("\n")
+                os.remove(txt_path)
         path_text = "%s/2-name2text.txt" % opt_dir
         with open(path_text, "w", encoding="utf8") as f:
             f.write("\n".join(opt) + "\n")