Update app.py

app.py

@@ -5,8 +5,8 @@ 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
+def CustomChatGPT(category, user_input):
+    user_input = f"In the context of BioTrack and cannabis regulations in New Mexico, and specifically about {category}, " + user_input
     messages.append({"role": "user", "content": user_input})
     response = openai.ChatCompletion.create(
     model="gpt-3.5-turbo",
@@ -18,17 +18,19 @@ def CustomChatGPT(user_input):
 
 iface = gr.Interface(
     fn=CustomChatGPT, 
-    inputs=gr.inputs.Textbox(lines=5, placeholder='Type here...', label='Your Question'), 
+    inputs=[gr.inputs.Dropdown(choices=['Biotrack', 'Regulations', 'Best Practices', 'General Question'], label='Category'), 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. 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?"]
+        ["Biotrack", "How to add wholesale flower into my inventory?"],
+        ["Regulations", "Tell me about cannabis regulations in New Mexico."],
+        ["Best Practices", "What are the best practices for managing inventory?"],
+        ["General Question", "What is BioTrack?"]
     ],
     theme="huggingface"
 )
 
 iface.launch()
 
+