From 26d9b46888ac5321cb001f138a072913a7c05310 Mon Sep 17 00:00:00 2001
From: Barabazs <31799121+Barabazs@users.noreply.github.com>
Date: Sun, 5 Jan 2025 18:16:46 +0100
Subject: [PATCH] feat: include speaker information in  WriteTXT when diarizing

---
 whisperx/utils.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/whisperx/utils.py b/whisperx/utils.py
index 16ce116..0b440b7 100644
--- 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):