Fix Windows DLL loading (chidiwilliams#162)

.github/workflows/ci.yml

@@ -3,7 +3,7 @@ name: CI
 on:
   push:
     branches:
-      - master
+      - main
     tags:
       - '*'
   pull_request:

gui.py

@@ -21,7 +21,7 @@
 
 from __version__ import VERSION
 from transcriber import FileTranscriber, OutputFormat, RecordingTranscriber
-from whispr import Task
+from whispr import LOADED_WHISPER_DLL, Task
 
 APP_NAME = 'Buzz'
 
@@ -553,6 +553,8 @@ def __init__(self, parent: Optional[QWidget], *args):
         super().__init__('Buzz', 'Buzz', parent, *args)
 
     def enable_ggml_inference(self):
+        if LOADED_WHISPER_DLL is False:
+            return False
         return self.value(self.ENABLE_GGML_INFERENCE, False)
 
 
@@ -804,6 +806,7 @@ def __init__(self, title: str, w: int, h: int, parent: Optional[QWidget], *args)
             bool(self.settings.enable_ggml_inference()))
         enable_ggml_inference_action.triggered.connect(
             self.on_toggle_enable_ggml_inference)
+        enable_ggml_inference_action.setDisabled(LOADED_WHISPER_DLL is False)
 
         settings_menu = menu.addMenu("&Settings")
         settings_menu.addAction(enable_ggml_inference_action)

main.py

@@ -17,6 +17,9 @@
     os.path.abspath(__file__)))
 os.environ["PATH"] += os.pathsep + app_dir
 
+# Add the app directory to the DLL list: https://stackoverflow.com/a/64303856
+os.add_dll_directory(app_dir)
+
 if __name__ == "__main__":
     # Fixes opening new window when app has been frozen on Windows:
     # https://stackoverflow.com/a/33979091

whispr.py

@@ -18,9 +18,18 @@
 from tqdm import tqdm
 from whisper import Whisper
 
-import whisper_cpp
 from conn import pipe_stderr
 
+# Catch exception from whisper.dll not getting loaded.
+# TODO: Remove flag and try-except when issue with loading
+# the DLL in some envs is fixed.
+LOADED_WHISPER_DLL = False
+try:
+    import whisper_cpp
+    LOADED_WHISPER_DLL = True
+except ImportError:
+    logging.exception('')
+
 
 class Stopped(Exception):
     pass
@@ -41,7 +50,8 @@ class Task(enum.Enum):
 def whisper_cpp_params(
         language: str, task: Task, word_level_timings: bool,
         print_realtime=False, print_progress=False,):
-    params = whisper_cpp.whisper_full_default_params(whisper_cpp.WHISPER_SAMPLING_GREEDY)
+    params = whisper_cpp.whisper_full_default_params(
+        whisper_cpp.WHISPER_SAMPLING_GREEDY)
     params.print_realtime = print_realtime
     params.print_progress = print_progress
     params.language = whisper_cpp.String(language.encode('utf-8'))