Update app.py

app.py

@@ -9,11 +9,17 @@ 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
     messages.append({"role": "user", "content": user_input})
-    response = openai.ChatCompletion.create(
-    model="gpt-3.5-turbo",
-    messages=messages
-    )
-    ChatGPT_reply = response.choices[0].message["content"]
+    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."
     return ChatGPT_reply
 
 iface = gr.Interface(
@@ -24,17 +30,10 @@ 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?"],
-        ["BioTrack", "How do I update inventory quantities in BioTrack?"],
         ["Regulations", "How much cannabis can I buy in New Mexico?"],
-        ["Regulations", "What are the packaging requirements for cannabis products in New Mexico?"],
         ["Best Practices", "What are the best practices for managing inventory?"],
-        ["Best Practices", "How can I improve my dispensary's customer service?"],
         ["General Question", "How to increase sales for my dispensary?"],
-        ["General Question", "What are the benefits of offering delivery service for my dispensary?"],
-    
     ],
     theme=gr.themes.Monochrome()
 )
 
-iface.launch()
-