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)