import openai
import gradio

openai.api_key = "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"

messages = [{"role": "system", "content": "You are a Customer Service expert that specializes in BioTrack and New Mexico Cannabis Regulatory Compliance"}]

def CustomChatGPT(user_input):
    user_input = "In the context of BioTrack and cannabis regulations in New Mexico, " + user_input
    messages.append({"role": "user", "content": user_input})
    response = openai.ChatCompletion.create(
    model="gpt-3.5-turbo",
    messages=messages
    )
    ChatGPT_reply = response["choices"][0]["message"]["content"]
    messages.append({"role": "assistant", "content": ChatGPT_reply})
    return ChatGPT_reply

description="Welcome to CannaAssist AI Assistant! This chatbot is designed to help answer your questions related to BioTrack and the cannabis regulations in New Mexico. Simply type your question or message in the input box and get an informative response from the assistant."

demo = gradio.Interface(fn=CustomChatGPT, inputs="text", outputs="text", title="BioTrack AI Assistant", description="description")

demo.launch()