nehalelkaref's picture
Update app.py
d1446ae verified
raw
history blame contribute delete
No virus
1.15 kB
import gradio as gr
from openai import OpenAI
import os
key= os.environ.get('key')
client = OpenAI(api_key=key)
class english_buddy:
def __init__(self):
system_message = "Your name is Tom and I'm practising English with you. Be friendly, converse with me and advise me if my grammar is wrong"
self.messages = []
self.messages.append({'role':'system','content':system_message})
def __call__(self, user_input):
self.messages.append({'role':'user',
'content': user_input})
response = client.chat.completions.create(model='gpt-3.5-turbo',
messages=self.messages)
resonse_message = response.choices[0].message.content
self.messages.append({'role':'assistant',
'content': resonse_message})
return resonse_message
my_english_buddy = english_buddy()
iface = gr.Interface(fn=my_english_buddy, inputs="textbox", outputs="text",
clear_btn="Clear", theme="gradio/monochrome",
title="Chat with Tom and practice your English!",)
iface.launch(show_api=False)