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

Expose the srt file location of Transcription client

Browse Source
This commit is contained in:
chien-liu
2024-06-01 00:14:59 +02:00
committed by chien-liu
parent e1a42c22d2
commit cfba5b3e54
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
whisper_live/client.py
+6 -2
View File
@@ -642,6 +642,7 @@ class TranscriptionClient(TranscriptionTeeClient):
translate (bool, optional): Indicates whether translation tasks are required (default is False). translate (bool, optional): Indicates whether translation tasks are required (default is False).
save_output_recording (bool, optional): Indicates whether to save recording from microphone. save_output_recording (bool, optional): Indicates whether to save recording from microphone.
output_recording_filename (str, optional): File to save the output recording. output_recording_filename (str, optional): File to save the output recording.
output_transcription_path (str, optional): File to save the output transcription.
Attributes: Attributes:
client (Client): An instance of the underlying Client class responsible for handling the WebSocket connection. client (Client): An instance of the underlying Client class responsible for handling the WebSocket connection.
@@ -662,11 +663,14 @@ class TranscriptionClient(TranscriptionTeeClient):
model="small", model="small",
use_vad=True, use_vad=True,
save_output_recording=False, save_output_recording=False,
output_recording_filename="./output_recording.wav" output_recording_filename="./output_recording.wav",
output_transcription_path="./output.srt"
): ):
self.client = Client(host, port, lang, translate, model, srt_file_path="output.srt", use_vad=use_vad) self.client = Client(host, port, lang, translate, model, srt_file_path=output_transcription_path, use_vad=use_vad)
if save_output_recording and not output_recording_filename.endswith(".wav"): if save_output_recording and not output_recording_filename.endswith(".wav"):
raise ValueError(f"Please provide a valid `output_recording_filename`: {output_recording_filename}") raise ValueError(f"Please provide a valid `output_recording_filename`: {output_recording_filename}")
if not output_transcription_path.endswith(".srt"):
raise ValueError(f"Please provide a valid `output_transcription_path`: {output_transcription_path}. The file extension should be `.srt`.")
TranscriptionTeeClient.__init__( TranscriptionTeeClient.__init__(
self, self,
[self.client], [self.client],