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)