Update app.py

app.py

@@ -1,25 +1,20 @@
+import os
 import openai
 import gradio as gr
-import os
 
 openai.api_key = os.getenv("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"}]
 
 def CustomChatGPT(category, user_input):
-    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"}]
-    user_input = f"In the context of {category} specifically and using your expertise and knowledge of cannabis regulations in New Mexico and BioTrack, " + user_input
+    user_input = f"In the context of {category} specifically and using your expertise and knowledge of cannabis regulations in New Mexico and BioTrack" + user_input
     messages.append({"role": "user", "content": user_input})
-    try:
-        response = openai.ChatCompletion.create(
-            model="gpt-3.5-turbo",
-            messages=messages
-        )
-        ChatGPT_reply = response.choices[0].message["content"]
-    except openai.api_resources.abstract.api_error.APIError as e:
-        if e.type == 'rate_limit':
-            ChatGPT_reply = "Rate limit exceeded. Please try again later."
-        else:
-            ChatGPT_reply = "An error occurred. Please try again later."
+    response = openai.ChatCompletion.create(
+    model="gpt-3.5-turbo",
+    messages=messages
+    )
+    ChatGPT_reply = response["choices"][0]["message"]["content"]
+    messages.append({"role": "assistant", "content": ChatGPT_reply})
     return ChatGPT_reply
 
 iface = gr.Interface(
@@ -30,10 +25,11 @@ iface = gr.Interface(
     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?"],
+        ["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?"],
+        ["General Question", "How to increase sales for my dispensary?"]
     ],
-    theme=gr.themes.Monochrome()
+    theme="upscale"
 )
 
+iface.launch()