mirror of
https://github.com/Priler/jarvis.git
synced 2026-06-06 20:39:46 +00:00
50 lines
1.3 KiB
Python
50 lines
1.3 KiB
Python
import vosk
|
|
import queue
|
|
import json
|
|
import sys
|
|
import sounddevice as sd
|
|
from rich import print
|
|
from utils.benchmark import Benchmark
|
|
|
|
# import speech_recognition as sr
|
|
# for index, name in enumerate(sr.Microphone.list_microphone_names()):
|
|
# print("Microphone with name \"{1}\" found for `Microphone(device_index={0})`".format(index, name))
|
|
# exit(1)
|
|
|
|
bench = Benchmark()
|
|
|
|
|
|
def va_respond(voice: str):
|
|
print(f"[light_sea_green]Vosk thinks you said[/]: [dodger_blue1]{voice}[/]")
|
|
|
|
|
|
model = vosk.Model("model_small")
|
|
samplerate = 16000
|
|
device = 2
|
|
|
|
q = queue.Queue()
|
|
|
|
|
|
def q_callback(indata, frames, time, status):
|
|
if status:
|
|
print(status, file=sys.stderr)
|
|
q.put(bytes(indata))
|
|
|
|
|
|
def va_listen(callback):
|
|
with sd.RawInputStream(samplerate=samplerate, blocksize=8000, device=device, dtype='int16',
|
|
channels=1, callback=q_callback):
|
|
|
|
rec = vosk.KaldiRecognizer(model, samplerate)
|
|
while True:
|
|
bench.start()
|
|
data = q.get()
|
|
if rec.AcceptWaveform(data):
|
|
callback(json.loads(rec.Result())["text"])
|
|
end_time = bench.end()
|
|
print(f"[grey37]This took[/] [red]{end_time[1]}[/]")
|
|
#else:
|
|
# print(rec.PartialResult())
|
|
|
|
# начать прослушивание команд
|
|
va_listen(va_respond) |