diff options
Diffstat (limited to 'client')
| -rw-r--r-- | client/brain.py | 15 | ||||
| -rw-r--r-- | client/stt.py | 34 | ||||
| -rw-r--r-- | client/test.py | 23 |
3 files changed, 56 insertions, 16 deletions
diff --git a/client/brain.py b/client/brain.py index f2a2efd..1ed6db5 100644 --- a/client/brain.py +++ b/client/brain.py @@ -32,13 +32,22 @@ class Brain(object): module, a priority of 0 is assumed. """ + logger = logging.getLogger(__name__) module_locations = [jasperpath.PLUGIN_PATH] + logger.debug("Looking for modules in: %s", ', '.join(["'%s'" % location for location in module_locations])) module_names = [name for loader, name, ispkg in pkgutil.walk_packages(module_locations, prefix='modules.')] modules = [] for name in module_names: - mod = importlib.import_module(name) - if hasattr(mod, 'WORDS'): - modules.append(mod) + try: + mod = importlib.import_module(name) + except: + logger.warning("Skipped module '%s' due to an error.", name, exc_info=True) + else: + if hasattr(mod, 'WORDS'): + logger.debug("Found module '%s' with words: %r", name, mod.WORDS) + modules.append(mod) + else: + logger.warning("Skipped module '%s' because it misses the WORDS constant.", name) modules.sort(key=lambda mod: mod.PRIORITY if hasattr(mod, 'PRIORITY') else 0, reverse=True) return modules diff --git a/client/stt.py b/client/stt.py index bcfdfba..a937658 100644 --- a/client/stt.py +++ b/client/stt.py @@ -3,6 +3,8 @@ import os import traceback import json +import tempfile +import logging import requests import yaml @@ -27,6 +29,8 @@ class PocketSphinxSTT(object): dictd_persona -- filename of the 'Persona' dictionary (.dic) """ + self._logger = logging.getLogger(__name__) + # quirky bug where first import doesn't work try: import pocketsphinx as ps @@ -46,11 +50,23 @@ class PocketSphinxSTT(object): if not hmm_dir: hmm_dir = "/usr/local/share/pocketsphinx/model/hmm/en_US/hub4wsj_sc_8k" + with tempfile.NamedTemporaryFile(prefix='psdecoder_music_', suffix='.log', delete=False) as f: + self.logfile_music = f.name + with tempfile.NamedTemporaryFile(prefix='psdecoder_persona_', suffix='.log', delete=False) as f: + self.logfile_persona = f.name + with tempfile.NamedTemporaryFile(prefix='psdecoder_default_', suffix='.log', delete=False) as f: + self.logfile_default = f.name + if lmd_music and dictd_music: - self.speechRec_music = ps.Decoder(hmm=hmm_dir, lm=lmd_music, dict=dictd_music) + self.speechRec_music = ps.Decoder(hmm=hmm_dir, lm=lmd_music, dict=dictd_music, logfn=self.logfile_music) self.speechRec_persona = ps.Decoder( - hmm=hmm_dir, lm=lmd_persona, dict=dictd_persona) - self.speechRec = ps.Decoder(hmm=hmm_dir, lm=lmd, dict=dictd) + hmm=hmm_dir, lm=lmd_persona, dict=dictd_persona, logfn=self.logfile_persona) + self.speechRec = ps.Decoder(hmm=hmm_dir, lm=lmd, dict=dictd, logfn=self.logfile_default) + + def __del__(self): + os.remove(self.logfile_music) + os.remove(self.logfile_persona) + os.remove(self.logfile_default) def transcribe(self, audio_file_path, PERSONA_ONLY=False, MUSIC=False): """ @@ -68,12 +84,24 @@ class PocketSphinxSTT(object): if MUSIC: self.speechRec_music.decode_raw(wavFile) result = self.speechRec_music.get_hyp() + with open(self.logfile_music, 'r+') as f: + for line in f: + self._logger.debug("speechRec_music %s", line.strip()) + f.truncate() elif PERSONA_ONLY: self.speechRec_persona.decode_raw(wavFile) result = self.speechRec_persona.get_hyp() + with open(self.logfile_persona, 'r+') as f: + for line in f: + self._logger.debug("speechRec_persona %s", line.strip()) + f.truncate() else: self.speechRec.decode_raw(wavFile) result = self.speechRec.get_hyp() + with open(self.logfile_default, 'r+') as f: + for line in f: + self._logger.debug("speechRec_default %s", line.strip()) + f.truncate() print "===================" print "JASPER: " + result[0] diff --git a/client/test.py b/client/test.py index 184a777..0ad37a3 100644 --- a/client/test.py +++ b/client/test.py @@ -4,7 +4,7 @@ import os import sys import unittest import argparse -from mock import patch +from mock import patch, Mock from urllib2 import URLError, urlopen import test_mic @@ -37,20 +37,23 @@ class TestVocabCompiler(unittest.TestCase): languagemodel = "temp_languagemodel.lm" words = [ - 'HACKER', 'LIFE', 'FACEBOOK', 'THIRD', 'NO', 'JOKE', - 'NOTIFICATION', 'MEANING', 'TIME', 'TODAY', 'SECOND', - 'BIRTHDAY', 'KNOCK KNOCK', 'INBOX', 'OF', 'NEWS', 'YES', - 'TOMORROW', 'EMAIL', 'WEATHER', 'FIRST', 'MUSIC', 'SPOTIFY' + 'MUSIC', 'SPOTIFY' ] + mock_module = Mock() + mock_module.WORDS = [ + 'MOCK' + ] + + words.extend(mock_module.WORDS) + with patch.object(g2p, 'translateWords') as translateWords: with patch.object(vocabcompiler, 'text2lm') as text2lm: - vocabcompiler.compile(sentences, dictionary, languagemodel) + with patch.object(brain.Brain, 'get_modules', classmethod(lambda cls: [mock_module])) as modules: + vocabcompiler.compile(sentences, dictionary, languagemodel) - # 'words' is appended with ['MUSIC', 'SPOTIFY'] - # so must be > 2 to have received WORDS from modules - translateWords.assert_called_once_with(UnorderedList(words)) - self.assertTrue(text2lm.called) + translateWords.assert_called_once_with(UnorderedList(words)) + self.assertTrue(text2lm.called) os.remove(sentences) os.remove(dictionary) |