GameSentenceMiner 2.0.0__py3-none-any.whl → 2.0.0.dev1__py3-none-any.whl

GameSentenceMiner/config_gui.py

@@ -7,8 +7,6 @@ from . import configuration
 from . import obs
 from .configuration import *
 
-TOML_CONFIG_FILE = '../../config.toml'
-CONFIG_FILE = os.path.join(os.path.dirname(__file__), 'config.json')
 settings_saved = False
 on_save = []
 
@@ -186,7 +184,7 @@ class ConfigApp:
             self.master_config.set_config_for_profile(current_profile, config)
 
         # Serialize the config instance to JSON
-        with open('../../config.json', 'w') as file:
+        with open(get_config_path(), 'w') as file:
             file.write(self.master_config.to_json(indent=4))
 
         print("Settings saved successfully!")

GameSentenceMiner/configuration.py

@@ -216,7 +216,7 @@ class ProfileConfig:
 
         self.anki.anki_custom_fields = config_data.get('anki_custom_fields', {})
 
-        with open('config.json', 'w') as f:
+        with open(get_config_path(), 'w') as f:
             f.write(self.to_json(indent=4))
             print(
                 'config.json successfully generated from previous settings. config.toml will no longer be used.')
@@ -247,38 +247,33 @@ class Config:
     def get_all_profile_names(self):
         return list(self.configs.keys())
 
-
-logger = logging.getLogger("GameSentenceMiner")
-logger.setLevel(logging.DEBUG)  # Set the base level to DEBUG so that all messages are captured
-
-# Create console handler with level INFO
-console_handler = logging.StreamHandler()
-console_handler.setLevel(logging.INFO)
-
-# Create rotating file handler with level DEBUG
-file_handler = RotatingFileHandler("gamesentenceminer.log", maxBytes=10_000_000, backupCount=2, encoding='utf-8')
-file_handler.setLevel(logging.DEBUG)
-
-# Create a formatter
-formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
-
-# Add formatter to handlers
-console_handler.setFormatter(formatter)
-file_handler.setFormatter(formatter)
-
-# Add handlers to the logger
-logger.addHandler(console_handler)
-logger.addHandler(file_handler)
-
-CONFIG_FILE = os.path.join(os.path.dirname(__file__), 'get_config().json')
-temp_directory = ''
-
 def get_app_directory():
     appdata_dir = os.getenv('APPDATA')  # Get the AppData directory
     config_dir = os.path.join(appdata_dir, 'GameSentenceMiner')
     os.makedirs(config_dir, exist_ok=True)  # Create the directory if it doesn't exist
     return config_dir
 
+def get_log_path():
+    return os.path.join(get_app_directory(), 'gamesentenceminer.log')
+
+temp_directory = ''
+
+def get_temporary_directory():
+    global temp_directory
+    if not temp_directory:
+        temp_directory = os.path.join(get_app_directory(), 'temp')
+        os.makedirs(temp_directory, exist_ok=True)
+        for filename in os.listdir(temp_directory):
+            file_path = os.path.join(temp_directory, filename)
+            try:
+                if os.path.isfile(file_path) or os.path.islink(file_path):
+                    os.unlink(file_path)
+                elif os.path.isdir(file_path):
+                    shutil.rmtree(file_path)
+            except Exception as e:
+                logger.error(f"Failed to delete {file_path}. Reason: {e}")
+    return temp_directory
+
 def get_config_path():
     return os.path.join(get_app_directory(), 'config.json')
 
@@ -354,6 +349,29 @@ def get_master_config():
 def switch_profile_and_save(profile_name):
     global config_instance
     config_instance.current_profile = profile_name
-    with open('config.json', 'w') as file:
+    with open(get_config_path(), 'w') as file:
         json.dump(config_instance.to_dict(), file, indent=4)
     return config_instance.get_config()
+
+
+logger = logging.getLogger("GameSentenceMiner")
+logger.setLevel(logging.DEBUG)  # Set the base level to DEBUG so that all messages are captured
+
+# Create console handler with level INFO
+console_handler = logging.StreamHandler()
+console_handler.setLevel(logging.INFO)
+
+# Create rotating file handler with level DEBUG
+file_handler = RotatingFileHandler(get_log_path(), maxBytes=10_000_000, backupCount=2, encoding='utf-8')
+file_handler.setLevel(logging.DEBUG)
+
+# Create a formatter
+formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
+
+# Add formatter to handlers
+console_handler.setFormatter(formatter)
+file_handler.setFormatter(formatter)
+
+# Add handlers to the logger
+logger.addHandler(console_handler)
+logger.addHandler(file_handler)

GameSentenceMiner/ffmpeg.py

