Files
VRCT/test_main.py
misyaguziya 55e7fab281 rename files
2023-06-30 22:02:48 +09:00

40 lines
1.4 KiB
Python

import time
import threading
import queue
import audio_transcriber
import audio_recorder
import audio_utils
try:
mic_audio_queue = queue.Queue()
mic_device = audio_utils.get_default_input_device()
mic_audio_recorder = audio_recorder.SelectedMicRecorder(mic_device)
mic_audio_recorder.record_into_queue(mic_audio_queue)
mic_transcriber = audio_transcriber.AudioTranscriber(speaker=False, source=mic_audio_recorder.source, language="ja-JP")
mic_transcribe = threading.Thread(target=mic_transcriber.transcribe_audio_queue, args=(mic_audio_queue,))
mic_transcribe.daemon = True
mic_transcribe.start()
time.sleep(2)
spk_audio_queue = queue.Queue()
spk_device = audio_utils.get_default_output_device()
spk_audio_recorder = audio_recorder.SelectedSpeakerRecorder(spk_device)
spk_audio_recorder.record_into_queue(spk_audio_queue)
spk_transcriber = audio_transcriber.AudioTranscriber(speaker=True, source=spk_audio_recorder.source, language="ja-JP")
spk_transcribe = threading.Thread(target=spk_transcriber.transcribe_audio_queue, args=(spk_audio_queue,))
spk_transcribe.daemon = True
spk_transcribe.start()
while True:
text = mic_transcriber.get_transcript()
if len(text) > 0:
print("mic:", text)
text = spk_transcriber.get_transcript()
if len(text) > 0:
print("spk:", text)
time.sleep(0.1)
except Exception as e:
print(e)