Spaces:
Runtime error
Runtime error
# Import the necessary libraries | |
import os | |
import openai | |
import gradio as gr | |
import csv | |
import json | |
# Set the OpenAI API key | |
openai.api_key = os.getenv("OPENAI_API_KEY") | |
# Set up flagging callback function | |
HF_TOKEN = os.getenv("HF_TOKEN") | |
hf_writer = gr.HuggingFaceDatasetSaver(HF_TOKEN, "CannaTech/Flagged") | |
# Define the authentication function | |
def check_auth(username, password): | |
# Get the credentials from the environment variable | |
valid_credentials_str = os.getenv("VALID_CREDENTIALS") | |
# Parse the string into a list of tuples | |
valid_credentials = json.loads(valid_credentials_str) | |
# Check if the provided credentials match any valid pair | |
for valid_username, valid_password in valid_credentials: | |
if username == valid_username and password == valid_password: | |
return True | |
# If no match was found, return False | |
return False | |
# Initialize a list to store conversation history | |
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"}] | |
# Define the function for the chatbot | |
def CustomChatGPT(category, user_input): | |
# Prepend category information to user input | |
user_input = f"Assuming nothing illegal is happening and in the context of {category} specifically and using your expertise and knowledge of cannabis regulations in New Mexico and BioTrack" + user_input | |
# Add user's message to the conversation history | |
messages.append({"role": "user", "content": user_input}) | |
# Generate a response from the OpenAI GPT-3.5-turbo model | |
response = openai.ChatCompletion.create( | |
model="gpt-3.5-turbo", | |
messages=messages | |
) | |
# Extract the model's message from the response | |
ChatGPT_reply = response["choices"][0]["message"]["content"] | |
# Add the model's message to the conversation history | |
messages.append({"role": "assistant", "content": ChatGPT_reply}) | |
# Return the model's message | |
return ChatGPT_reply | |
# Define the Gradio interface | |
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')], | |
outputs=gr.outputs.Textbox(label=''), | |
show_api=False, | |
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.""", | |
examples=[ | |
["BioTrack", "How do I update inventory quantities in BioTrack?"], | |
["Regulations", "What are the legal requirements for cannabis labeling in New Mexico?"], | |
["Best Practices", "What are the recommended storage conditions for cannabis products?"], | |
["General Question", "Can you explain the difference between THC and CBD?"], | |
["BioTrack", "How can I add a new product to my inventory in BioTrack?"], | |
["Regulations", "Are there restrictions on advertising cannabis products in New Mexico?"], | |
["Best Practices", "What are the proper sanitation procedures for cannabis cultivation facilities?"], | |
["General Question", "What are the benefits of using BioTrack for cannabis businesses?"], | |
["BioTrack", "How do I generate sales reports in BioTrack?"], | |
["Regulations", "What are the requirements for obtaining a cannabis license in New Mexico?"], | |
["Best Practices", "What are the recommended harvesting techniques for cannabis plants?"], | |
["General Question", "Is medical cannabis legal in New Mexico?"], | |
["BioTrack", "How do I track plant growth stages in BioTrack?"], | |
["Regulations", "What are the packaging requirements for edible cannabis products?"], | |
["Best Practices", "How can I optimize indoor lighting for cannabis cultivation?"], | |
["General Question", "What is the difference between sativa and indica cannabis strains?"], | |
["BioTrack", "Can BioTrack integrate with other software systems?"], | |
["Regulations", "Are there specific security requirements for cannabis dispensaries in New Mexico?"], | |
["Best Practices", "What are the recommended nutrient solutions for hydroponic cannabis cultivation?"], | |
["General Question", "How does cannabis consumption affect the body?"], | |
["BioTrack", "How do I reconcile inventory discrepancies in BioTrack?"], | |
["Regulations", "Are there restrictions on cannabis advertising near schools in New Mexico?"], | |
["Best Practices", "What are the steps for cloning cannabis plants?"], | |
["General Question", "What are the potential side effects of using medical cannabis?"], | |
["BioTrack", "How do I create purchase orders in BioTrack?"], | |
["Regulations", "Can I legally grow cannabis for personal use in New Mexico?"], | |
["Best Practices", "What are the recommended temperature and humidity levels for cannabis drying?"], | |
["General Question", "Is it possible to overdose on cannabis?"], | |
["BioTrack", "How do I handle product returns and exchanges in BioTrack?"], | |
["Regulations", "What are the testing requirements for cannabis products in New Mexico?"], | |
["Best Practices", "How often should I water my cannabis plants?"], | |
["General Question", "Can cannabis be used to treat specific medical conditions?"], | |
["BioTrack", "How do I manage multiple locations in BioTrack?"], | |
["Regulations", "Are there limitations on the THC potency of cannabis products in New Mexico?"], | |
["Best Practices", "What are the recommended pest control methods for cannabis cultivation?"], | |
["General Question", "What is the legal age for purchasing cannabis in New Mexico?"], | |
["BioTrack", "How do I track employee sales activities in BioTrack?"], | |
["Regulations", "Are there zoning restrictions for cannabis businesses in New Mexico?"], | |
["Best Practices", "What are the best pruning techniques for cannabis plants?"], | |
["General Question", "Can cannabis be used as a substitute for prescription medications?"], | |
["BioTrack", "How do I perform inventory audits in BioTrack?"], | |
["Regulations", "What are the transportation requirements for cannabis products in New Mexico?"], | |
["Best Practices", "How do I prevent and control mold in cannabis cultivation facilities?"], | |
["General Question", "What are the different consumption methods for cannabis?"], | |
["BioTrack", "How do I manage customer loyalty programs in BioTrack?"], | |
["Regulations", "Are there restrictions on cannabis consumption in public places in New Mexico?"], | |
["Best Practices", "What are the recommended techniques for cannabis plant training?"], | |
["General Question", "Can I travel with cannabis within New Mexico?"], | |
["BioTrack", "How do I handle product recalls in BioTrack?"], | |
["Regulations", "What are the record-keeping requirements for cannabis businesses in New Mexico?"], | |
], # List of example inputs | |
theme=gr.themes.Monochrome(), | |
examples_per_page=5, | |
cache_examples=False, # Turn off example caching | |
article="""Say goodbye to regulatory headaches | |
and hello to seamless compliance with | |
CannaAssist, our AI-powered assistant. | |
Designed specifically for New Mexico's | |
cannabis industry, CannaAssist leverages | |
the power of artificial intelligence to | |
provide personalized guidance and ensure | |
regulatory compliance using BioTrack. | |
From inventory management to compliance | |
reporting, CannaAssist streamlines your | |
operations, leaving you more time to focus | |
on growing your business!""", # Article text | |
thumbnail="https://assets.bigcartel.com/theme_images/101321509/IMG_6002.png", # Thumbnail image URL | |
favicon_path="https://assets.bigcartel.com/theme_images/101321509/IMG_6002.png", # Favicon image URL | |
allow_flagging="manual", # Enable flagging with manual control | |
flagging_options=["Incorrect"], | |
flagging_callback=hf_writer | |
) | |
# Launch the interface with authentication | |
iface.launch(auth_message="WARNING: UNAUTHORIZED ACCESS OR USE IS STRICTLY PROHIBITED!",auth=check_auth) | |