chat-bot-test / test.py
alliwene's picture
Upload folder using huggingface_hub
33bc31b
Raw
History Blame Contribute Delete
929 Bytes
import gradio as gr
import openai
from dotenv import load_dotenv
import os
load_dotenv()
openai.api_key = os.getenv("OPENAI_API_KEY")
messages = [{"role": "system", "content": "You are a teacher"}]
def transcribe(audio):
global messages
file = open(audio, "rb")
transcription = openai.Audio.transcribe("whisper-1", file)
print(transcription)
messages.append({"role": "user", "content": transcription["text"]})
response = openai.ChatCompletion.create(model="gpt-3.5-turbo", messages=messages)
AImessage = response["choices"][0]["message"]["content"]
messages.append({"role": "assistant", "content": AImessage})
chat = ""
for message in messages:
if message["role"] != "system":
chat += message["role"] + ":" + message["content"] + "\n\n"
return chat
ui = gr.Interface(
fn=transcribe, inputs=gr.Audio(type="filepath"), outputs="text"
)
ui.launch()