latest

run/tests/_transcribe.py

@@ -0,0 +1,34 @@
+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()