RedditVideoMakerBot/TTS/streamlabs_polly.py

import requests
from requests.exceptions import JSONDecodeError
from utils import settings
from attr import attrs, attrib
from attr.validators import instance_of

from TTS.common import BaseApiTTS, get_random_voice
from utils.voice import check_ratelimit

voices = [
    'Brian',
    'Emma',
    'Russell',
    'Joey',
    'Matthew',
    'Joanna',
    'Kimberly',
    'Amy',
    'Geraint',
    'Nicole',
    'Justin',
    'Ivy',
    'Kendra',
    'Salli',
    'Raveena',
]


# valid voices https://lazypy.ro/tts/


@attrs
class StreamlabsPolly(BaseApiTTS):
    random_voice: bool = attrib(
        validator=instance_of(bool),
        default=False
    )
    url: str = 'https://streamlabs.com/polly/speak',
    max_chars: int = 550

    def make_request(
            self,
            text,
    ):
        """
        Makes a requests to remote TTS service

        Args:
            text: text to be voice over

        Returns:
            Request's response
        """
        voice = (
            get_random_voice(voices)
            if self.random_voice
            else str(settings.config['settings']['tts']['streamlabs_polly_voice']).capitalize()
            if str(settings.config['settings']['tts']['streamlabs_polly_voice']).lower() in [
                voice.lower() for voice in voices]
            else get_random_voice(voices)
        )
        response = requests.post(
            self.url,
            data={
                'voice': voice,
                'text': text,
                'service': 'polly',
            })
        if not check_ratelimit(response):
            return self.make_request(text)
        else:
            try:
                results = requests.get(response.json()['speak_url'])
                return results
            except (KeyError, JSONDecodeError):
                try:
                    if response.json()['error'] == 'No text specified!':
                        raise ValueError('Please specify a text to convert to speech.')
                except (KeyError, JSONDecodeError):
                    print('Error occurred calling Streamlabs Polly')