From 97e16702a142ec68015669a47687f331f15cc57a Mon Sep 17 00:00:00 2001
From: Adam Blank <blank@caltech.edu>
Date: Wed, 25 Sep 2024 00:09:04 +0000
Subject: [PATCH] Update util.py

---
 00/util.py | 17 ++++++++++++-----
 1 file changed, 12 insertions(+), 5 deletions(-)

diff --git a/00/util.py b/00/util.py
index 8350474..a4a70b1 100644
--- a/00/util.py
+++ b/00/util.py
@@ -1,7 +1,14 @@
-from gtts import gTTS
-import playsound
+from pyt2s.services import stream_elements
+from pydub import AudioSegment
+from pydub.playback import play as p
+import io
+
 
 def play(script):
-    tts = gTTS(text=script, lang='en')
-    tts.save("sound.mp3")
-    playsound.playsound("sound.mp3")
+    play_with_voice(script, "Amy")
+
+
+def play_with_voice(script, voice):
+    data = stream_elements.requestTTS(script, voice)
+    song = AudioSegment.from_file(io.BytesIO(data), format="mp3")
+    p(song)
-- 
GitLab