diff options
Diffstat (limited to 'client/stt.py')
| -rw-r--r-- | client/stt.py | 23 |
1 files changed, 14 insertions, 9 deletions
diff --git a/client/stt.py b/client/stt.py index 29fb79c..4a46e71 100644 --- a/client/stt.py +++ b/client/stt.py @@ -33,7 +33,7 @@ class AbstractSTTEngine(object): return True @abstractmethod - def transcribe(self, audio_file_path, mode=TranscriptionMode.NORMAL): + def transcribe(self, fp, mode=TranscriptionMode.NORMAL): pass class PocketSphinxSTT(AbstractSTTEngine): @@ -107,7 +107,7 @@ class PocketSphinxSTT(AbstractSTTEngine): config['dictd_music'] = profile['pocketsphinx']['dictd_music'] return config - def transcribe(self, audio_file_path, mode=TranscriptionMode.NORMAL): + def transcribe(self, fp, mode=TranscriptionMode.NORMAL): """ Performs STT, transcribing an audio file and returning the result. @@ -116,12 +116,17 @@ class PocketSphinxSTT(AbstractSTTEngine): PERSONA_ONLY -- if True, uses the 'Persona' language model and dictionary MUSIC -- if True, uses the 'Music' language model and dictionary """ + decoder = self._decoders[mode] - wavFile = file(audio_file_path, 'rb') - wavFile.seek(44) + fp.seek(44) - decoder = self._decoders[TranscriptionMode.NORMAL] - decoder.decode_raw(wavFile) + # FIXME: Can't use the Decoder.decode_raw() here, because + # pocketsphinx segfaults with tempfile.SpooledTemporaryFile() + data = fp.read() + decoder.start_utt() + decoder.process_raw(data, False, True) + decoder.end_utt() + result = decoder.get_hyp() with open(self._logfiles[mode], 'r+') as f: if mode == TranscriptionMode.KEYWORD: @@ -197,7 +202,7 @@ class GoogleSTT(AbstractSTTEngine): config['api_key'] = profile['keys']['GOOGLE_SPEECH'] return config - def transcribe(self, audio_fp, mode=TranscriptionMode.NORMAL): + def transcribe(self, fp, mode=TranscriptionMode.NORMAL): """ Performs STT via the Google Speech API, transcribing an audio file and returning an English string. @@ -205,11 +210,11 @@ class GoogleSTT(AbstractSTTEngine): Arguments: audio_file_path -- the path to the .wav file to be transcribed """ + url = "https://www.google.com/speech-api/v2/recognize?output=json&client=chromium&key=%s&lang=%s&maxresults=6&pfilter=2" % ( self.api_key, "en-us") - with open(audio_file_path, 'rb') as f: - data = f.read() + data = fp.read() try: headers = {'Content-type': 'audio/l16; rate=%s' % GoogleSTT.RATE} |