diff --git a/docker-compose.yml b/docker-compose.yml index c90a012..c9ca94e 100644 --- a/docker-compose.yml +++ b/docker-compose.yml @@ -25,11 +25,13 @@ services: - "PATH_MAPPING_TO=/Volumes/TV" - "TRANSCRIBE_DEVICE=cpu" - "CLEAR_VRAM_ON_COMPLETE=True" - - "HF_TRANSFORMERS=False" - - "HF_BATCH_SIZE=24" - "MODEL_PATH=./models" - "UPDATE=False" - "APPEND=False" + - "USE_MODEL_PROMPT=False" + - "CUSTOM_MODEL_PROMPT=" + - "LRC_FOR_AUDIO_FILES=True" + - "CUSTOM_REGROUP=cm_sl=84_sl=42++++++1" volumes: - "${TV}:/tv" - "${MOVIES}:/movies" diff --git a/subgen.py b/subgen.py index f361ba3..3adc458 100644 --- a/subgen.py +++ b/subgen.py @@ -24,6 +24,8 @@ import av import ffmpeg import whisper import re +from watchdog.observers.polling import PollingObserver as Observer +from watchdog.events import FileSystemEventHandler def convert_to_bool(in_bool): # Convert the input to string and lower case, then check against true values @@ -60,6 +62,7 @@ def update_env_variables(): global transcribe_folders, transcribe_or_translate, force_detected_language_to global clear_vram_on_complete, compute_type, append, reload_script_on_change global model_prompt, custom_model_prompt, lrc_for_audio_files, custom_regroup + global subextension, subextensionSDH plextoken = os.getenv('PLEXTOKEN', 'token here') plexserver = os.getenv('PLEXSERVER', 'http://192.168.1.111:32400') @@ -97,18 +100,15 @@ def update_env_variables(): if transcribe_device == "gpu": transcribe_device = "cuda" + + subextension = f".subgen.{whisper_model.split('.')[0]}.{namesublang}.srt" + subextensionSDH = f".subgen.{whisper_model.split('.')[0]}.{namesublang}.sdh.srt" update_env_variables() -if monitor: - from watchdog.observers.polling import PollingObserver as Observer - from watchdog.events import FileSystemEventHandler - app = FastAPI() model = None files_to_transcribe = [] -subextension = f".subgen.{whisper_model.split('.')[0]}.{namesublang}.srt" -subextensionSDH = f".subgen.{whisper_model.split('.')[0]}.{namesublang}.sdh.srt" in_docker = os.path.exists('/.dockerenv') docker_status = "Docker" if in_docker else "Standalone"