Add error handling

app.py

@@ -11,6 +11,7 @@ from langchain_chroma import Chroma
 from langchain_openai import OpenAIEmbeddings
 from langchain_core.prompts import ChatPromptTemplate, MessagesPlaceholder, PromptTemplate
 from langchain_openai import ChatOpenAI
+from openai import RateLimitError
 from dotenv import load_dotenv
 import gradio as gr
 
@@ -80,9 +81,18 @@ def predict(message, history):
         chat_history.append(HumanMessage(content=m))
         chat_history.append(a)
 
-    ai_response = rag_chain.invoke({"input": message, "chat_history": chat_history})
-
-    return ai_response["answer"]
+    try:
+        ai_response = rag_chain.invoke({"input": message, "chat_history": chat_history})
+        answer = ai_response["answer"]
+    except RateLimitError as e:
+        if "You exceeded your current quota" in e.message:
+            answer = "This demo has ran out of credits! Come back again soon and try it out"
+        else:
+            answer = "There seems to be a problem with the OpenAI API. We'll look into it!"
+    except Exception as e:
+        answer = "There seems to be a problem with the OpenAI API. We'll look into it!"
+
+    return answer
 
 theme = gr.themes.Default().set(
     button_primary_background_fill_dark="#26C8A1",