Added try except

app.py

@@ -3,22 +3,35 @@ 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"},
+    ]
 
 messages = [
     {"role": "system", "content": ""},
 ]
 
 def chatbot(input):
-    if input:
-        messages.append({"role": "user", "content": input})
-        prompt = "\n".join([f"{m['role']}: {m['content']}" for m in messages])
-        chat = openai.Completion.create(
-            engine="text-davinci-003", prompt=prompt, max_tokens=1024, n=1, stop=None, temperature=0.7
-        )
-        reply = chat.choices[0].text.strip()
-        messages.append({"role": "assistant", "content": reply})
-        return reply
+    try:
+        if input:
+            messages.append({"role": "user", "content": input})
+            prompt = "\n".join([f"{m['role']}: {m['content']}" for m in messages])
+            chat = openai.Completion.create(
+                engine="text-davinci-003", prompt=prompt, max_tokens=1024, n=1, stop=None, temperature=0.7
+            )
+            reply = chat.choices[0].text.strip()
+            messages.append({"role": "assistant", "content": reply})
+            return reply
+    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")