jhj0517 commited on
Commit
2240740
·
1 Parent(s): 4e9280e

Add constants and apply

Browse files
app.py CHANGED
@@ -53,7 +53,7 @@ class App:
53
  with gr.Row():
54
  dd_model = gr.Dropdown(choices=self.whisper_inf.available_models, value=whisper_params["model_size"],
55
  label="Model")
56
- dd_lang = gr.Dropdown(choices=["Automatic Detection"] + self.whisper_inf.available_langs,
57
  value=whisper_params["lang"], label="Language")
58
  dd_file_format = gr.Dropdown(choices=["SRT", "WebVTT", "txt"], value="SRT", label="File Format")
59
  with gr.Row():
 
53
  with gr.Row():
54
  dd_model = gr.Dropdown(choices=self.whisper_inf.available_models, value=whisper_params["model_size"],
55
  label="Model")
56
+ dd_lang = gr.Dropdown(choices=[AUTOMATIC_DETECTION] + self.whisper_inf.available_langs,
57
  value=whisper_params["lang"], label="Language")
58
  dd_file_format = gr.Dropdown(choices=["SRT", "WebVTT", "txt"], value="SRT", label="File Format")
59
  with gr.Row():
modules/translation/deepl_api.py CHANGED
@@ -5,6 +5,7 @@ from datetime import datetime
5
  import gradio as gr
6
 
7
  from modules.utils.paths import TRANSLATION_OUTPUT_DIR, DEFAULT_PARAMETERS_CONFIG_PATH
 
8
  from modules.utils.subtitle_manager import *
9
  from modules.utils.files_manager import load_yaml, save_yaml
10
 
@@ -50,7 +51,7 @@ DEEPL_AVAILABLE_TARGET_LANGS = {
50
  }
51
 
52
  DEEPL_AVAILABLE_SOURCE_LANGS = {
53
- 'Automatic Detection': None,
54
  'Bulgarian': 'BG',
55
  'Czech': 'CS',
56
  'Danish': 'DA',
 
5
  import gradio as gr
6
 
7
  from modules.utils.paths import TRANSLATION_OUTPUT_DIR, DEFAULT_PARAMETERS_CONFIG_PATH
8
+ from modules.utils.constants import AUTOMATIC_DETECTION
9
  from modules.utils.subtitle_manager import *
10
  from modules.utils.files_manager import load_yaml, save_yaml
11
 
 
51
  }
52
 
53
  DEEPL_AVAILABLE_SOURCE_LANGS = {
54
+ AUTOMATIC_DETECTION: None,
55
  'Bulgarian': 'BG',
56
  'Czech': 'CS',
57
  'Danish': 'DA',
modules/utils/constants.py ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ from gradio_i18n import Translate, gettext as _
2
+
3
+ AUTOMATIC_DETECTION = _("Automatic Detection")
modules/whisper/whisper_base.py CHANGED
@@ -14,6 +14,7 @@ from dataclasses import astuple
14
  from modules.uvr.music_separator import MusicSeparator
15
  from modules.utils.paths import (WHISPER_MODELS_DIR, DIARIZATION_MODELS_DIR, OUTPUT_DIR, DEFAULT_PARAMETERS_CONFIG_PATH,
16
  UVR_MODELS_DIR)
 
17
  from modules.utils.subtitle_manager import get_srt, get_vtt, get_txt, write_file, safe_filename
18
  from modules.utils.youtube_manager import get_ytdata, get_ytaudio
19
  from modules.utils.files_manager import get_media_files, format_gradio_files, load_yaml, save_yaml
@@ -107,7 +108,7 @@ class WhisperBase(ABC):
107
 
108
  if params.lang is None:
109
  pass
110
- elif params.lang == "Automatic Detection":
111
  params.lang = None
112
  else:
113
  language_code_dict = {value: key for key, value in whisper.tokenizer.LANGUAGES.items()}
 
14
  from modules.uvr.music_separator import MusicSeparator
15
  from modules.utils.paths import (WHISPER_MODELS_DIR, DIARIZATION_MODELS_DIR, OUTPUT_DIR, DEFAULT_PARAMETERS_CONFIG_PATH,
16
  UVR_MODELS_DIR)
17
+ from modules.utils.constants import AUTOMATIC_DETECTION
18
  from modules.utils.subtitle_manager import get_srt, get_vtt, get_txt, write_file, safe_filename
19
  from modules.utils.youtube_manager import get_ytdata, get_ytaudio
20
  from modules.utils.files_manager import get_media_files, format_gradio_files, load_yaml, save_yaml
 
108
 
109
  if params.lang is None:
110
  pass
111
+ elif params.lang == AUTOMATIC_DETECTION:
112
  params.lang = None
113
  else:
114
  language_code_dict = {value: key for key, value in whisper.tokenizer.LANGUAGES.items()}
modules/whisper/whisper_parameter.py CHANGED
@@ -3,6 +3,8 @@ import gradio as gr
3
  from typing import Optional, Dict
4
  import yaml
5
 
 
 
6
 
7
  @dataclass
8
  class WhisperParameters:
@@ -306,7 +308,7 @@ class WhisperValues:
306
  data = {
307
  "whisper": {
308
  "model_size": self.model_size,
309
- "lang": "Automatic Detection" if self.lang is None else self.lang,
310
  "is_translate": self.is_translate,
311
  "beam_size": self.beam_size,
312
  "log_prob_threshold": self.log_prob_threshold,
 
3
  from typing import Optional, Dict
4
  import yaml
5
 
6
+ from modules.utils.constants import AUTOMATIC_DETECTION
7
+
8
 
9
  @dataclass
10
  class WhisperParameters:
 
308
  data = {
309
  "whisper": {
310
  "model_size": self.model_size,
311
+ "lang": AUTOMATIC_DETECTION if self.lang is None else self.lang,
312
  "is_translate": self.is_translate,
313
  "beam_size": self.beam_size,
314
  "log_prob_threshold": self.log_prob_threshold,