Update app.py

app.py

@@ -7,8 +7,8 @@ import os
 os.system("pip install openai")
 import openai
 
-api_key = os.environ.get('api_key')
-openai.api_key = "youlou"
+#api_key = os.environ.get('api_key')
+
 
 whisper = gr.Interface.load(name="spaces/sanchit-gandhi/whisper-large-v2")
 
@@ -19,21 +19,21 @@ token = os.environ.get('HF_TOKEN')
 tts = gr.Interface.load(name="spaces/Flux9665/IMS-Toucan")
 talking_face = gr.Blocks.load(name="spaces/fffiloni/one-shot-talking-face", api_key=token)
 
-def infer(audio):
+def infer(audio, openai_api_key):
     
     whisper_result = whisper(audio, None, "translate", fn_index=0)
     
-    gpt_response = try_api(whisper_result)
+    gpt_response = try_api(whisper_result, openai_api_key)
     
     audio_response = tts(gpt_response[0], "English Text", "English Accent", "English Speaker's Voice", fn_index=0)
     
     portrait_link = talking_face("wise_woman_portrait.png", audio_response, fn_index=0)
     
-    return whisper_result, portrait_link, gr.update(visible=True)
+    return whisper_result, portrait_link, gpt_response[1], gr.update(visible=True)
 
-def try_api(message):
+def try_api(message, openai_api_key):
     try:
-        response = call_api(message)
+        response = call_api(message, openai_api_key)
         return response, "no error"
     except openai.error.Timeout as e:
         #Handle timeout error, e.g. retry or log
@@ -64,10 +64,12 @@ def try_api(message):
         print(f"OpenAI API request exceeded rate limit: {e}")
         return "oups", f"OpenAI API request exceeded rate limit: {e}"
 
-def call_api(message):
+def call_api(message, openai_api_key):
           
     print("starting open ai")
     
+    openai.api_key = openai_api_key
+    
     response = openai.Completion.create(
         model="text-davinci-003",
         prompt=message,
@@ -108,10 +110,12 @@ with gr.Blocks(css="style.css") as demo:
         gr.HTML(title)
         
         gpt_response = gr.Video(label="Talking Portrait response", elem_id="video_out")
+        error_handler = gr.Text()
              
     with gr.Column(elem_id="col-container-2"):
-          
-        record_input = gr.Audio(source="microphone",type="filepath", label="Audio input", show_label=True, elem_id="record_btn")
+        with gr.Row():
+            record_input = gr.Audio(source="microphone",type="filepath", label="Audio input", show_label=True, elem_id="record_btn")
+            openai_api_key = gr.Text(label="Your OpenAI API Key")
         whisper_tr = gr.Textbox(label="whisper english translation", elem_id="text_inp")
 
         send_btn = gr.Button("Send my request !")
@@ -121,7 +125,7 @@ with gr.Blocks(css="style.css") as demo:
                 loading_icon = gr.HTML(loading_icon_html)
                 share_button = gr.Button("Share to community", elem_id="share-btn")
     
-    send_btn.click(infer, inputs=[record_input], outputs=[whisper_tr, gpt_response, share_group])
+    send_btn.click(infer, inputs=[record_input, openai_api_key], outputs=[whisper_tr, gpt_response, error_handler, share_group])
     share_button.click(None, [], [], _js=share_js)
 
 demo.queue(max_size=32, concurrency_count=20).launch(debug=True)