Spaces:
Running
Running
import gradio as gr | |
import os | |
import tempfile | |
from openai import OpenAI | |
def generate_tts(text, model, voice, api_key,response_format, speed): | |
if api_key == '': | |
raise gr.Error('Please enter your OpenAI API key to run this generator') | |
else: | |
try: | |
client = OpenAI(api_key=api_key) | |
response = client.audio.speech.create( | |
model=model, # tts-1, tts-1-hd | |
voice=voice, # alloy, echo, fable, onyx, nova, shimmer | |
response_format=response_format, # mp3, opus, aac, flac | |
speed=speed, # 0.25x - 4x | |
input=text, | |
) | |
except Exception as error: | |
# Handle any exception that occurs | |
raise gr.Error("An error occurred while generating speech output. Please check your API key and try again.") | |
print(str(error)) | |
# Create a temp file to save the audio from OpenAI api | |
output_suffix = "." + response_format | |
with tempfile.NamedTemporaryFile(suffix=output_suffix, delete=False) as temp_file: | |
temp_file.write(response.content) | |
# Get the file path of the temp file | |
temp_file_path = temp_file.name | |
return temp_file_path | |
css = """ | |
.header-text p {line-height: 80px !important; text-align: left; font-size: 26px;} | |
.header-logo {text-align: left} | |
""" | |
with gr.Blocks(css=css) as tts_demo: | |
with gr.Row(): | |
with gr.Column(scale=1, min_width=80): | |
gr.Image("tt-logo.jpg", width=80, height=80, show_download_button=False, show_share_button=False, interactive=False, show_label=False, elem_id="thinktecture-logo", elem_classes="header-logo", container=False) | |
with gr.Column(scale=11): | |
gr.Markdown("OpenAI Text-To-Speech Generator (TTS-1 Model)", elem_classes="header-text") | |
with gr.Row(variant='panel'): | |
api_key = gr.Textbox(type='password', label='OpenAI API Key', placeholder='Your OpenAI API key') | |
model = gr.Dropdown(choices=['tts-1','tts-1-hd'], label='Model', value='tts-1') | |
voice = gr.Dropdown(choices=['alloy', 'echo', 'fable', 'onyx', 'nova', 'shimmer'], label='Voice', value='fable') | |
response_format = gr.Dropdown(choices=['mp3', 'opus', 'aac', 'flac'], label='Output format', value='mp3') | |
speed = gr.Slider(0.25, 4, value=1, step=.25, label="Speed", info="Speach speed (0.25x - 4x)") | |
with gr.Row(variant='panel'): | |
text = gr.Textbox(label="Input text", placeholder="Your input for Text-To-Speech - press 'Say it' or hit 'Enter' key when ready.") | |
with gr.Row(variant='panel'): | |
with gr.Column(scale=1): | |
gr.Markdown(" ") | |
with gr.Column(scale=1): | |
btn = gr.Button("Say it") | |
with gr.Column(scale=1): | |
gr.Markdown(" ") | |
output_audio = gr.Audio(label="Text-To-Speech Output", waveform_options={"waveform_progress_color" : "#ff584f"}) | |
text.submit(fn=generate_tts, inputs=[text, model, voice, api_key, response_format, speed], outputs=output_audio, api_name="tts_enter_key", concurrency_limit=None) | |
btn.click(fn=generate_tts, inputs=[text, model, voice, api_key, response_format, speed], outputs=output_audio, api_name="tts_button", concurrency_limit=None) | |
tts_demo.launch() |