Update app.py

app.py

@@ -4,17 +4,32 @@ import openai
 import whisper
 from gtts import gTTS
 from IPython.display import Audio
+
 model = whisper.load_model("base")
 
-openai.api_key = "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"
+mes1 = [
+    {"role": "system", "content": "You are a TOEFL examiner who will ask me questions in a TOEFL test."}
+]
+
+mes2 = [
+    {"role": "system", "content": "You are a mental health therapist. Your name is Tina."}
+]
+
+mes3 = [
+    {"role": "system", "content": "You are my personal assistant. Your name is Alice."}
+]
 
 res = []
 
 question = []
 
-def transcribe(audio):
+def transcribe(apikey, audio, choice1, choice2):
+
+    #global messages
+
+    openai.api_key = apikey
     
-    #time.sleep(3)
+    # time.sleep(3)
     # load audio and pad/trim it to fit 30 seconds
     audio = whisper.load_audio(audio)
     audio = whisper.pad_or_trim(audio)
@@ -31,6 +46,13 @@ def transcribe(audio):
     result = whisper.decode(model, mel, options)
     res.append(result.text)
 
+    if choice1 == "TOEFL":
+      messages = mes1
+    elif choice1 == "Therapist":
+      messages = mes2
+    else:
+      messages = mes3
+
     # chatgpt
     n = len(res)
     content = res[n-1]
@@ -43,10 +65,12 @@ def transcribe(audio):
 
     chat_response = completion.choices[0].message.content
 
-    messages.append({"role": "assistant", "content": chat_response})
+    messages.append({"role": "assistant", "content": chat_response})   
 
-    tts = gTTS(chat_response, lang='en', tld='us')
-    #tts = gTTS(chat_response, lang='zh-CN')
+    if choice2 == "Chinese":
+      tts = gTTS(chat_response, lang='zh-CN')
+    elif choice2 == "English":
+      tts = gTTS(chat_response, lang='en', tld='us')
 
     tts.save('1.wav')
     sound_file = '1.wav'
@@ -61,11 +85,15 @@ output_3 = gr.Audio(label="Audio")
 
 gr.Interface(
     title = 'TalktoAI，随时随地，谈天说地！', 
+    description = "让具有人文关怀的AI造福每一个人！科技向善，AI璀璨！",
     fn=transcribe, 
     inputs=[
-        gr.inputs.Audio(source="microphone", type="filepath")
+        gr.Textbox(lines=1, label = "请填写您的OpenAI_API_key"),
+        gr.inputs.Audio(source="microphone", type="filepath"),
+        gr.Radio(["TOEFL", "Therapist", "Alice"], label="TOEFL Examiner, Therapist Tina, or Assistant Alice?"),
+        gr.Radio(["Chinese", "English"], label="Chinese or English?")
     ],
     outputs=[
       output_1, output_2, output_3
     ],
-    live=True).launch()
+    ).launch()