gpt_api / app.py
jozzy's picture
Update app.py
39874b2
raw
history blame
1.55 kB
import os
import gradio as gr
import openai
openai.api_key = os.environ['OPENAI_API_KEY']
messages = [{"role": "system", "content": 'You are a helpful technology assistant.'}]
def audioGPT(audio):
global messages
audio_file = open(audio, "rb")
transcript = openai.Audio.transcribe("whisper-1", audio_file)
messages.append({"role": "user", "content": transcript["text"]})
response = openai.ChatCompletion.create(model="gpt-3.5-turbo", messages=messages)
system_message = response["choices"][0]["message"]
messages.append(system_message)
chats = ""
for msg in messages:
if msg['role'] != 'system':
chats += msg['role'] + ": " + msg['content'] + "\n\n"
return chats
def textGPT(text):
global messages
messages.append({"role": "user", "content": text})
response = openai.ChatCompletion.create(model="gpt-3.5-turbo", messages=messages)
system_message = response["choices"][0]["message"]
messages.append(system_message)
chats = ""
for msg in messages:
if msg['role'] != 'system':
chats += msg['role'] + ": " + msg['content'] + "\n\n"
return chats
text = gr.Interface(fn=textGPT, inputs="text", outputs="text")
audio = gr.Interface(fn=audioGPT, inputs=gr.Audio(source="microphone", type="filepath"), outputs="text")
demo = gr.TabbedInterface([text, audio], [ "chatGPT", "audioGPT"])
if __name__ == "__main__":
demo.launch(share=True, auth=("os.environ['username']", "os.environ['password']"))
#demo.launch()