Update modules/vad/silero_vad.py

modules/vad/silero_vad.py

@@ -215,13 +215,6 @@ class SileroVAD:
 
         return np.concatenate([audio[chunk["start"]: chunk["end"]] for chunk in chunks])
 
-    def get_chunk_index(self, time: float) -> int:
-        sample = int(time * self.sampling_rate)
-        return min(
-            bisect.bisect(self.chunk_end_sample, sample),
-            len(self.chunk_end_sample) - 1,
-        )
-
     @staticmethod
     def format_timestamp(
         seconds: float,
@@ -260,5 +253,24 @@ class SileroVAD:
             segment["start"] = ts_map.get_original_time(segment["start"])
             segment["end"] = ts_map.get_original_time(segment["end"])
 
+        for segment in segments:
+            if segment.words:
+                words = []
+                for word in segment.words:
+                    # Ensure the word start and end times are resolved to the same chunk.
+                    middle = (word.start + word.end) / 2
+                    chunk_index = ts_map.get_chunk_index(middle)
+                    word.start = ts_map.get_original_time(word.start, chunk_index)
+                    word.end = ts_map.get_original_time(word.end, chunk_index)
+                    words.append(word)
+
+                segment["start"] = words[0].start
+                segment["end"] = words[-1].end
+                segment["words"] = words
+
+            else:
+                segment["start"] = ts_map.get_original_time(segment["start"])
+                segment["end"] = ts_map.get_original_time(segment["end"])
+        
         return segments