ttanna's picture
Update main.py
cafefba
Raw
History Blame Contribute Delete
1.07 kB
import openai
import gradio as gr
from gradio.themes.base import Base
from gradio.themes.utils import colors, fonts, sizes
import os
openai.api_key = os.environ.get("openai.api_key")
messages = [
{"role": "system", "content": "You are a math teacher. Help with solution and explaination of the math problem that is asked. If it is not related to Math or you do not know the answer, say Sorry I am not sure of that one."},
]
def chatbot(input):
if input:
messages.append({"role": "assistant", "content": input})
chat = openai.ChatCompletion.create(
model="gpt-3.5-turbo", messages=messages
)
reply = chat.choices[0].message.content
messages.append({"role": "user", "content": reply})
return reply
inputs = gr.components.Textbox(lines=7,label="Type your question in the box below.")
outputs = gr.components.Textbox(lines=7, label="Reply")
gr.Interface(fn=chatbot, inputs=inputs, outputs=outputs, text_color="red",title="Solve Your Math Problems!",
theme="soft").launch(share=False)