From 3778ca2a1413f94f8b432f843c468f4a774a74c3 Mon Sep 17 00:00:00 2001
From: Jason <thebosssg19@gmail.com>
Date: Thu, 16 Jun 2022 23:25:51 -0400
Subject: [PATCH] fix: fixed language options

---
 TTS/POLLY.py | 39 ++++-----------------------------------
 1 file changed, 4 insertions(+), 35 deletions(-)

diff --git a/TTS/POLLY.py b/TTS/POLLY.py
index 95db5e6..55e7471 100644
--- a/TTS/POLLY.py
+++ b/TTS/POLLY.py
@@ -6,37 +6,7 @@ import requests
 import sox
 from moviepy.audio.AudioClip import concatenate_audioclips, CompositeAudioClip
 from moviepy.audio.io.AudioFileClip import AudioFileClip
-voices = {'neural': [
-    'Ivy',
-    'Joanna',
-    'Kendra',
-    'Kimberly',
-    'Salli',
-    'Joey',
-    'Justin',
-    'Matthew',
-    'Amy',
-    'Emma',
-    'Brian'
-
-], 'standard': [
-    'Ivy',
-    'Joanna',
-    'Kendra',
-    'Kimberly',
-    'Salli',
-    'Joey',
-    'Justin',
-    'Matthew',
-    "Russell",
-    "Nicole",
-    "Amy",
-    "Emma",
-    "Brian",
-    "Aditi",
-    "Raveena",
-    "Geraint"
-]}
+voices = ['Brian', 'Emma', 'Russell', 'Joey', 'Matthew', 'Joanna', 'Kimberly', 'Amy', 'Geraint', 'Nicole', 'Justin', 'Ivy', 'Kendra', 'Salli', 'Raveena']
 
 
 # valid voices https://lazypy.ro/tts/
@@ -59,7 +29,7 @@ class POLLY:
             if not os.getenv('VOICE'):
                 return ValueError('Please set the environment variable VOICE to a valid voice. options are: {}'.format(voices))
             voice = str(os.getenv("VOICE")).capitalize()
-        body = {'voice': voice, 'text': req_text}
+        body = {'voice': voice, 'text': req_text, 'service': 'polly'}
         response = requests.post(self.url, data=body)
         try:
             voice_data = requests.get(response.json()['speak_url'])
@@ -76,7 +46,7 @@ class POLLY:
 
                 chunkId = 0
                 for chunk in chunks:
-                    body = {'voice': voice, 'text': chunk}
+                    body = {'voice': voice, 'text': chunk, 'service': 'polly'}
                     resp = requests.post(self.url, data=body)
                     voice_data = requests.get(resp.json()['speak_url'])
                     with open(filename.replace(".mp3", f"-{chunkId}.mp3"), "wb") as out:
@@ -111,5 +81,4 @@ class POLLY:
         return text
 
     def randomvoice(self):
-        valid = voices['neural'] + voices['standard']
-        return random.choice(valid)
+        return random.choice(voices)