Added try except

app.py

@@ -2,15 +2,28 @@ import os
 import openai
 import gradio as gr
 
-openai.api_key = os.environ["OPENAPI_API_KEY"]
+try:
+    openai.api_key = os.environ["OPENAPI_API_KEY"]
+except KeyError:
+    error_message = "System is at capacity right now.Please try again later"
+    print(error_message)
+    def chatbot(input):
+        return error_message
+else:
+    messages = [
+        {"role": "system", "content": "My AI Assistant"},
+    ]
 
 def chatbot(input):
-    response = openai.Moderation.create(input=input)
-    reply = response["results"][0].flagged
-    if reply == True:
-        return "This content is offensive and needs to be moderated"
-    else:
-        return "This content doesn't need moderation"
+    try:
+        response = openai.Moderation.create(input=input)
+        reply = response["results"][0].flagged
+        if reply == True:
+            return "This content is offensive and needs to be moderated"
+        else:
+            return "This content doesn't need moderation"
+    except openai.error.OpenAIError as e:
+        return "System is at capacity right now.Please try again later"
 
 inputs = gr.inputs.Textbox(lines=7, label="Query")
 outputs = gr.outputs.Textbox(label="Response")