-
Notifications
You must be signed in to change notification settings - Fork 2
/
Copy paththerapist.py
28 lines (19 loc) · 950 Bytes
/
therapist.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
import gradio as gr
import openai, config, subprocess
openai.api_key = config.OPENAI_API_KEY
messages = [{"role": "system", "content": 'You are a music recommender, please respond with a list of 5 songs based on my heart rate in bpm which is currently '}]
def transcribe(heartrate可加中文):
global messages
transcript = heartrate可加中文
messages.append({"role": "user", "content": heartrate可加中文})
response = openai.ChatCompletion.create(model="gpt-4", messages=messages)
system_message = response["choices"][0]["message"]
messages.append(system_message)
subprocess.call(["say", system_message['content']])
chat_transcript = ""
for message in messages:
if message['role'] != 'system':
chat_transcript += message['role'] + ": " + message['content'] + "\n\n"
return chat_transcript
ui = gr.Interface(fn=transcribe, inputs="text", outputs="text").launch()
ui.launch()