File size: 835 Bytes
5f6834d
 
 
 
 
 
 
 
70b765e
5f6834d
 
70b765e
 
5f6834d
 
 
 
 
70b765e
5f6834d
56fe814
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
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

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

demo.launch()