jonathanagustin's picture
Upload folder using huggingface_hub
2449b0e verified
raw
history blame
6.21 kB
import gradio as gr
import tempfile
import openai
def tts(input_text: str, model: str, voice: str, api_key: str) -> str:
"""
Convert input text to speech using OpenAI's Text-to-Speech API.
:param input_text: The text to be converted to speech.
:type input_text: str
:param model: The model to use for synthesis (e.g., 'tts-1', 'tts-1-hd').
:type model: str
:param voice: The voice profile to use (e.g., 'alloy', 'echo', 'fable', etc.).
:type voice: str
:param api_key: OpenAI API key.
:type api_key: str
:return: File path to the generated audio file.
:rtype: str
:raises gr.Error: If input parameters are invalid or API call fails.
"""
if not input_text.strip():
raise gr.Error("Input text cannot be empty.")
if not api_key.strip():
raise gr.Error("API key is required.")
openai.api_key = api_key
try:
response = openai.Audio.create(
text=input_text,
voice=voice,
model=model
)
except openai.APITimeoutError as e:
raise gr.Error(f"OpenAI API request timed out: {e}")
except openai.APIConnectionError as e:
raise gr.Error(f"OpenAI API request failed to connect: {e}")
except openai.AuthenticationError as e:
raise gr.Error(f"OpenAI API request was not authorized: {e}")
except openai.PermissionDeniedError as e:
raise gr.Error(f"OpenAI API request was not permitted: {e}")
except openai.RateLimitError as e:
raise gr.Error(f"OpenAI API request exceeded rate limit: {e}")
except openai.BadRequestError as e:
raise gr.Error(f"OpenAI API bad request: {e}")
except openai.UnprocessableEntityError as e:
raise gr.Error(f"OpenAI API request could not be processed: {e}")
except openai.NotFoundError as e:
raise gr.Error(f"OpenAI API resource not found: {e}")
except openai.InternalServerError as e:
raise gr.Error(f"OpenAI API internal server error: {e}")
except openai.APIError as e:
# This will catch other API-related errors not captured above
raise gr.Error(f"OpenAI API Error: {e}")
except openai.OpenAIError as e:
# Catch-all for any other OpenAI exceptions
raise gr.Error(f"An OpenAI error occurred: {e}")
except Exception as e:
# Catch any other exceptions
raise gr.Error(f"An unexpected error occurred: {e}")
if not hasattr(response, 'audio'):
raise gr.Error("Invalid response from OpenAI API. The response does not contain audio content.")
with tempfile.NamedTemporaryFile(suffix=".mp3", delete=False) as temp_file:
temp_file.write(response.audio)
temp_file_path = temp_file.name
return temp_file_path
def main():
"""
Main function to create and launch the Gradio interface with input validation and error handling.
"""
MODEL_OPTIONS = ["tts-1", "tts-1-hd"]
VOICE_OPTIONS = ["alloy", "echo", "fable", "onyx", "nova", "shimmer"]
with gr.Blocks() as demo:
with gr.Row():
with gr.Column(scale=1):
api_key_input = gr.Textbox(
label="OpenAI API Key",
info="Get OpenAI key at: https://platform.openai.com/account/api-keys",
type="password",
placeholder="Enter your OpenAI API Key",
value="",
)
model_dropdown = gr.Dropdown(
choices=MODEL_OPTIONS, label="Model", value="tts-1"
)
voice_dropdown = gr.Dropdown(
choices=VOICE_OPTIONS, label="Voice Options", value="echo"
)
with gr.Column(scale=2):
input_textbox = gr.Textbox(
label="Input Text",
lines=10,
placeholder="Type your text here..."
)
submit_button = gr.Button(
"Convert Text to Speech",
variant="primary",
interactive=False # Initially disabled
)
with gr.Column(scale=1):
output_audio = gr.Audio(label="Output Audio")
# Define the event handler for the submit button with error handling
def on_submit(input_text, model, voice, api_key):
try:
audio_file = tts(input_text, model, voice, api_key)
return audio_file
except gr.Error as err:
# Re-raise gr.Error exceptions to display message without traceback
raise err
except Exception as e:
# Handle any other exceptions and display error message
raise gr.Error(f"An unexpected error occurred: {e}")
# Function to update the submit button state
def update_submit_button_state(api_key, input_text):
if api_key.strip() and input_text.strip():
return gr.update(interactive=True)
else:
return gr.update(interactive=False)
# Update the submit button state when the API key or input text changes
api_key_input.change(
fn=update_submit_button_state,
inputs=[api_key_input, input_textbox],
outputs=submit_button
)
input_textbox.change(
fn=update_submit_button_state,
inputs=[api_key_input, input_textbox],
outputs=submit_button
)
# Allow pressing Enter in the input textbox to trigger the conversion
input_textbox.submit(
fn=on_submit,
inputs=[input_textbox, model_dropdown, voice_dropdown, api_key_input],
outputs=output_audio,
api_name="tts",
)
# Trigger the conversion when the submit button is clicked
submit_button.click(
fn=on_submit,
inputs=[input_textbox, model_dropdown, voice_dropdown, api_key_input],
outputs=output_audio,
api_name="tts",
)
# Launch the Gradio app with error display enabled
demo.launch(show_error=True)
if __name__ == "__main__":
main()