GameSentenceMiner 2.9.25__py3-none-any.whl → 2.9.26__py3-none-any.whl

GameSentenceMiner/vad.py

@@ -93,15 +93,26 @@ class VADProcessor(ABC):
         return float(result.stdout.strip())
 
     @staticmethod
-    def extract_audio_and_combine_segments(input_audio, segments, output_audio, padding=0.2):
+    def extract_audio_and_combine_segments(input_audio, segments, output_audio, padding=0.1):
         files = []
         ffmpeg_threads = []
         logger.info(f"Extracting {len(segments)} segments from {input_audio} with padding {padding} seconds.")
-        for segment in segments:
+
+        current_start = None
+        for i in range(len(segments)):
+            segment = segments[i]
             logger.info(segment)
-            temp_file = make_unique_file_name(os.path.join(get_temporary_directory(), "segment." + get_config().audio.extension))
+            if i < len(segments) - 1 and (segments[i + 1]['start'] - segment['end']) < padding * 2:
+                logger.info(f"Adjusting segment {segments[i + 1]} due to insufficient padding.")
+                current_start = segment['start'] if current_start is None else current_start
+                continue
+            temp_file = make_unique_file_name(
+                os.path.join(get_temporary_directory(), "segment." + get_config().audio.extension))
             files.append(temp_file)
-            ffmpeg_threads.append(run_new_thread(lambda: ffmpeg.trim_audio(input_audio, segment['start'] - padding, segment['end'] + padding, temp_file, trim_beginning=True)))
+            ffmpeg_threads.append(run_new_thread(
+                lambda: ffmpeg.trim_audio(input_audio, (current_start if current_start else segment['start']) - padding, segment['end'] + padding, temp_file,
+                                          trim_beginning=True)))
+            current_start = None
             time.sleep(0.1)  # Small delay to ensure unique file names
 
         for thread in ffmpeg_threads:
@@ -109,8 +120,8 @@ class VADProcessor(ABC):
 
         if len(files) > 1:
             ffmpeg.combine_audio_files(files, output_audio)
-            # for file in files:
-            #     os.remove(file)
+            for file in files:
+                os.remove(file)
         else:
             shutil.move(files[0], output_audio)
 
@@ -361,7 +372,7 @@ class GroqVADProcessor(VADProcessor):
 vad_processor = VADSystem()
 
 # test_vad = SileroVADProcessor()
-#
+# #
 # if os.path.exists(r"C:\Users\Beangate\GSM\Electron App\test\after_splice.opus"):
 #     os.remove(r"C:\Users\Beangate\GSM\Electron App\test\after_splice.opus")
 # test_vad.process_audio(r"C:\Users\Beangate\GSM\Electron App\test\before_splice.opus", r"C:\Users\Beangate\GSM\Electron App\test\after_splice.opus", None)

{gamesentenceminer-2.9.25.dist-info → gamesentenceminer-2.9.26.dist-info}/METADATA

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

{gamesentenceminer-2.9.25.dist-info → gamesentenceminer-2.9.26.dist-info}/RECORD

@@ -4,7 +4,7 @@ GameSentenceMiner/config_gui.py,sha256=xQXFrDfsjbWxuX0PwGn2jjv0mfRZEZ_MdUU3iMASi
 GameSentenceMiner/gametext.py,sha256=nAwGMQSrmc6sUAw-OAURK2n6MG5Ecm6psJ7YF9q5KTA,6623
 GameSentenceMiner/gsm.py,sha256=oM4l0UP6_h66AVEZqdIShs264A3LEgmSDTlp-gp048o,24167
 GameSentenceMiner/obs.py,sha256=YG8LwBf9BTsGbROm_Uq6LhFDSrbf3jgogp78rBbJq94,14728
-GameSentenceMiner/vad.py,sha256=kWJ6c3v0iJInIiwU9ANYQvweqGi3PpeN_eegJLybfIM,16148
+GameSentenceMiner/vad.py,sha256=PATjcvpxkieaujDtqHBaVyRMkStWbphqmkfBPJeDuOE,16678
 GameSentenceMiner/ai/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 GameSentenceMiner/ai/ai_prompting.py,sha256=0jBAnngNwmc3dqJiVWe_QRy4Syr-muV-ML2rq0FiUtU,10215
 GameSentenceMiner/assets/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -61,9 +61,9 @@ GameSentenceMiner/web/templates/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm
 GameSentenceMiner/web/templates/index.html,sha256=HZKiIjiGJV8PGQ9T2aLDUNSfJn71qOwbYCjbRuSIjpY,213583
 GameSentenceMiner/web/templates/text_replacements.html,sha256=tV5c8mCaWSt_vKuUpbdbLAzXZ3ATZeDvQ9PnnAfqY0M,8598
 GameSentenceMiner/web/templates/utility.html,sha256=3flZinKNqUJ7pvrZk6xu__v67z44rXnaK7UTZ303R-8,16946
-gamesentenceminer-2.9.25.dist-info/licenses/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-gamesentenceminer-2.9.25.dist-info/METADATA,sha256=fH0PSv9qJAA6wBq72A3c1xEhB0Y2jE6fkWyen-i6M1Y,7276
-gamesentenceminer-2.9.25.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-gamesentenceminer-2.9.25.dist-info/entry_points.txt,sha256=2APEP25DbfjSxGeHtwBstMH8mulVhLkqF_b9bqzU6vQ,65
-gamesentenceminer-2.9.25.dist-info/top_level.txt,sha256=V1hUY6xVSyUEohb0uDoN4UIE6rUZ_JYx8yMyPGX4PgQ,18
-gamesentenceminer-2.9.25.dist-info/RECORD,,
+gamesentenceminer-2.9.26.dist-info/licenses/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+gamesentenceminer-2.9.26.dist-info/METADATA,sha256=zzFPI8lHQRvkEfTKNSaWRC5EGiGLxzk1jsE1KzgvZX8,7276
+gamesentenceminer-2.9.26.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+gamesentenceminer-2.9.26.dist-info/entry_points.txt,sha256=2APEP25DbfjSxGeHtwBstMH8mulVhLkqF_b9bqzU6vQ,65
+gamesentenceminer-2.9.26.dist-info/top_level.txt,sha256=V1hUY6xVSyUEohb0uDoN4UIE6rUZ_JYx8yMyPGX4PgQ,18
+gamesentenceminer-2.9.26.dist-info/RECORD,,