2019-11-04 12:10:42 +00:00
|
|
|
"""Support for the demo for speech to text service."""
|
|
|
|
from typing import List
|
|
|
|
|
|
|
|
from aiohttp import StreamReader
|
|
|
|
|
|
|
|
from homeassistant.components.stt import Provider, SpeechMetadata, SpeechResult
|
|
|
|
from homeassistant.components.stt.const import (
|
2019-11-04 22:01:10 +00:00
|
|
|
AudioBitRates,
|
2019-11-04 12:10:42 +00:00
|
|
|
AudioFormats,
|
2019-11-04 22:01:10 +00:00
|
|
|
AudioSampleRates,
|
2019-11-04 12:10:42 +00:00
|
|
|
AudioCodecs,
|
|
|
|
SpeechResultState,
|
|
|
|
)
|
|
|
|
|
|
|
|
SUPPORT_LANGUAGES = ["en", "de"]
|
|
|
|
|
|
|
|
|
|
|
|
async def async_get_engine(hass, config):
|
|
|
|
"""Set up Demo speech component."""
|
|
|
|
return DemoProvider()
|
|
|
|
|
|
|
|
|
|
|
|
class DemoProvider(Provider):
|
|
|
|
"""Demo speech API provider."""
|
|
|
|
|
|
|
|
@property
|
|
|
|
def supported_languages(self) -> List[str]:
|
|
|
|
"""Return a list of supported languages."""
|
|
|
|
return SUPPORT_LANGUAGES
|
|
|
|
|
|
|
|
@property
|
|
|
|
def supported_formats(self) -> List[AudioFormats]:
|
|
|
|
"""Return a list of supported formats."""
|
|
|
|
return [AudioFormats.WAV]
|
|
|
|
|
|
|
|
@property
|
|
|
|
def supported_codecs(self) -> List[AudioCodecs]:
|
|
|
|
"""Return a list of supported codecs."""
|
|
|
|
return [AudioCodecs.PCM]
|
|
|
|
|
|
|
|
@property
|
2019-11-04 22:01:10 +00:00
|
|
|
def supported_bit_rates(self) -> List[AudioBitRates]:
|
|
|
|
"""Return a list of supported bit rates."""
|
|
|
|
return [AudioBitRates.BITRATE_16]
|
2019-11-04 12:10:42 +00:00
|
|
|
|
|
|
|
@property
|
2019-11-04 22:01:10 +00:00
|
|
|
def supported_sample_rates(self) -> List[AudioSampleRates]:
|
|
|
|
"""Return a list of supported sample rates."""
|
|
|
|
return [AudioSampleRates.SAMPLERATE_16000, AudioSampleRates.SAMPLERATE_44100]
|
2019-11-04 12:10:42 +00:00
|
|
|
|
|
|
|
async def async_process_audio_stream(
|
|
|
|
self, metadata: SpeechMetadata, stream: StreamReader
|
|
|
|
) -> SpeechResult:
|
|
|
|
"""Process an audio stream to STT service."""
|
|
|
|
|
|
|
|
# Read available data
|
|
|
|
async for _ in stream.iter_chunked(4096):
|
|
|
|
pass
|
|
|
|
|
|
|
|
return SpeechResult("Turn the Kitchen Lights on", SpeechResultState.SUCCESS)
|