cjones/WhisperLive
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Send completed bool with each segment

Browse Source 
Completed bool represents if the segment is completely processed by the server

Signed-off-by: makaveli10 <suryanvineet47@gmail.com>
This commit is contained in:
makaveli10
2024-11-05 18:11:32 +05:30
parent 00f0ff1112
commit 81c57ae40c
2 changed files with 11 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
whisper_live/client.py
+3 -3
View File
@@ -112,9 +112,9 @@ class Client:
for i, seg in enumerate(segments):
if not text or text[-1] != seg["text"]:
text.append(seg["text"])
if i == len(segments) - 1:
if i == len(segments) - 1 and not seg["completed"]:
self.last_segment = seg
elif (self.server_backend == "faster_whisper" and
elif (self.server_backend == "faster_whisper" and seg["completed"] and
(not self.transcript or
float(seg['start']) >= float(self.transcript[-1]['end']))):
self.transcript.append(seg)
@@ -259,7 +259,7 @@ class Client:
"""
if self.server_backend == "faster_whisper":
if (self.last_segment):
if (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)