RedditVideoMakerBot/TTS/aws_polly.py

#!/usr/bin/env python3
from boto3 import Session
from botocore.exceptions import BotoCoreError, ClientError, ProfileNotFound

import sys
from utils import settings
from attr import attrs, attrib
from attr.validators import instance_of

from TTS.common import get_random_voice


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


@attrs
class AWSPolly:
    random_voice: bool = attrib(
        validator=instance_of(bool),
        default=False
    )
    max_chars: int = 0

    def run(
            self,
            text,
            filepath,
    ) -> None:
        """
        Calls for TTS api

        Args:
            text: text to be voiced over
            filepath: name of the audio file
        """
        try:
            session = Session(profile_name='polly')
            polly = session.client('polly')
            voice = (
                get_random_voice(voices)
                if self.random_voice
                else str(settings.config['settings']['tts']['aws_polly_voice']).capitalize()
                if str(settings.config['settings']['tts']['aws_polly_voice']).lower() in [voice.lower() for voice in
                                                                                          voices]
                else get_random_voice(voices)
            )
            try:
                # Request speech synthesis
                response = polly.synthesize_speech(
                    Text=text, OutputFormat='mp3', VoiceId=voice, Engine='neural'
                )
            except (BotoCoreError, ClientError) as error:
                # The service returned an error, exit gracefully
                print(error)
                sys.exit(-1)

            # Access the audio stream from the response
            if 'AudioStream' in response:
                file = open(filepath, 'wb')
                file.write(response['AudioStream'].read())
                file.close()
                # print_substep(f"Saved Text {idx} to MP3 files successfully.", style="bold green")

            else:
                # The response didn't contain audio data, exit gracefully
                print('Could not stream audio')
                sys.exit(-1)
        except ProfileNotFound:
            print("You need to install the AWS CLI and configure your profile")
            print(
                """
            Linux: https://docs.aws.amazon.com/polly/latest/dg/setup-aws-cli.html
            Windows: https://docs.aws.amazon.com/polly/latest/dg/install-voice-plugin2.html
            """
            )
            sys.exit(-1)