Update app.py

app.py

@@ -1,7 +1,17 @@
+import os
+from getpass import getpass
 import openai
 import gradio as gr
+from huggingface_hub import HfApi
 
-openai.api_key = "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"
+# Get API key from Hugging Face secret repository
+# Make sure to replace 'CannaTech' and 'OPENAI_API_KEY' with your Hugging Face username and the name of your secret repository
+# You will be prompted to enter your Hugging Face password
+api = HfApi()
+password = getpass("Enter your Hugging Face password: ")
+token = api.login(username="CannaTech", password=password)
+api.create_repo(token=token, name="OPENAI_API_KEY", exist_ok=True)
+openai.api_key = api.get_secret(token=token, repo_id="CannaTech/OPENAI_API_KEY", secret_name="openai-api-key")
 
 messages = [{"role": "system", "content": "You are an expert in Technical Support and Customer Service that specializes in New Mexico Cannabis Regulatory Compliance and training people how to use software called BioTrack"}]
 
@@ -16,19 +26,45 @@ def CustomChatGPT(category, user_input):
     messages.append({"role": "assistant", "content": ChatGPT_reply})
     return ChatGPT_reply
 
+examples = [
+    ["BioTrack", "What is the process to register for BioTrack in New Mexico?"],
+    ["BioTrack", "How to add a new product in BioTrack?"],
+    ["BioTrack", "What are the steps to update inventory in BioTrack?"],
+    ["BioTrack", "How to generate sales reports in BioTrack?"],
+    ["BioTrack", "Can I integrate BioTrack with my existing POS system?"],
+    ["BioTrack", "What are the data backup options in BioTrack?"],
+    ["BioTrack", "How to handle product returns in BioTrack?"],
+    ["BioTrack", "What are the security features of BioTrack?"],
+    ["BioTrack", "How to train my staff to use BioTrack?"],
+    ["BioTrack", "What are the system requirements to run BioTrack?"],
+    ["Regulations", "What is the legal age to purchase cannabis in New Mexico?"],
+    ["Regulations", "What are the packaging and labeling requirements for cannabis in New Mexico?"],
+    ["Regulations", "What are the limits on cannabis possession in New Mexico?"],
+    ["Regulations", "Can I grow my own cannabis in New Mexico?"],
+    ["Regulations", "What are the regulations for cannabis edibles in New Mexico?"],
+    ["Regulations", "What are the licensing requirements for opening a dispensary in New Mexico?"],
+    ["Regulations", "What are the regulations for cannabis advertising in New Mexico?"],
+    ["Regulations", "What are the penalties for non-compliance with cannabis regulations in New Mexico?"],
+    ["Regulations", "Can I use medical cannabis in public in New Mexico?"],
+    ["Regulations", "What are the regulations for transporting cannabis in New Mexico?"],
+    ["BioTrack", "How to report a lost or stolen product in BioTrack?"],
+    ["Regulations", "What are the regulations for cannabis waste disposal in New Mexico?"],
+    ["BioTrack", "How to handle    customer complaints in BioTrack?"],
+    ["Regulations", "What are the regulations for cannabis testing in New Mexico?"],
+    ["BioTrack", "How to manage multiple locations in BioTrack?"]
+]
+
 iface = gr.Interface(
     fn=CustomChatGPT, 
-    inputs=[gr.inputs.Dropdown(choices=['BioTrack', 'Regulations', 'Best Practices', 'General Question'], label='Category'), gr.inputs.Textbox(lines=1, placeholder='Type here...', label='Your Question')], 
+    inputs=[
+        gr.inputs.Dropdown(choices=['BioTrack', 'Regulations', 'Best Practices', 'General Question'], label='Category', type='value'),
+        gr.inputs.Textbox(lines=1, 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=[
-        ["BioTrack", "How to add wholesale flower into my inventory?"],
-        ["Regulations", "How much cannabis can I buy in New Mexico."],
-        ["Best Practices", "What are the best practices for managing inventory?"],
-        ["General Question", "How to increase sales for my dispensary?"]
-    ],
-    theme="huggingface"
+    examples=examples,
+    theme="upscale"
 )
 
 iface.launch()