From 0771a9781cd642ea791c0e99029f496fbb96a029 Mon Sep 17 00:00:00 2001 From: McCloudS <64094529+McCloudS@users.noreply.github.com> Date: Tue, 7 Jan 2025 13:51:43 -0700 Subject: [PATCH] Fix subtitle naming for translate Always default the subtitle name to eng, unless namesublang set to something else. --- subgen.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/subgen.py b/subgen.py index 52f3897..acc587e 100644 --- a/subgen.py +++ b/subgen.py @@ -1,4 +1,4 @@ -subgen_version = '2024.12.10' +subgen_version = '2024.12.11' from language_code import LanguageCode from datetime import datetime @@ -57,7 +57,7 @@ path_mapping_to = os.getenv('PATH_MAPPING_TO', r'/Volumes/TV') model_location = os.getenv('MODEL_PATH', './models') monitor = convert_to_bool(os.getenv('MONITOR', False)) transcribe_folders = os.getenv('TRANSCRIBE_FOLDERS', '') -transcribe_or_translate = os.getenv('TRANSCRIBE_OR_TRANSLATE', 'transcribe') +transcribe_or_translate = os.getenv('TRANSCRIBE_OR_TRANSLATE', 'transcribe').lower() clear_vram_on_complete = convert_to_bool(os.getenv('CLEAR_VRAM_ON_COMPLETE', True)) compute_type = os.getenv('COMPUTE_TYPE', 'auto') append = convert_to_bool(os.getenv('APPEND', False)) @@ -729,6 +729,9 @@ def define_subtitle_language_naming(language: LanguageCode, type): """ if namesublang: return namesublang + if transcribe_or_translate == 'translate' + return 'eng' + # If we are translating, then we ALWAYS output an english file. switch_dict = { "ISO_639_1": language.to_iso_639_1, "ISO_639_2_T": language.to_iso_639_2_t,