Spaces:
Runtime error
Runtime error
| # Import the necessary libraries | |
| import os | |
| import openai | |
| import gradio as gr | |
| # Set the OpenAI API key | |
| openai.api_key = os.getenv("OPENAI_API_KEY") | |
| # Define the authentication function | |
| def check_auth(username, password): | |
| # Define valid username-password pairs | |
| valid_credentials = [ | |
| ("user1", "password1"), | |
| ("user2", "password2"), | |
| ("user3", "password3"), | |
| ("user4", "password4"), | |
| ("user5", "password5") | |
| ] | |
| # 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='AI Response'), | |
| 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=[...], # List of example inputs | |
| theme=gr.themes.Monochrome(), | |
| examples_per_page=5, | |
| cache_examples=False, # Turn off example caching | |
| article="""CannaTech Solutions - CannaAssist AI Assistant...""", # 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 | |
| ) | |
| # Launch the interface with authentication | |
| iface.launch(auth_message="WARNING: UNAUTHORIZED ACCESS OR USE OF THIS CLOSED BETA IS STRICTLY PROHIBITED",auth=check_auth) | |