apiai.py 2.2 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556
  1. import speech_recognition as sr
  2. from core import Utils
  3. from core.OrderListener import OrderListener
  4. class Apiai(OrderListener):
  5. def __init__(self, callback=None, **kwargs):
  6. """
  7. Start recording the microphone and analyse audio with Apiai api
  8. :param callback: The callback function to call to send the text
  9. :param kwargs:
  10. """
  11. OrderListener.__init__(self)
  12. # callback function to call after the translation speech/tex
  13. self.callback = callback
  14. # obtain audio from the microphone
  15. r = sr.Recognizer()
  16. with sr.Microphone() as source:
  17. # listen for 1 second to calibrate the energy threshold for ambient noise levels
  18. r.adjust_for_ambient_noise(source)
  19. Utils.print_info("Say something!")
  20. audio = r.listen(source)
  21. # recognize speech using Apiai Speech Recognition
  22. try:
  23. key = kwargs.get('key', None)
  24. language = kwargs.get('language', "en")
  25. session_id= kwargs.get('session_id', None)
  26. show_all = kwargs.get('show_all', False)
  27. captured_audio = r.recognize_api(audio, client_access_token=key, language=language, session_id=session_id, show_all=show_all)
  28. Utils.print_success("Apiai Speech Recognition thinks you said %s" % captured_audio)
  29. self._analyse_audio(captured_audio)
  30. except sr.UnknownValueError as e:
  31. Utils.print_warning("Apiai Speech Recognition could not understand audio; {0}".format(e))
  32. # callback anyway, we need to listen again for a new order
  33. self._analyse_audio(audio=None)
  34. except sr.RequestError as e:
  35. Utils.print_danger("Could not request results from Apiai Speech Recognition service; {0}".format(e))
  36. # callback anyway, we need to listen again for a new order
  37. self._analyse_audio(audio=None)
  38. def _analyse_audio(self, audio):
  39. """
  40. Confirm the audio exists annd run it in a Callback
  41. :param audio: the captured audio
  42. """
  43. # if self.main_controller is not None:
  44. # self.main_controller.analyse_order(audio)
  45. if self.callback is not None:
  46. self.callback(audio)