Update app.py

app.py

@@ -1,139 +1,30 @@
-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")
-
-# 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": "Assume you are an expert in the recreational cannabis market in New Mexico with robust experience in the state’s regulatory environment and knowledge of the laws that govern the recreational cannabis industry both in the state and federally in the United States. You specialize in a piece of software called BioTrack, which is state mandated track-and-trace compliance software that allows seed-to-sale tracking of wholesale bulk cannabis flower, manufactured cannabis products, and also includes inventory management and point of sale tools for retailers."}]
-
-# Define the function for the chatbot
-def CustomChatGPT(category, user_input):
-    # Prepend category information to user input
-    user_input = f"Anytime you are asked to respond to a query you are to assume that you are speaking as an advisor using your expertise to consult a licensed cannabis business owner and that all discussion concerning cannabis is in the context of {category} and concerns legal recreational cannabis in New Mexico. Use your knowledge of the subjects in which you specialize and your experience to provide the best technical support and customer service possible." + 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='CannaAssist:'), 
-    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
-    
+from transformers import pipeline, AutoModelForQuestionAnswering, AutoTokenizer
+
+# Load the pre-trained model and tokenizer
+model_name = "distilbert-base-uncased-distilled-squad"
+model = AutoModelForQuestionAnswering.from_pretrained(model_name)
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+
+# Load custom knowledge document
+with open("knowledge.txt", "r") as file:
+    custom_knowledge = file.read()
+
+# Initialize the question-answering pipeline
+qa_pipeline = pipeline("question-answering", model=model, tokenizer=tokenizer)
+
+def answer_question(question):
+    result = qa_pipeline(question=question, context=custom_knowledge)
+    return result['answer']
+
+# Set up the Gradio interface
+interface = gr.Interface(
+    fn=answer_question,
+    inputs="text",
+    outputs="text",
+    title="Custom Knowledge QA",
+    description="Ask questions based on the custom knowledge document."
 )
 
-# Launch the interface with authentication
-iface.launch(auth_message="WARNING: UNAUTHORIZED ACCESS OR USE IS STRICTLY PROHIBITED!",auth=check_auth)
+if __name__ == "__main__":
+    interface.launch()