wit.py 2.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354
  1. import speech_recognition as sr
  2. from core import Utils
  3. from core.OrderListener import OrderListener
  4. class Wit(OrderListener):
  5. def __init__(self, callback=None, **kwargs):
  6. """
  7. Start recording the microphone and analyse audio with Wit.ai 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 Wit.ai Speech Recognition
  22. try:
  23. key = kwargs.get('key', None)
  24. show_all = kwargs.get('show_all', False)
  25. captured_audio = r.recognize_wit(audio, key=key, show_all=show_all)
  26. Utils.print_success("Wit.ai Speech Recognition thinks you said %s" % captured_audio)
  27. self._analyse_audio(captured_audio)
  28. except sr.UnknownValueError:
  29. Utils.print_warning("Wit.ai Speech Recognition could not understand audio")
  30. # callback anyway, we need to listen again for a new order
  31. self._analyse_audio(audio=None)
  32. except sr.RequestError as e:
  33. Utils.print_danger("Could not request results from Wit.ai Speech Recognition service; {0}".format(e))
  34. # callback anyway, we need to listen again for a new order
  35. self._analyse_audio(audio=None)
  36. def _analyse_audio(self, audio):
  37. """
  38. Confirm the audio exists annd run it in a Callback
  39. :param audio: the captured audio
  40. """
  41. # if self.main_controller is not None:
  42. # self.main_controller.analyse_order(audio)
  43. if self.callback is not None:
  44. self.callback(audio)