[BUGFIX] signal only works in main thread of the main interpreter

Models/languageClassification.py

@@ -18,14 +18,26 @@
     }
 }
 
+model = None
 
-def classify(text):
+
+def download_model():
     pretrained_lang_model_file = Path(ct_model_path / "lid218e.bin")
-    if not pretrained_lang_model_file.exists():
+    if not pretrained_lang_model_file.is_file():
         print(f"Downloading LID (language identification) model...")
         downloader.download_extract(MODEL_LINKS["lid218e"]["urls"], str(ct_model_path.resolve()), MODEL_LINKS["lid218e"]["checksum"])
 
-    model = fasttext.load_model(str(pretrained_lang_model_file.resolve()))
+
+def classify(text):
+    global model
+    pretrained_lang_model_file = Path(ct_model_path / "lid218e.bin")
+    if not pretrained_lang_model_file.is_file():
+        print(f"Error: LID (language identification model missing.")
+        return ""
+
+    if model is None:
+        print(f"Loading LID (language identification) model...")
+        model = fasttext.load_model(str(pretrained_lang_model_file.resolve()))
 
     text = text.replace("\n", " ")
     predictions = model.predict(text, k=1)

audioWhisper.py

@@ -13,6 +13,8 @@
 import settings
 import remote_opener
 from Models.TextTranslation import texttranslate
+from Models import languageClassification
+from Models.LLM import flanLanguageModel
 import pyaudiowpatch as pyaudio
 from whisper import available_models, audio as whisper_audio
 
@@ -165,6 +167,12 @@ def main(ctx, devices, device_index, sample_rate, dynamic_energy, open_browser,
             print(e)
             pass
 
+    # Load language identification dependencies
+    languageClassification.download_model()
+
+    # Load FLAN-T5 dependencies
+    flanLanguageModel.init()
+
     # load the speech recognizer and set the initial energy threshold and pause threshold
     r = sr.Recognizer()
     r.energy_threshold = energy