35 lines
961 B
Python
35 lines
961 B
Python
from modules.whisper_live.client import TranscriptionClient
|
|
import os
|
|
import time
|
|
|
|
def setup_directories(user_dir, user_id):
|
|
user_transcript_dir = f"{user_dir}/{user_id}/transcripts"
|
|
if not os.path.exists(user_transcript_dir):
|
|
os.makedirs(user_transcript_dir)
|
|
return user_transcript_dir
|
|
|
|
def timestamped_callback(text, is_final):
|
|
if is_final:
|
|
print(f"Timestamp: {time.strftime('%H:%M:%S')}, Transcription: {text}")
|
|
|
|
def main():
|
|
user_dir = "../../data/users"
|
|
user_id = "kcar"
|
|
user_transcript_dir = setup_directories(user_dir, user_id)
|
|
|
|
client = TranscriptionClient(
|
|
"localhost",
|
|
9090,
|
|
lang="en",
|
|
translate=False,
|
|
use_vad=True,
|
|
save_output_recording=True,
|
|
output_recording_filename=f"{user_transcript_dir}/output_recording.wav",
|
|
output_transcription_path=f"{user_transcript_dir}/output.srt",
|
|
)
|
|
|
|
client()
|
|
|
|
if __name__ == "__main__":
|
|
main()
|