tts.py

Go to the documentation of this file.

"""Support for the MaryTTS service."""
 
from __future__ import annotations
 
from speak2mary import MaryTTS
import voluptuous as vol
 
from homeassistant.components.tts import (
    CONF_LANG,
    PLATFORM_SCHEMA as TTS_PLATFORM_SCHEMA,
    Provider,
)
from homeassistant.const import CONF_EFFECT, CONF_HOST, CONF_PORT
import homeassistant.helpers.config_validation as cv
 
CONF_VOICE = "voice"
CONF_CODEC = "codec"
 
SUPPORT_LANGUAGES = MaryTTS.supported_locales()
SUPPORT_CODEC = MaryTTS.supported_codecs()
SUPPORT_OPTIONS = [CONF_EFFECT]
SUPPORT_EFFECTS = MaryTTS.supported_effects().keys()
 
DEFAULT_HOST = "localhost"
DEFAULT_PORT = 59125
DEFAULT_LANG = "en_US"
DEFAULT_VOICE = "cmu-slt-hsmm"
DEFAULT_CODEC = "WAVE_FILE"
DEFAULT_EFFECTS: dict[str, str] = {}
 
MAP_MARYTTS_CODEC = {"WAVE_FILE": "wav", "AIFF_FILE": "aiff", "AU_FILE": "au"}
 
PLATFORM_SCHEMA = TTS_PLATFORM_SCHEMA.extend(
    {
        vol.Optional(CONF_HOST, default=DEFAULT_HOST): cv.string,
        vol.Optional(CONF_PORT, default=DEFAULT_PORT): cv.port,
        vol.Optional(CONF_LANG, default=DEFAULT_LANG): vol.In(SUPPORT_LANGUAGES),
        vol.Optional(CONF_VOICE, default=DEFAULT_VOICE): cv.string,
        vol.Optional(CONF_CODEC, default=DEFAULT_CODEC): vol.In(SUPPORT_CODEC),
        vol.Optional(CONF_EFFECT, default=DEFAULT_EFFECTS): {
            vol.All(cv.string, vol.In(SUPPORT_EFFECTS)): cv.string
        },
    }
)
 
 
def get_engine(hass, config, discovery_info=None):
    """Set up MaryTTS speech component."""
    return MaryTTSProvider(hass, config)
 
 
class MaryTTSProvider(Provider):
    """MaryTTS speech api provider."""
 
    def __init__(self, hass, conf):
        """Init MaryTTS TTS service."""
        self.hasshass = hass
        self._mary_mary = MaryTTS(
            conf.get(CONF_HOST),
            conf.get(CONF_PORT),
            conf.get(CONF_CODEC),
            conf.get(CONF_LANG),
            conf.get(CONF_VOICE),
        )
        self._effects_effects = conf.get(CONF_EFFECT)
        self.namename = "MaryTTS"
 
    @property
    def default_language(self):
        """Return the default language."""
        return self._mary_mary.locale
 
    @property
    def supported_languages(self):
        """Return list of supported languages."""
        return SUPPORT_LANGUAGES
 
    @property
    def default_options(self):
        """Return dict include default options."""
        return {CONF_EFFECT: self._effects_effects}
 
    @property
    def supported_options(self):
        """Return a list of supported options."""
        return SUPPORT_OPTIONS
 
    def get_tts_audio(self, message, language, options):
        """Load TTS from MaryTTS."""
        effects = options[CONF_EFFECT]
 
        data = self._mary_mary.speak(message, effects)
        audiotype = MAP_MARYTTS_CODEC[self._mary_mary.codec]
 
        return audiotype, data