mirror of
https://github.com/RVC-Boss/GPT-SoVITS.git
synced 2025-04-06 03:57:44 +08:00
62 lines
1.6 KiB
Python
62 lines
1.6 KiB
Python
import json
|
|
import locale
|
|
import os
|
|
|
|
import gradio_i18n
|
|
|
|
I18N_JSON_DIR: os.PathLike = os.path.join(
|
|
os.path.dirname(os.path.relpath(__file__)), "locale"
|
|
)
|
|
|
|
|
|
def load_language_list(language):
|
|
with open(
|
|
os.path.join(I18N_JSON_DIR, f"{language}.json"), "r", encoding="utf-8"
|
|
) as f:
|
|
language_list = json.load(f)
|
|
return language_list
|
|
|
|
|
|
def scan_language_list():
|
|
language_list = []
|
|
for name in os.listdir(I18N_JSON_DIR):
|
|
if name.endswith(".json"):
|
|
language_list.append(name.split(".")[0])
|
|
return language_list
|
|
|
|
|
|
class I18nAuto:
|
|
def __init__(self, language=None):
|
|
if language in ["Auto", None] or not os.path.exists(
|
|
os.path.join(I18N_JSON_DIR, f"{language}.json")
|
|
):
|
|
self.language = "Auto"
|
|
else:
|
|
self.language = language
|
|
self.language_map = load_language_list(language)
|
|
|
|
def __call__(self, key):
|
|
if self.language == "Auto":
|
|
return gradio_i18n.gettext(key)
|
|
else:
|
|
return self.language_map.get(key, key)
|
|
|
|
def __repr__(self):
|
|
return "Use Language: " + self.language
|
|
|
|
def translate_blocks(self, block):
|
|
translation = {}
|
|
for lang in scan_language_list():
|
|
short_lang = lang.partition("_")[0]
|
|
if short_lang == "zh" and "CN" not in lang:
|
|
continue
|
|
|
|
translation[short_lang] = load_language_list(lang)
|
|
|
|
gradio_i18n.translate_blocks(block, translation)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
i18n = I18nAuto(language="en_US")
|
|
print(i18n)
|