From a1650eaa4ffcd6eb523df8d2cf96137470b9aa9e Mon Sep 17 00:00:00 2001 From: makaveli10 Date: Tue, 19 Nov 2024 13:31:55 +0530 Subject: [PATCH] Fix client tests to write srt file Signed-off-by: makaveli10 --- whisper_live/client.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/whisper_live/client.py b/whisper_live/client.py index 15b6306..a748bb6 100644 --- a/whisper_live/client.py +++ b/whisper_live/client.py @@ -259,7 +259,9 @@ class Client: """ if self.server_backend == "faster_whisper": - if (self.last_segment) and self.transcript[-1]["text"] != self.last_segment["text"]: + if not self.transcript and self.last_segment is not None: + self.transcript.append(self.last_segment) + elif self.last_segment and self.transcript[-1]["text"] != self.last_segment["text"]: self.transcript.append(self.last_segment) utils.create_srt_file(self.transcript, output_path)