Merge pull request #341 from jhj0517/feature/support-i18n

Support localization
jhj0517 · Oct 20, 2024 · ed6e918 · ed6e918
2 parents 8dc115b + 84fd983
commit ed6e918
Show file tree

Hide file tree

Showing 8 changed files with 563 additions and 203 deletions.
diff --git a/app.py b/app.py
diff --git a/configs/translation.yaml b/configs/translation.yaml
diff --git a/modules/translation/deepl_api.py b/modules/translation/deepl_api.py
@@ -5,6 +5,7 @@
 import gradio as gr
 
 from modules.utils.paths import TRANSLATION_OUTPUT_DIR, DEFAULT_PARAMETERS_CONFIG_PATH
+from modules.utils.constants import AUTOMATIC_DETECTION
 from modules.utils.subtitle_manager import *
 from modules.utils.files_manager import load_yaml, save_yaml
 
@@ -50,7 +51,7 @@
 }
 
 DEEPL_AVAILABLE_SOURCE_LANGS = {
-    'Automatic Detection': None,
+    AUTOMATIC_DETECTION: None,
     'Bulgarian': 'BG',
     'Czech': 'CS',
     'Danish': 'DA',

diff --git a/modules/utils/constants.py b/modules/utils/constants.py
@@ -0,0 +1,3 @@
+from gradio_i18n import Translate, gettext as _
+
+AUTOMATIC_DETECTION = _("Automatic Detection")
diff --git a/modules/utils/paths.py b/modules/utils/paths.py
@@ -10,6 +10,7 @@
 UVR_MODELS_DIR = os.path.join(MODELS_DIR, "UVR", "MDX_Net_Models")
 CONFIGS_DIR = os.path.join(WEBUI_DIR, "configs")
 DEFAULT_PARAMETERS_CONFIG_PATH = os.path.join(CONFIGS_DIR, "default_parameters.yaml")
+I18N_YAML_PATH = os.path.join(CONFIGS_DIR, "translation.yaml")
 OUTPUT_DIR = os.path.join(WEBUI_DIR, "outputs")
 TRANSLATION_OUTPUT_DIR = os.path.join(OUTPUT_DIR, "translations")
 UVR_OUTPUT_DIR = os.path.join(OUTPUT_DIR, "UVR")

diff --git a/modules/whisper/whisper_base.py b/modules/whisper/whisper_base.py
@@ -14,6 +14,7 @@
 from modules.uvr.music_separator import MusicSeparator
 from modules.utils.paths import (WHISPER_MODELS_DIR, DIARIZATION_MODELS_DIR, OUTPUT_DIR, DEFAULT_PARAMETERS_CONFIG_PATH,
                                  UVR_MODELS_DIR)
+from modules.utils.constants import AUTOMATIC_DETECTION
 from modules.utils.subtitle_manager import get_srt, get_vtt, get_txt, write_file, safe_filename
 from modules.utils.youtube_manager import get_ytdata, get_ytaudio
 from modules.utils.files_manager import get_media_files, format_gradio_files, load_yaml, save_yaml
@@ -107,7 +108,7 @@ def run(self,
 
         if params.lang is None:
             pass
-        elif params.lang == "Automatic Detection":
+        elif params.lang == AUTOMATIC_DETECTION:
             params.lang = None
         else:
             language_code_dict = {value: key for key, value in whisper.tokenizer.LANGUAGES.items()}

diff --git a/modules/whisper/whisper_parameter.py b/modules/whisper/whisper_parameter.py
@@ -3,6 +3,8 @@
 from typing import Optional, Dict
 import yaml
 
+from modules.utils.constants import AUTOMATIC_DETECTION
+
 
 @dataclass
 class WhisperParameters:
@@ -306,7 +308,7 @@ def to_yaml(self) -> Dict:
         data = {
             "whisper": {
                 "model_size": self.model_size,
-                "lang": "Automatic Detection" if self.lang is None else self.lang,
+                "lang": AUTOMATIC_DETECTION.unwrap() if self.lang is None else self.lang,
                 "is_translate": self.is_translate,
                 "beam_size": self.beam_size,
                 "log_prob_threshold": self.log_prob_threshold,

diff --git a/requirements.txt b/requirements.txt
@@ -11,6 +11,7 @@ git+https://github.com/jhj0517/jhj0517-whisper.git
 faster-whisper==1.0.3
 transformers
 gradio
+git+https://github.com/jhj0517/gradio-i18n.git@fix/encoding-error
 pytubefix
 ruamel.yaml==0.18.6
 pyannote.audio==3.3.1
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1,3 @@
		from gradio_i18n import Translate, gettext as _

		AUTOMATIC_DETECTION = _("Automatic Detection")