liuhuayong/ollama-voice
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Small changes

Browse Source
This commit is contained in:
M 2023-11-12 02:49:25 +01:00
parent 96c3b0feb9
commit b2f53b9075
1 changed files with 14 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
assistant.py
View File

@ -12,10 +12,11 @@ from yaml import Loader
if sys.version_info[0:3] != (3, 9, 13): if sys.version_info[0:3] != (3, 9, 13):
print('Warning, it was only tested with python 3.9.13, it may fail') print('Warning, it was only tested with python 3.9.13, it may fail')
FORMAT = pyaudio.paInt16 INPUT_DEFAULT_DURATION_SECONDS = 5
CHANNELS = 1 INPUT_FORMAT = pyaudio.paInt16
RATE = 16000 INPUT_CHANNELS = 1
CHUNK = 1024 INPUT_RATE = 16000
INPUT_CHUNK = 1024
OLLAMA_REST_HEADERS = {'Content-Type': 'application/json',} OLLAMA_REST_HEADERS = {'Content-Type': 'application/json',}
INPUT_CONFIG_PATH ="assistant.yaml" INPUT_CONFIG_PATH ="assistant.yaml"
@ -24,11 +25,11 @@ class Assistant:
def __init__(self): def __init__(self):
self.config = self.initConfig() self.config = self.initConfig()
self.audio = pyaudio.PyAudio()
print("Loading Whisper model...") print("Loading Whisper model...")
self.model = whisper.load_model(self.config.whisperRecognition.modelPath) self.model = whisper.load_model(self.config.whisperRecognition.modelPath)
self.tts = pyttsx3.init() self.tts = pyttsx3.init()
self.audio = pyaudio.PyAudio()
self.conversation_history = [self.config.conversation.context+self.config.conversation.greeting+"\n"] self.conversation_history = [self.config.conversation.context+self.config.conversation.greeting+"\n"]
@ -60,15 +61,15 @@ class Assistant:
return config return config
def waveform_from_mic(self, duration=5) -> np.ndarray: def waveform_from_mic(self, duration=INPUT_DEFAULT_DURATION_SECONDS) -> np.ndarray:
stream = self.audio.open(format=FORMAT, channels=CHANNELS, stream = self.audio.open(format=INPUT_FORMAT, channels=INPUT_CHANNELS,
rate=RATE, input=True, rate=INPUT_RATE, input=True,
frames_per_buffer=CHUNK) frames_per_buffer=INPUT_CHUNK)
frames = [] frames = []
for _ in range(0, int(RATE / CHUNK * duration)): for _ in range(0, int(INPUT_RATE / INPUT_CHUNK * duration)):
data = stream.read(CHUNK) data = stream.read(INPUT_CHUNK)
frames.append(data) frames.append(data)
stream.stop_stream() stream.stop_stream()