saving the virtual assistant
parent
8c7309cb61
commit
3dce8528db
Binary file not shown.
@ -0,0 +1,2 @@
|
|||||||
|
black
|
||||||
|
pip-tools
|
@ -0,0 +1,41 @@
|
|||||||
|
#
|
||||||
|
# This file is autogenerated by pip-compile with python 3.8
|
||||||
|
# To update, run:
|
||||||
|
#
|
||||||
|
# pip-compile dev_requirements.in
|
||||||
|
#
|
||||||
|
black==22.10.0
|
||||||
|
# via -r dev_requirements.in
|
||||||
|
build==0.9.0
|
||||||
|
# via pip-tools
|
||||||
|
click==8.1.3
|
||||||
|
# via
|
||||||
|
# black
|
||||||
|
# pip-tools
|
||||||
|
mypy-extensions==0.4.3
|
||||||
|
# via black
|
||||||
|
packaging==21.3
|
||||||
|
# via build
|
||||||
|
pathspec==0.10.2
|
||||||
|
# via black
|
||||||
|
pep517==0.13.0
|
||||||
|
# via build
|
||||||
|
pip-tools==6.10.0
|
||||||
|
# via -r dev_requirements.in
|
||||||
|
platformdirs==2.5.4
|
||||||
|
# via black
|
||||||
|
pyparsing==3.0.9
|
||||||
|
# via packaging
|
||||||
|
tomli==2.0.1
|
||||||
|
# via
|
||||||
|
# black
|
||||||
|
# build
|
||||||
|
# pep517
|
||||||
|
typing-extensions==4.4.0
|
||||||
|
# via black
|
||||||
|
wheel==0.38.4
|
||||||
|
# via pip-tools
|
||||||
|
|
||||||
|
# The following packages are considered to be unsafe in a requirements file:
|
||||||
|
# pip
|
||||||
|
# setuptools
|
@ -1,6 +1,6 @@
|
|||||||
import simpleaudio as sa
|
import simpleaudio as sa
|
||||||
|
|
||||||
filename = 'myfile.wav'
|
filename = './data/hello.wav'
|
||||||
wave_obj = sa.WaveObject.from_wave_file(filename)
|
wave_obj = sa.WaveObject.from_wave_file(filename)
|
||||||
play_obj = wave_obj.play()
|
play_obj = wave_obj.play()
|
||||||
play_obj.wait_done() # Wait until sound has finished playing
|
play_obj.wait_done() # Wait until sound has finished playing
|
@ -0,0 +1,3 @@
|
|||||||
|
pyaudio
|
||||||
|
SpeechRecognition
|
||||||
|
gTTs
|
@ -0,0 +1,26 @@
|
|||||||
|
#
|
||||||
|
# This file is autogenerated by pip-compile with python 3.8
|
||||||
|
# To update, run:
|
||||||
|
#
|
||||||
|
# pip-compile requirement.in
|
||||||
|
#
|
||||||
|
certifi==2022.9.24
|
||||||
|
# via requests
|
||||||
|
charset-normalizer==2.1.1
|
||||||
|
# via requests
|
||||||
|
click==8.1.3
|
||||||
|
# via gtts
|
||||||
|
gtts==2.2.4
|
||||||
|
# via -r requirement.in
|
||||||
|
idna==3.4
|
||||||
|
# via requests
|
||||||
|
pyaudio==0.2.12
|
||||||
|
# via -r requirement.in
|
||||||
|
requests==2.28.1
|
||||||
|
# via gtts
|
||||||
|
six==1.16.0
|
||||||
|
# via gtts
|
||||||
|
speechrecognition==3.8.1
|
||||||
|
# via -r requirement.in
|
||||||
|
urllib3==1.26.12
|
||||||
|
# via requests
|
@ -0,0 +1,102 @@
|
|||||||
|
import webbrowser
|
||||||
|
import typing
|
||||||
|
import os
|
||||||
|
import sys
|
||||||
|
from abc import ABC, abstractmethod
|
||||||
|
|
||||||
|
import speech_recognition as sr
|
||||||
|
|
||||||
|
class Action(ABC):
|
||||||
|
|
||||||
|
action_key: str
|
||||||
|
|
||||||
|
@abstractmethod
|
||||||
|
def action_callback(self, action_text) -> typing.Any:
|
||||||
|
pass
|
||||||
|
|
||||||
|
def register_va(self, virtual_assisant):
|
||||||
|
self._va = virtual_assisant
|
||||||
|
|
||||||
|
def sans_action_key(self, action_text):
|
||||||
|
return action_text[len(self.action_key)+1:]
|
||||||
|
|
||||||
|
class BrowserAction(Action):
|
||||||
|
|
||||||
|
action_key = "open browser"
|
||||||
|
|
||||||
|
def action_callback(self, action_text):
|
||||||
|
domain = self.sans_action_key(action_text)
|
||||||
|
url = f"https://{domain}"
|
||||||
|
print(f"Opening: {url}")
|
||||||
|
self._va.("Opening {domain}")
|
||||||
|
webbrowser.open(url)
|
||||||
|
|
||||||
|
class EchoAction(Action):
|
||||||
|
|
||||||
|
action_key = "echo"
|
||||||
|
|
||||||
|
def action_callback(self, action_text) -> typing.Any:
|
||||||
|
print(action_text)
|
||||||
|
self._va.text_to_speech(self.sans_action_key(action_text))
|
||||||
|
|
||||||
|
class StopAction(Action):
|
||||||
|
|
||||||
|
action_key = "stop listening"
|
||||||
|
|
||||||
|
def action_callback(self, action_text) -> typing.Any:
|
||||||
|
print("Stopping Virtual Assistant...")
|
||||||
|
sys.exit(0)
|
||||||
|
|
||||||
|
|
||||||
|
class VirtualAssistant:
|
||||||
|
|
||||||
|
def __init__(self, timeout: int = 3, actions: typing.List[Action]=[]):
|
||||||
|
self.listener = sr.Recognizer()
|
||||||
|
self.timeout = timeout
|
||||||
|
self.action_register = []
|
||||||
|
for action in actions:
|
||||||
|
self.register_action(action)
|
||||||
|
|
||||||
|
def register_action(self, action: Action):
|
||||||
|
"""Adds new Action instances to the Virtual Assistant's action register."""
|
||||||
|
self.action_register.append(action)
|
||||||
|
action.register_va(self)
|
||||||
|
|
||||||
|
def parse_action(self, text: str):
|
||||||
|
"""Checks the VA command register and executes callbacks when found."""
|
||||||
|
for action in self.action_register:
|
||||||
|
if text.lower().startswith(action.action_key):
|
||||||
|
action.action_callback(text)
|
||||||
|
return
|
||||||
|
print(f"No actions taken on: {text}")
|
||||||
|
|
||||||
|
def speech_to_text(self) -> str:
|
||||||
|
""""""
|
||||||
|
try:
|
||||||
|
with sr.Microphone() as mic:
|
||||||
|
print("Arni is listening...")
|
||||||
|
self.listener.adjust_for_ambient_noise(mic)
|
||||||
|
utterance = self.listener.listen(mic, timeout=self.timeout)
|
||||||
|
content = self.listener.recognize_google(utterance)
|
||||||
|
return content
|
||||||
|
except (sr.UnknownValueError, sr.RequestError, sr.WaitTimeoutError):
|
||||||
|
return ""
|
||||||
|
|
||||||
|
def text_to_speech(self, text: str):
|
||||||
|
print(f"Saying: {text}")
|
||||||
|
cmd = f'gtts-cli --nocheck "{text}" | mpg123 -q -'
|
||||||
|
print(cmd)
|
||||||
|
os.system(cmd)
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
def main():
|
||||||
|
print("Creating virtual assistant...")
|
||||||
|
va = VirtualAssistant(actions=[BrowserAction(),EchoAction(), StopAction()])
|
||||||
|
# breakpoint()
|
||||||
|
while True:
|
||||||
|
content = va.speech_to_text()
|
||||||
|
va.parse_action(content)
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
main()
|
Loading…
Reference in New Issue