Update app.py

app.py

@@ -4,12 +4,14 @@ import tempfile
 from openai import OpenAI
 
 
-def tts(text, model, voice, api_key):
+def tts(text, model, voice, api_key, api_base):
     if api_key == '':
         raise gr.Error('Please enter your OpenAI API Key')
+    if api_base == '':
+        api_base = "https://api.openai.com/v1"
     else:
         try:
-            client = OpenAI(api_key=api_key)
+            client = OpenAI(api_key=api_key, api_base=api_base)
 
             response = client.audio.speech.create(
                 model=model, # "tts-1","tts-1-hd"
@@ -36,6 +38,7 @@ with gr.Blocks() as demo:
     gr.Markdown("# <center> OpenAI Text-To-Speech API with Gradio </center>")
     #gr.HTML("You can also access the Streaming demo for OpenAI TTS by clicking this <a href='https://huggingface.co/spaces/ysharma/OpenAI_TTS_Streaming'>Gradio demo link</a>")
     with gr.Row(variant='panel'):
+      api_base = gr.Textbox(type='text', label='OpenAI base URL', placeholder='Enter your base URL to access the TTS demo')
       api_key = gr.Textbox(type='password', label='OpenAI API Key', placeholder='Enter your API key to access the TTS demo')
       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 Options', value='alloy')
@@ -44,7 +47,7 @@ with gr.Blocks() as demo:
     btn = gr.Button("Text-To-Speech")
     output_audio = gr.Audio(label="Speech Output")
     
-    text.submit(fn=tts, inputs=[text, model, voice, api_key], outputs=output_audio, api_name="tts_enter_key", concurrency_limit=None)
-    btn.click(fn=tts, inputs=[text, model, voice, api_key], outputs=output_audio, api_name="tts_button", concurrency_limit=None)
+    text.submit(fn=tts, inputs=[text, model, voice, api_key, api_base], outputs=output_audio, api_name="tts_enter_key", concurrency_limit=None)
+    btn.click(fn=tts, inputs=[text, model, voice, api_key, api_base], outputs=output_audio, api_name="tts_button", concurrency_limit=None)
 
 demo.launch()