Update app.py

app.py

@@ -5,13 +5,14 @@ import requests
 from pydub import AudioSegment as am
 from xml.etree import ElementTree
 
-api_base = "https://mvp-azureopenai.openai.azure.com/"
-api_key = os.getenv("OPENAI_API_KEY")
+aoai_url = "https://mvp-azureopenai.openai.azure.com/"
+aoai_key = ""
+stts_key = ""
 
 openai.api_type = "azure"
-openai.api_base = api_base
+openai.api_base = "https://mvp-azureopenai.openai.azure.com/"
 openai.api_version = "2023-03-15-preview"
-openai.api_key = api_key
+# openai.api_key = aoai_key
 
 messages_gpt = []
 messages_chat = [
@@ -23,9 +24,21 @@ messages_vchat = [
  {"role": "system", "content": "You are an AI assistant that helps people find information and just response with SSML."},
 ]
 
+def get_aoai_key(txt_inpout):
+ if txt_inpout:
+ openai.api_key = txt_inpout
+ return gr.update(value=txt_inpout)
+
 with gr.Blocks() as page:
+ with gr.Row():
+ with gr.Column(scale=0.4):
+ gr.Markdown("Your Key will not be saved or viewed by anyone.")
+ with gr.Column(scale=0.6):
+ ui_aoai_key = gr.Textbox(placeholder="Please enter your Azure OpenAI API key here.", lines=1, show_label=False, type='password').style(container=False)
+ ui_aoai_key.submit(get_aoai_key, ui_aoai_key, ui_aoai_key)
+ 
  with gr.Tabs():
- with gr.TabItem("GPT Playgroud"):
+ with gr.TabItem("GPT Playground"):
  ui_chatbot_gpt = gr.Chatbot(label="GPT Playground:")
  with gr.Row():
  with gr.Column(scale=0.9):
@@ -102,7 +115,7 @@ with gr.Blocks() as page:
  ui_prompt_sys = gr.Textbox(value="You are an AI assistant that helps people find information.", show_label=False, interactive=True).style(container=False)
  with gr.Row():
  ui_temp_chat = gr.Slider(0.1, 1.0, 0.7, step=0.1, label="Temperature", interactive=True)
- ui_max_tokens_chat = gr.Slider(100, 8000, 800, step=100, label="Max Tokens", interactive=True)
+ ui_max_tokens_chat = gr.Slider(100, 8000, 2000, step=100, label="Max Tokens", interactive=True)
  ui_top_p_chat = gr.Slider(0.05, 1.0, 0.9, step=0.1, label="Top P", interactive=True)
  with gr.Accordion("Select radio button to see detail:", open=False):
  ui_res_radio_chat = gr.Radio(["Response from OpenAI Model", "Prompt messages history"], label="Show OpenAI response:", interactive=True)
@@ -163,7 +176,7 @@ with gr.Blocks() as page:
  ui_clear_chat.click(lambda: None, None, ui_chatbot_chat, queue=False).then(reset_sys, ui_prompt_sys)
 
 
- with gr.TabItem("WALL·E 2"):
+ with gr.TabItem("DALL·E 2"):
  ui_prompt_walle = gr.Textbox(placeholder="Please enter your prompt here to generate image.", show_label=False).style(container=False)
  ui_image_walle = gr.Image()
  with gr.Accordion("Select radio button to see detail:", open=False): 
@@ -172,8 +185,8 @@ with gr.Blocks() as page:
  def get_image_walle(prompt_walle):
  global response_walle
  walle_api_version = '2022-08-03-preview'
- url = "{}dalle/text-to-image?api-version={}".format(api_base, walle_api_version)
- headers= { "api-key": api_key, "Content-Type": "application/json" }
+ url = "{}dalle/text-to-image?api-version={}".format(openai.api_base, walle_api_version)
+ headers= { "api-key": openai.api_key, "Content-Type": "application/json" }
  body = {
  "caption": prompt_walle,
  "resolution": "1024x1024"
@@ -204,6 +217,11 @@ with gr.Blocks() as page:
  with gr.Column():
  with gr.Accordion("Expand to config parameters:", open=False):
  ui_prompt_sys_vchat = gr.Textbox(value="You are an AI assistant that helps people find information and just response with SSML.", show_label=False, interactive=True).style(container=False)
+ ui_stts_key = gr.Textbox(placeholder="Input your Azure Speech service API key", show_label=False, interactive=True).style(container=False)
+ ui_stts_key.submit(lambda x: None, ui_stts_key)
+ stts_key = ui_aoai_key.value
+ print(stts_key)
+
  ui_voice_inc_vchat = gr.Audio(source="microphone", type="filepath")
  ui_voice_out_vchat = gr.Audio(value=None, type="filepath", interactive=False).style(container=False)
  with gr.Accordion("Expand to config parameters:", open=False):
@@ -238,7 +256,7 @@ with gr.Blocks() as page:
  voice_wav = voice_wav.set_frame_rate(16000)
  voice_wav.export(voice_message, format='wav')
  # STT
- OASK_Speech = os.getenv("OASK_Speech")
+ OASK = stts_key
  service_region = "westus"
 
  base_url = "https://"+service_region+".stt.speech.microsoft.com/"
@@ -267,7 +285,7 @@ with gr.Blocks() as page:
  return sst_text 
  
  def text_to_speech():
- OASK_Speech = os.getenv("OASK_Speech")
+ OASK_Speech = stts_key
  service_region = "westus"
 
  base_url = "https://"+service_region+".tts.speech.microsoft.com/"
@@ -329,7 +347,4 @@ with gr.Blocks() as page:
  bot_vchat, ui_chatbot_vchat, ui_chatbot_vchat, queue=False).then(text_to_speech, None, ui_voice_out_vchat)
 
 
-page.launch(share=False)
-
-
-
+page.launch(share=False)