dora-vad 0.3.9__py3-none-any.whl → 0.3.10__py3-none-any.whl

dora_vad/main.py

@@ -7,11 +7,11 @@ from dora import Node
 from silero_vad import get_speech_timestamps, load_silero_vad
 
 model = load_silero_vad()
-MIN_SILENCE_DURATION_MS = int(os.getenv("MIN_SILENCE_DURATION_MS", "100"))
+MIN_SILENCE_DURATION_MS = int(os.getenv("MIN_SILENCE_DURATION_MS", "200"))
 MIN_SPEECH_DURATION_MS = int(os.getenv("MIN_SPEECH_DURATION_MS", "300"))
-
-MIN_AUDIO_SAMPLING_DURAION_S = int(os.getenv("MAX_AUDIO_DURATION_S", "20"))
-MAX_AUDIO_DURAION_S = int(os.getenv("MAX_AUDIO_DURATION_S", "75"))
+THRESHOLD = float(os.getenv("THRESHOLD", "0.4"))
+MAX_AUDIO_DURATION_S = float(os.getenv("MAX_AUDIO_DURATION_S", "75"))
+MIN_AUDIO_SAMPLING_DURATION_MS = int(os.getenv("MIN_AUDIO_SAMPLING_DURATION_MS", "500"))
 
 
 def main():
@@ -26,10 +26,11 @@ def main():
             last_audios += [audio]
             last_audios = last_audios[-100:]
             audio = np.concatenate(last_audios)
+            sr = event["metadata"].get("sample_rate", 16000)
             speech_timestamps = get_speech_timestamps(
                 torch.from_numpy(audio),
                 model,
-                threshold=0.2,
+                threshold=THRESHOLD,
                 min_speech_duration_ms=MIN_SPEECH_DURATION_MS,
                 min_silence_duration_ms=MIN_SILENCE_DURATION_MS,
             )
@@ -37,16 +38,20 @@ def main():
             # Check ig there is timestamp
             if (
                 len(speech_timestamps) > 0
-                and len(last_audios) > MIN_AUDIO_SAMPLING_DURAION_S
+                and len(audio) > MIN_AUDIO_SAMPLING_DURATION_MS * sr / 1000
             ):
                 # Check if the audio is not cut at the end. And only return if there is a long time spent
                 if speech_timestamps[-1]["end"] == len(audio):
+                    node.send_output(
+                        "timestamp_start",
+                        pa.array([speech_timestamps[-1]["start"]]),
+                    )
                     continue
                 audio = audio[0 : speech_timestamps[-1]["end"]]
                 node.send_output("audio", pa.array(audio))
                 last_audios = [audio[speech_timestamps[-1]["end"] :]]
 
             # If there is no sound for too long return the audio
-            elif len(last_audios) > 75:
+            elif len(last_audios) > MAX_AUDIO_DURATION_S:
                 node.send_output("audio", pa.array(audio))
                 last_audios = []

{dora_vad-0.3.9.dist-info → dora_vad-0.3.10.dist-info}/METADATA

@@ -1,12 +1,12 @@
 Metadata-Version: 2.2
 Name: dora-vad
-Version: 0.3.9
+Version: 0.3.10
 Summary: Dora Node for Text translating using Argostranslate
 Author-email: Haixuan Xavier Tao <tao.xavier@outlook.com>
 License: MIT
 Requires-Python: >=3.8
 Description-Content-Type: text/markdown
-Requires-Dist: dora-rs>=0.3.6
+Requires-Dist: dora-rs>=0.3.9
 Requires-Dist: numpy<2.0.0
 Requires-Dist: silero-vad>=5.1
 

dora_vad-0.3.10.dist-info/RECORD

@@ -0,0 +1,7 @@
+dora_vad/__init__.py,sha256=HuSK3dnyI9Pb5QAuaKFwQQ3J5SIZnLcKHPJO0norGzc,353
+dora_vad/main.py,sha256=z1-nS4_tnhTasV75Bsc6s2Der_ISbpjafmNJLYrsvws,2241
+dora_vad-0.3.10.dist-info/METADATA,sha256=izl0Gjyg6Z1xQ2-N-b7jDn0LaZSR15aghlv-nwF-VE8,1461
+dora_vad-0.3.10.dist-info/WHEEL,sha256=jB7zZ3N9hIM9adW7qlTAyycLYW9npaWKLRzaoVcLKcM,91
+dora_vad-0.3.10.dist-info/entry_points.txt,sha256=X82-C0mpNH9bVjwn24mzfoWX5ssrK6W1Ylk1HPDR0ME,48
+dora_vad-0.3.10.dist-info/top_level.txt,sha256=1EUgMajnck6tHR4mFagv_uv-t2A6Cabd2wYtEldKm2Q,9
+dora_vad-0.3.10.dist-info/RECORD,,

{dora_vad-0.3.9.dist-info → dora_vad-0.3.10.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (75.8.0)
+Generator: setuptools (75.8.2)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

dora_vad-0.3.9.dist-info/RECORD

@@ -1,7 +0,0 @@
-dora_vad/__init__.py,sha256=HuSK3dnyI9Pb5QAuaKFwQQ3J5SIZnLcKHPJO0norGzc,353
-dora_vad/main.py,sha256=HfdT87XdtlZfoNRfKbWKl-ylVNFoq9t8acfrzB6n3wE,1913
-dora_vad-0.3.9.dist-info/METADATA,sha256=B4iuvHVM6YfqEOqkyw-dZ6_kBJU1fjGU--2Ds6Bfq0g,1460
-dora_vad-0.3.9.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-dora_vad-0.3.9.dist-info/entry_points.txt,sha256=X82-C0mpNH9bVjwn24mzfoWX5ssrK6W1Ylk1HPDR0ME,48
-dora_vad-0.3.9.dist-info/top_level.txt,sha256=1EUgMajnck6tHR4mFagv_uv-t2A6Cabd2wYtEldKm2Q,9
-dora_vad-0.3.9.dist-info/RECORD,,