change to TTS

app.py

@@ -1,35 +1,48 @@
 import streamlit as st
-from gtts import gTTS
+# from gtts import gTTS
 import io
+import torch
+from TTS.api import TTS
 
 # Function to generate audio from text
-def text_to_audio(text):
-    tts = gTTS(text)
-    audio = tts.save("output.mp3")
-    return audio
+# def text_to_audio(text):
+#     tts = gTTS(text)
+#     audio = tts.save("output.mp3")
+#     return audio
 
 st.title("Voice Synthesis App")
+# Get device
+device = "cuda" if torch.cuda.is_available() else "cpu"
+
+# List available 🐸TTS models
+print(TTS().list_models())
+
+# Init TTS
+tts = TTS("tts_models/en/ljspeech/overflow").to(device)
 
 # Input form for the user to enter text
 user_input = st.text_area("Enter the text:")
-
 if st.button("Generate"):
     if user_input:
-        audio = text_to_audio(user_input)
+        # audio = text_to_audio(user_input)
+        
+        # Text to speech to a file
+        tts.tts_to_file(text="Hello world!", file_path="output.wav")
 
         # Display the audio
-        st.audio("output.mp3", format="audio/mp3")
+        st.audio("output.wav", format="audio/wav")
 
         # Create a download link for the audio file
         st.markdown("### Download the audio")
-        with open("output.mp3", "rb") as audio_file:
+        with open("output.wav", "rb") as audio_file:
             audio_bytes = audio_file.read()
+            
         st.download_button(
             label="Click to download",
             data=audio_bytes,
             key="download_audio",
-            file_name="output.mp3",
-            mime="audio/mp3",
+            file_name="output.wav",
+            mime="audio/wav",
         )
 
 st.write("Note: This app uses gTTS for text-to-speech conversion.")

requirements.txt

@@ -1 +1,2 @@
-gTTS
+gTTS
+TTS