feat: include speaker information in WriteTXT when diarizing

m-bain · Jan 5, 2025 · 26d9b46 · 26d9b46
1 parent 9a8967f
commit 26d9b46
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/whisperx/utils.py b/whisperx/utils.py
@@ -214,7 +214,12 @@ class WriteTXT(ResultWriter):
 
     def write_result(self, result: dict, file: TextIO, options: dict):
         for segment in result["segments"]:
-            print(segment["text"].strip(), file=file, flush=True)
+            speaker = segment.get("speaker")
+            text = segment["text"].strip()
+            if speaker is not None:
+                print(f"[{speaker}]: {text}", file=file, flush=True)
+            else:
+                print(text, file=file, flush=True)
 
 
 class SubtitlesWriter(ResultWriter):