app.py

import openai
import gradio as gr

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

iface = gr.Interface(
    fn=CustomChatGPT, 
    inputs=gr.inputs.Textbox(lines=5, placeholder='Type here...', label='Your Question'), 
    outputs=gr.outputs.Textbox(label='AI Response'), 
    title="CannaAssist AI Assistant",
    description="Welcome to the CannaAssist AI Assistant. This tool is designed to provide expert guidance on BioTrack and cannabis regulations in New Mexico. Feel free to ask any questions related to these topics. DISCLAIMER: This is a proof of concept and not an official product.",
    examples=[
        ["How to add wholesale flower into my inventory?"],
        ["How to check in a customer?"],
        ["How to generate a manifest?"]
    ],
    theme="huggingface"
)

iface.launch()