Spaces:
Running
Running
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(auth=("os.environ['username']", "os.environ['password']")) | |
#demo.launch() | |