123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354 |
- import speech_recognition as sr
- from core import Utils
- from core.OrderListener import OrderListener
- class Wit(OrderListener):
- def __init__(self, callback=None, **kwargs):
- """
- Start recording the microphone and analyse audio with Wit.ai api
- :param callback: The callback function to call to send the text
- :param kwargs:
- """
- OrderListener.__init__(self)
- # callback function to call after the translation speech/tex
- self.callback = callback
- # obtain audio from the microphone
- r = sr.Recognizer()
- with sr.Microphone() as source:
- # listen for 1 second to calibrate the energy threshold for ambient noise levels
- r.adjust_for_ambient_noise(source)
- Utils.print_info("Say something!")
- audio = r.listen(source)
- # recognize speech using Wit.ai Speech Recognition
- try:
- key = kwargs.get('key', None)
- show_all = kwargs.get('show_all', False)
- captured_audio = r.recognize_wit(audio, key=key, show_all=show_all)
- Utils.print_success("Wit.ai Speech Recognition thinks you said %s" % captured_audio)
- self._analyse_audio(captured_audio)
- except sr.UnknownValueError:
- Utils.print_warning("Wit.ai Speech Recognition could not understand audio")
- # callback anyway, we need to listen again for a new order
- self._analyse_audio(audio=None)
- except sr.RequestError as e:
- Utils.print_danger("Could not request results from Wit.ai Speech Recognition service; {0}".format(e))
- # callback anyway, we need to listen again for a new order
- self._analyse_audio(audio=None)
- def _analyse_audio(self, audio):
- """
- Confirm the audio exists annd run it in a Callback
- :param audio: the captured audio
- """
- # if self.main_controller is not None:
- # self.main_controller.analyse_order(audio)
- if self.callback is not None:
- self.callback(audio)
|