pico2wave.py 1.2 KB

1234567891011121314151617181920212223242526272829303132
  1. import subprocess
  2. from tts import TTS
  3. import logging, sys
  4. class Pico2wave(TTS):
  5. PICO2WAVE_LANGUAGES = dict(fr="fr-FR", us="en-US", uk="en-GB", de="de-DE", es="es-ES", it="it-IT")
  6. PICO2WAVE_LANGUAGES_DEFAULT = PICO2WAVE_LANGUAGES['fr']
  7. def __init__(self, audio_player_type=None):
  8. TTS.__init__(self, audio_player_type, "wav")
  9. def say(self, words=None, language=PICO2WAVE_LANGUAGES_DEFAULT, cache=False):
  10. language = self.get_voice(language)
  11. file_path = self.cache.get_audio_file_cache_path(words, language=language)
  12. self.get_audio(words, language, file_path)
  13. self.play_audio(file_path, cache=cache)
  14. def get_voice(self, language):
  15. if language in self.PICO2WAVE_LANGUAGES:
  16. return self.PICO2WAVE_LANGUAGES[language]
  17. logging.warn("Cannot find language matching language: %s voice: %s replace by default voice: %s", language, self.PICO2WAVE_LANGUAGES_DEFAULT)
  18. return self.PICO2WAVE_LANGUAGES_DEFAULT
  19. @staticmethod
  20. def get_audio(words, language, file_path):
  21. subprocess.check_output(["/usr/bin/pico2wave", "-l=%s" % language, "-w=%s" % file_path, words], stderr=sys.stderr)
  22. # logging.basicConfig(stream=sys.stderr, level=logging.DEBUG)