Somewhat fancy voice command recognition software
您最多选择25个主题 主题必须以字母或数字开头,可以包含连字符 (-),并且长度不得超过35个字符

kaylee.py 8.3KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246
  1. # This is part of Kaylee
  2. # -- this code is licensed GPLv3
  3. # Copyright 2015-2017 Clayton G. Hobbs
  4. # Portions Copyright 2013 Jezra
  5. import importlib
  6. import queue
  7. import sys
  8. import subprocess
  9. import signal
  10. import os.path
  11. from gi.repository import GObject, GLib
  12. from .recognizer import Recognizer
  13. from .util import *
  14. from .numbers import NumberParser
  15. import kayleevc.plugins
  16. class Kaylee:
  17. def __init__(self):
  18. self.ui = None
  19. self.options = {}
  20. self.continuous_listen = False
  21. # Load configuration
  22. self.config = Config()
  23. self.options = vars(self.config.options)
  24. # Make sure some plugins are configured to be loaded
  25. if self.config.plugins is None:
  26. print("error: no plugins configured", file=sys.stderr)
  27. sys.exit(1)
  28. # Load plugins
  29. self.plugins = []
  30. for plugin in self.config.plugins.keys():
  31. pmod = importlib.import_module(plugin, 'kayleevc.plugins')
  32. pobj = pmod.Plugin(self.config, plugin)
  33. pobj.connect('tts', self.plugin_tts)
  34. self.plugins.append(pobj)
  35. # Create a hasher
  36. self.hasher = Hasher(self.config)
  37. # Create the strings file
  38. self.update_voice_commands_if_changed()
  39. if self.options['interface']:
  40. if self.options['interface'] == "g":
  41. from kayleevc.gui import GTKInterface as UI
  42. elif self.options['interface'] == "gt":
  43. from kayleevc.gui import GTKTrayInterface as UI
  44. else:
  45. print("no GUI defined")
  46. sys.exit()
  47. self.ui = UI(self.options, self.options['continuous'])
  48. self.ui.connect("command", self.process_command)
  49. # Can we load the icon resource?
  50. icon = self.load_resource("icon_small.png")
  51. if icon:
  52. self.ui.set_icon_active_asset(icon)
  53. # Can we load the icon_inactive resource?
  54. icon_inactive = self.load_resource("icon_inactive_small.png")
  55. if icon_inactive:
  56. self.ui.set_icon_inactive_asset(icon_inactive)
  57. if self.options['history']:
  58. self.history = []
  59. # Update the language if necessary
  60. self.language_updater = LanguageUpdater(self.config)
  61. self.language_updater.update_language_if_changed()
  62. # Create the recognizer
  63. self.recognizer = Recognizer(self.config)
  64. # Connect the recognizer's finished signal to the appropriate method
  65. self.recognizer.connect('finished', self.recognizer_finished)
  66. def update_voice_commands_if_changed(self):
  67. """Use hashes to test if the voice commands have changed"""
  68. stored_hash = self.hasher['voice_commands']
  69. # Calculate the hash the voice commands have right now
  70. hasher = self.hasher.get_hash_object()
  71. for plugin in self.plugins:
  72. for string in sorted(plugin.corpus_strings):
  73. hasher.update(string.encode('utf-8'))
  74. # Add a separator to avoid odd behavior
  75. hasher.update('\n'.encode('utf-8'))
  76. new_hash = hasher.hexdigest()
  77. if new_hash != stored_hash:
  78. self.create_strings_file()
  79. self.hasher['voice_commands'] = new_hash
  80. self.hasher.store()
  81. def create_strings_file(self):
  82. # Open the strings file
  83. with open(self.config.strings_file, 'w') as strings:
  84. # Add command words to the corpus
  85. # FIXME: Doing this twice is a silly thing
  86. for plugin in self.plugins:
  87. for string in sorted(plugin.corpus_strings):
  88. strings.write(string + "\n")
  89. # Add number words to the corpus
  90. if NumberParser.number_words is not None:
  91. for word in NumberParser.number_words:
  92. strings.write(word + " ")
  93. strings.write("\n")
  94. def _log_history(self, plugin, text):
  95. """Log the recognized sentence to the history file"""
  96. if self.options['history']:
  97. self.history.append("{}: {}".format(plugin.name, text))
  98. if len(self.history) > self.options['history']:
  99. # Pop off the first item
  100. self.history.pop(0)
  101. # Open and truncate the history file
  102. with open(self.config.history_file, 'w') as hfile:
  103. for line in self.history:
  104. hfile.write(line + '\n')
  105. self._stop_ui(text)
  106. def _valid_cmd(self):
  107. """Run the valid_sentence_command if it's set"""
  108. if self.options['valid_sentence_command']:
  109. subprocess.call(self.options['valid_sentence_command'], shell=True)
  110. def _invalid_cmd(self, text):
  111. """Run the invalid_sentence_command if it's set"""
  112. if self.options['invalid_sentence_command']:
  113. subprocess.call(self.options['invalid_sentence_command'],
  114. shell=True)
  115. print("no matching command {0}".format(text))
  116. def recognizer_finished(self, recognizer, text):
  117. confidence_heap = queue.PriorityQueue()
  118. min_confidence = self.options['minimum_confidence']
  119. # Add plugins to the heap
  120. for index, plugin in enumerate(self.plugins):
  121. # Get plugin confidence
  122. confidence = plugin.confidence(text)
  123. # Clamp confidence to [0, 1]
  124. confidence = min(max(confidence, 0), 1)
  125. # If the plugin meets minimum confidence
  126. if ((min_confidence is not None and confidence >= min_confidence)
  127. or (min_confidence is None and confidence > 0)):
  128. # Add a triple to the heap, so plugins are sorted first by
  129. # confidence, then by index to break ties
  130. confidence_heap.put_nowait((1 - confidence, index, plugin))
  131. # Run valid or invalid sentence command
  132. if confidence_heap.empty():
  133. self._invalid_cmd(text)
  134. else:
  135. self._valid_cmd()
  136. # Give the command to the plugins that want it
  137. while True:
  138. try:
  139. plugin = confidence_heap.get_nowait()[2]
  140. except queue.Empty:
  141. break
  142. # If the plugin successfully handled the command
  143. if plugin.handle(text):
  144. self._log_history(plugin, text)
  145. break
  146. self._stop_ui(text)
  147. def plugin_tts(self, plugin, text):
  148. # Stop listening
  149. self.recognizer.pause()
  150. # Speak
  151. try:
  152. subprocess.call(self.options['tts'] + [text])
  153. except KeyError:
  154. print('TTS:', text)
  155. # Resume listening
  156. self.recognizer.listen()
  157. def _stop_ui(self, text):
  158. # If there is a UI and we are not continuous listen
  159. if self.ui:
  160. if not self.continuous_listen:
  161. # Stop listening
  162. self.recognizer.pause()
  163. # Let the UI know that there is a finish
  164. self.ui.finished(text)
  165. def run(self):
  166. if self.ui:
  167. self.ui.run()
  168. else:
  169. self.recognizer.listen()
  170. def quit(self):
  171. sys.exit()
  172. def process_command(self, UI, command):
  173. print(command)
  174. if command == "listen":
  175. self.recognizer.listen()
  176. elif command == "stop":
  177. self.continuous_listen = False
  178. self.recognizer.pause()
  179. elif command == "continuous_listen":
  180. self.continuous_listen = True
  181. self.recognizer.listen()
  182. elif command == "quit":
  183. self.quit()
  184. def load_resource(self, string):
  185. # TODO: Use the Config object for this path management
  186. local_data = os.path.join(os.path.dirname(__file__), '..', 'data')
  187. paths = ["/usr/share/kaylee/", "/usr/local/share/kaylee", local_data]
  188. for path in paths:
  189. resource = os.path.join(path, string)
  190. if os.path.exists(resource):
  191. return resource
  192. # If we get this far, no resource was found
  193. return False
  194. def run():
  195. # Make our kaylee object
  196. kaylee = Kaylee()
  197. # Init gobject threads
  198. GObject.threads_init()
  199. # We want a main loop
  200. main_loop = GObject.MainLoop()
  201. # Handle sigint
  202. signal.signal(signal.SIGINT, signal.SIG_DFL)
  203. # Run the kaylee
  204. kaylee.run()
  205. # Start the main loop
  206. try:
  207. main_loop.run()
  208. except:
  209. main_loop.quit()
  210. sys.exit()