From a7c54f48e864bd05f9b69ee5f8058b1de19ed17d Mon Sep 17 00:00:00 2001
From: Jason <thebosssg19@gmail.com>
Date: Mon, 11 Jul 2022 20:42:37 -0400
Subject: [PATCH] Improved AWS Polly eh

improved handling for errors like
like #929
---
 TTS/aws_polly.py | 64 +++++++++++++++++++++++++++---------------------
 1 file changed, 36 insertions(+), 28 deletions(-)

diff --git a/TTS/aws_polly.py b/TTS/aws_polly.py
index 13eaea1..e1d5318 100644
--- a/TTS/aws_polly.py
+++ b/TTS/aws_polly.py
@@ -1,6 +1,6 @@
 #!/usr/bin/env python3
 from boto3 import Session
-from botocore.exceptions import BotoCoreError, ClientError
+from botocore.exceptions import BotoCoreError, ClientError, ProfileNotFound
 import sys
 from utils import settings
 import random
@@ -30,36 +30,44 @@ class AWSPolly:
         self.voices = voices
 
     def run(self, text, filepath, random_voice: bool = False):
-        session = Session(profile_name="polly")
-        polly = session.client("polly")
-        if random_voice:
-            voice = self.randomvoice()
-        else:
-            if not settings.config["settings"]["tts"]["aws_polly_voice"]:
-                return ValueError(
-                    f"Please set the TOML variable AWS_VOICE to a valid voice. options are: {voices}"
-                )
-            voice = str(settings.config["settings"]["tts"]["aws_polly_voice"]).capitalize()
         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)
+            session = Session(profile_name="polly")
+            polly = session.client("polly")
+            if random_voice:
+                voice = self.randomvoice()
+            else:
+                if not settings.config["settings"]["tts"]["aws_polly_voice"]:
+                    return ValueError(
+                        f"Please set the TOML variable AWS_VOICE to a valid voice. options are: {voices}"
+                    )
+                voice = str(settings.config["settings"]["tts"]["aws_polly_voice"]).capitalize()
+            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")
+            # 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")
+            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)
 
     def randomvoice(self):