File size: 3,427 Bytes
a801f9e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 |
import streamlit as st
from openai import OpenAI
import os
from pathlib import Path
import base64
def create_audio_player(audio_path):
"""Create an HTML audio player for the generated audio file"""
audio_file = open(audio_path, 'rb')
audio_bytes = audio_file.read()
audio_base64 = base64.b64encode(audio_bytes).decode()
audio_player = f'<audio controls><source src="data:audio/mp3;base64,{audio_base64}" type="audio/mp3"></audio>'
return audio_player
def main():
st.title("🗣️ Text to Speech with OpenAI")
# Sidebar for API key
with st.sidebar:
st.header("Configuration")
api_key = st.text_input("Enter your OpenAI API key", type="password")
st.markdown("""
### How to get an API key
1. Go to [OpenAI API Keys](https://platform.openai.com/api-keys)
2. Create a new secret key
3. Copy and paste it here
""")
# Main content
st.markdown("Convert your text to natural-sounding speech using OpenAI's TTS API.")
# Text input
user_input = st.text_area("Enter the text you want to convert to speech:",
height=150,
placeholder="Enter your text here...")
# Voice selection
voice_option = st.selectbox(
"Select a voice:",
["alloy", "echo", "fable", "onyx", "nova", "shimmer"]
)
# Create a button to generate audio
if st.button("Generate Speech"):
if not api_key:
st.error("Please enter your OpenAI API key in the sidebar.")
return
if not user_input:
st.error("Please enter some text to convert to speech.")
return
try:
# Show spinning indicator while processing
with st.spinner("Generating audio..."):
# Initialize OpenAI client with the API key
client = OpenAI(api_key=api_key)
# Create output directory if it doesn't exist
output_dir = Path("generated_audio")
output_dir.mkdir(exist_ok=True)
# Generate unique filename
output_file = output_dir / "output.mp3"
# Generate speech
response = client.audio.speech.create(
model="tts-1",
voice=voice_option,
input=user_input
)
# Save the audio file
response.stream_to_file(str(output_file))
# Display success message
st.success("Audio generated successfully!")
# Create and display audio player
st.markdown("### Listen to the generated audio:")
st.markdown(create_audio_player(output_file), unsafe_allow_html=True)
# Add download button
with open(output_file, "rb") as file:
st.download_button(
label="Download Audio",
data=file,
file_name="generated_speech.mp3",
mime="audio/mp3"
)
except Exception as e:
st.error(f"An error occurred: {str(e)}")
if __name__ == "__main__":
main() |