added try except

app.py

@@ -2,25 +2,31 @@ 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": "My AI Assistant"},
-]
-
-def chatbot(input):
-    if input:
-        messages.append({"role": "user", "content": input})
-        chat = openai.ChatCompletion.create(
-            model="gpt-3.5-turbo", messages=messages
-        )
-        reply = chat.choices[0].message.content
-        messages.append({"role": "assistant", "content": reply})
-        return reply
-
-inputs = gr.inputs.Textbox(lines=7, label="Query")
-outputs = gr.outputs.Textbox(label="Response")
-
-gr.Interface(fn=chatbot, inputs=inputs, outputs=outputs, title="",
-             theme="compact").launch()
+    def chatbot(input):
+        if input:
+            messages.append({"role": "user", "content": input})
+            chat = openai.ChatCompletion.create(
+                model="gpt-3.5-turbo", messages=messages
+            )
+            reply = chat.choices[0].message.content
+            messages.append({"role": "assistant", "content": reply})
+            return reply
 
+gr.Interface(
+    fn=chatbot,
+    inputs=gr.inputs.Textbox(lines=7, label="Query"),
+    outputs=gr.outputs.Textbox(label="Response"),
+    theme="compact"
+).launch()