@@ -128,7 +128,7 @@ def get_audio_and_trim(video_path, line_time, next_line_time):
         codec_command = ["-c:a", f"{supported_formats[get_config().audio.extension]}"]
         logger.info(f"Re-encoding {codec} to {get_config().audio.extension}")
 
-    untrimmed_audio = tempfile.NamedTemporaryFile(dir=configuration.temp_directory,
+    untrimmed_audio = tempfile.NamedTemporaryFile(dir=configuration.get_temporary_directory(),
                                                   suffix=f"_untrimmed.{get_config().audio.extension}").name
 
     command = ffmpeg_base_command_list + [
@@ -161,7 +161,7 @@ def get_video_duration(file_path):
 
 
 def trim_audio_based_on_last_line(untrimmed_audio, video_path, line_time, next_line):
-    trimmed_audio = tempfile.NamedTemporaryFile(dir=configuration.temp_directory,
+    trimmed_audio = tempfile.NamedTemporaryFile(dir=configuration.get_temporary_directory(),
                                                 suffix=f".{get_config().audio.extension}").name
     file_mod_time = get_file_modification_time(video_path)
     file_length = get_video_duration(video_path)

GameSentenceMiner/gsm.py

@@ -105,7 +105,7 @@ class VideoToAudioHandler(FileSystemEventHandler):
     def get_audio(line_time, next_line_time, video_path):
         trimmed_audio = get_audio_and_trim(video_path, line_time, next_line_time)
         vad_trimmed_audio = make_unique_file_name(
-            f"{os.path.abspath(configuration.temp_directory)}/{obs.get_current_game(sanitize=True)}.{get_config().audio.extension}")
+            f"{os.path.abspath(configuration.get_temporary_directory())}/{obs.get_current_game(sanitize=True)}.{get_config().audio.extension}")
         final_audio_output = make_unique_file_name(
             f"{get_config().paths.audio_destination}{obs.get_current_game(sanitize=True)}.{get_config().audio.extension}")
         should_update_audio = True
@@ -153,15 +153,6 @@ def initialize(reloading=False):
             os.mkdir(get_config().paths.screenshot_destination)
         if not os.path.exists(get_config().paths.audio_destination):
             os.mkdir(get_config().paths.audio_destination)
-        if not os.path.exists("../temp_files"):
-            os.mkdir("../temp_files")
-        else:
-            for filename in os.scandir('../temp_files'):
-                file_path = os.path.join('../temp_files', filename.name)
-                if filename.is_file() or filename.is_symlink():
-                    os.remove(file_path)
-                elif filename.is_dir():
-                    shutil.rmtree(file_path)
     if get_config().vad.do_vad_postprocessing:
         if VOSK in (get_config().vad.backup_vad_model, get_config().vad.selected_vad_model):
             vosk_helper.get_vosk_model()
@@ -349,36 +340,34 @@ def main(reloading=False, do_config_input=True):
     global settings_window
     logger.info("Script started.")
     initialize(reloading)
-    with tempfile.TemporaryDirectory(dir="../temp_files") as temp_dir:
-        configuration.temp_directory = temp_dir
-        event_handler = VideoToAudioHandler()
-        observer = Observer()
-        observer.schedule(event_handler, get_config().paths.folder_to_watch, recursive=False)
-        observer.start()
+    event_handler = VideoToAudioHandler()
+    observer = Observer()
+    observer.schedule(event_handler, get_config().paths.folder_to_watch, recursive=False)
+    observer.start()
 
-        logger.info("Script Initialized. Happy Mining!")
-        if not is_linux():
-            register_hotkeys()
+    logger.info("Script Initialized. Happy Mining!")
+    if not is_linux():
+        register_hotkeys()
 
-        # Register signal handlers for graceful shutdown
-        signal.signal(signal.SIGTERM, handle_exit())  # Handle `kill` commands
-        signal.signal(signal.SIGINT, handle_exit())  # Handle Ctrl+C
-        win32api.SetConsoleCtrlHandler(handle_exit())
+    # Register signal handlers for graceful shutdown
+    signal.signal(signal.SIGTERM, handle_exit())  # Handle `kill` commands
+    signal.signal(signal.SIGINT, handle_exit())  # Handle Ctrl+C
+    win32api.SetConsoleCtrlHandler(handle_exit())
 
-        util.run_new_thread(run_tray)
+    util.run_new_thread(run_tray)
 
-        try:
-            settings_window = config_gui.ConfigApp()
-            settings_window.add_save_hook(update_icon)
-            settings_window.window.mainloop()
-        except KeyboardInterrupt:
-            cleanup()
+    try:
+        settings_window = config_gui.ConfigApp()
+        settings_window.add_save_hook(update_icon)
+        settings_window.window.mainloop()
+    except KeyboardInterrupt:
+        cleanup()
 
-        try:
-            observer.stop()
-            observer.join()
-        except Exception as e:
-            logger.error(f"Error stopping observer: {e}")
+    try:
+        observer.stop()
+        observer.join()
+    except Exception as e:
+        logger.error(f"Error stopping observer: {e}")
 
 
 if __name__ == "__main__":

GameSentenceMiner/obs.py

@@ -101,7 +101,7 @@ def get_source_from_scene(scene_name):
 
 def get_screenshot():
     try:
-        screenshot = util.make_unique_file_name(os.path.abspath(configuration.temp_directory) + '/screenshot.png')
+        screenshot = util.make_unique_file_name(os.path.abspath(configuration.get_temporary_directory()) + '/screenshot.png')
         update_current_game()
         current_source = get_source_from_scene(get_current_game())
         current_source_name = current_source.sourceName

GameSentenceMiner/vad/silero_trim.py

@@ -12,7 +12,7 @@ vad_model = load_silero_vad()
 # Use Silero to detect voice activity with timestamps in the audio
 def detect_voice_with_silero(input_audio):
     # Convert the audio to 16kHz mono WAV
-    temp_wav = tempfile.NamedTemporaryFile(dir=configuration.temp_directory, suffix='.wav').name
+    temp_wav = tempfile.NamedTemporaryFile(dir=configuration.get_temporary_directory(), suffix='.wav').name
     ffmpeg.convert_audio_to_wav(input_audio, temp_wav)
 
     # Load the audio and detect speech timestamps

GameSentenceMiner/vad/vosk_helper.py

@@ -67,7 +67,7 @@ def download_and_cache_vosk_model(model_dir="vosk_model_cache"):
 def detect_voice_with_vosk(input_audio):
     global vosk_model_path, vosk_model
     # Convert the audio to 16kHz mono WAV
-    temp_wav = tempfile.NamedTemporaryFile(dir=configuration.temp_directory, suffix='.wav').name
+    temp_wav = tempfile.NamedTemporaryFile(dir=configuration.get_temporary_directory(), suffix='.wav').name
     ffmpeg.convert_audio_to_wav(input_audio, temp_wav)
 
     if not vosk_model_path or not vosk_model:

GameSentenceMiner/vad/whisper_helper.py

@@ -24,7 +24,7 @@ def load_whisper_model():
 # Use Whisper to detect voice activity with timestamps in the audio
 def detect_voice_with_whisper(input_audio):
     # Convert the audio to 16kHz mono WAV
-    temp_wav = tempfile.NamedTemporaryFile(dir=configuration.temp_directory, suffix='.wav').name
+    temp_wav = tempfile.NamedTemporaryFile(dir=configuration.get_temporary_directory(), suffix='.wav').name
     ffmpeg.convert_audio_to_wav(input_audio, temp_wav)
 
     # Make sure Whisper is loaded

{GameSentenceMiner-2.0.0.dist-info → GameSentenceMiner-2.0.0.dev1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: GameSentenceMiner
-Version: 2.0.0
+Version: 2.0.0.dev1
 Summary: A tool for mining sentences from games.
 Author-email: Beangate <bpwhelan95@gmail.com>
 License: MIT License

GameSentenceMiner-2.0.0.dev1.dist-info/RECORD

@@ -0,0 +1,20 @@
+GameSentenceMiner/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+GameSentenceMiner/anki.py,sha256=3UT6K5PxzJMDoiXyOULrkeCoJ0KMq7JvBw6XhcLCirE,9114
+GameSentenceMiner/config_gui.py,sha256=s03e8F2pEPj0iG31RxFqWGfm2GLb8JoGQR7QhCMFu3M,45581
+GameSentenceMiner/configuration.py,sha256=7sXYNBNXyezPFbGKTu2XwjMAy6vQ1tJPfLg2aq2UGRA,13859
+GameSentenceMiner/ffmpeg.py,sha256=hdKimzkpAKsE-17qEAQg4uHy4-TtdFywYx48Skn9cPs,10418
+GameSentenceMiner/gametext.py,sha256=GpR9P8h3GmmKH46Dw13kJPx66n3jGjFCiV8Fcrqn9E8,3999
+GameSentenceMiner/gsm.py,sha256=Lnj59KHV4l5eEgffkswBlRCONgYB_Id3L0pTajUzsvI,14937
+GameSentenceMiner/model.py,sha256=oh8VVT8T1UKekbmP6MGNgQ8jIuQ_7Rg4GPzDCn2kJo8,1999
+GameSentenceMiner/notification.py,sha256=sWgIIXhaB9WV1K_oQGf5-IR6q3dakae_QS-RuIvbcEs,1939
+GameSentenceMiner/obs.py,sha256=VH9qXRbmxWxFgcp6iGzEQUseyxf_qINCOM54ByGa7_0,3698
+GameSentenceMiner/util.py,sha256=OYg0j_rT9F7v3aJRwWnHvdWMYyxGlimrvw7U2C9ifeY,4441
+GameSentenceMiner/vad/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+GameSentenceMiner/vad/silero_trim.py,sha256=r7bZYEj-NUXGKgD2UIhLrbTPyq0rau97qGtrMZcRK4A,1517
+GameSentenceMiner/vad/vosk_helper.py,sha256=lWmlGMhmg_0QoWeCHrXwz9wDKPqY37BckHCekGVtJUI,5794
+GameSentenceMiner/vad/whisper_helper.py,sha256=9kmPeSs6jRcKSVxYY-vtyTcqVUDORR4q1nl8fqxHHn4,3379
+GameSentenceMiner-2.0.0.dev1.dist-info/METADATA,sha256=32ajZFt816JeijN0hktOxA656mnWWT_5rSnSUK16PjI,13394
+GameSentenceMiner-2.0.0.dev1.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+GameSentenceMiner-2.0.0.dev1.dist-info/entry_points.txt,sha256=2APEP25DbfjSxGeHtwBstMH8mulVhLkqF_b9bqzU6vQ,65
+GameSentenceMiner-2.0.0.dev1.dist-info/top_level.txt,sha256=V1hUY6xVSyUEohb0uDoN4UIE6rUZ_JYx8yMyPGX4PgQ,18
+GameSentenceMiner-2.0.0.dev1.dist-info/RECORD,,

GameSentenceMiner-2.0.0.dist-info/RECORD

@@ -1,20 +0,0 @@
-GameSentenceMiner/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-GameSentenceMiner/anki.py,sha256=3UT6K5PxzJMDoiXyOULrkeCoJ0KMq7JvBw6XhcLCirE,9114
-GameSentenceMiner/config_gui.py,sha256=d1UDXbTNSZBFneWEytqdnySBL-_iJ1MGY0Jvyg3vn80,45691
-GameSentenceMiner/configuration.py,sha256=ilTnUXFIu9G7folA4AT78M57Wq9-j74gi5MbYWDtDRw,13165
-GameSentenceMiner/ffmpeg.py,sha256=XreJHjmCpmdI04g_UmTzXmh5pay1s5DLx41nrAJmQwo,10396
-GameSentenceMiner/gametext.py,sha256=GpR9P8h3GmmKH46Dw13kJPx66n3jGjFCiV8Fcrqn9E8,3999
-GameSentenceMiner/gsm.py,sha256=4kOb3eIiOZLpRv-ZhczoJ_ZtqMMxnxK0XW9gLWkdK8I,15557
-GameSentenceMiner/model.py,sha256=oh8VVT8T1UKekbmP6MGNgQ8jIuQ_7Rg4GPzDCn2kJo8,1999
-GameSentenceMiner/notification.py,sha256=sWgIIXhaB9WV1K_oQGf5-IR6q3dakae_QS-RuIvbcEs,1939
-GameSentenceMiner/obs.py,sha256=-hGz3D9roCDVecZ9IhIofPppcmTEAYO8cZf-lnkBBxU,3687
-GameSentenceMiner/util.py,sha256=OYg0j_rT9F7v3aJRwWnHvdWMYyxGlimrvw7U2C9ifeY,4441
-GameSentenceMiner/vad/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-GameSentenceMiner/vad/silero_trim.py,sha256=x1jTQF9x55Q2HF32xAE4QrLoa30U3ksOmY9XF_qulrs,1506
-GameSentenceMiner/vad/vosk_helper.py,sha256=6vPkfaj2TJtS0Ph3aUByTx98rb-InKcnhp7MZh-vY-c,5783
-GameSentenceMiner/vad/whisper_helper.py,sha256=sfGTXU3dt5ZAXMaZbBp3j2cvYc0QF-HXD72_Lgl1NTs,3368
-GameSentenceMiner-2.0.0.dist-info/METADATA,sha256=m4DB-eNUyyaygdHyAmO7gZVyWcXQijPtxlPc6AVPKzI,13389
-GameSentenceMiner-2.0.0.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-GameSentenceMiner-2.0.0.dist-info/entry_points.txt,sha256=2APEP25DbfjSxGeHtwBstMH8mulVhLkqF_b9bqzU6vQ,65
-GameSentenceMiner-2.0.0.dist-info/top_level.txt,sha256=V1hUY6xVSyUEohb0uDoN4UIE6rUZ_JYx8yMyPGX4PgQ,18
-GameSentenceMiner-2.0.0.dist-info/RECORD,,