From 58d5c418eafc772200059d6e000c0e715c1a6754 Mon Sep 17 00:00:00 2001
From: McCloudS <64094529+McCloudS@users.noreply.github.com>
Date: Wed, 27 Mar 2024 08:45:12 -0600
Subject: [PATCH 1/2] Update docker-compose.yml

---
 docker-compose.yml | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

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"

From 96d2426f5e93fab38d32de32168568c581857afa Mon Sep 17 00:00:00 2001
From: McCloudS <64094529+McCloudS@users.noreply.github.com>
Date: Wed, 27 Mar 2024 08:57:08 -0600
Subject: [PATCH 2/2] Fixed subextension not getting updated on setting change.

---
 subgen.py | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

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"