Update app.py

app.py

@@ -1,14 +1,18 @@
 import logging
 from typing import Optional
 
-import openai
 import streamlit as st
 from langchain_core.messages import AIMessage, HumanMessage
-from openai import OpenAI, OpenAIError
-
-from rag.runnable_and_memory import get_runnable_and_memory
 from utils.error_message_template import ERROR_MESSAGE
 
+try:
+    import openai
+    from openai import OpenAI, OpenAIError
+    from rag.runnable_and_memory import get_runnable_and_memory
+except OpenAIError:
+    handle_errors()
+    st.stop()
+    
 logging.basicConfig(level=logging.ERROR)
 
 # Streamlit page configuration
@@ -33,7 +37,7 @@ def initialize_session_state():
             )
             # Clear the memory
             st.session_state["memory"].clear()
-        except openai.AuthenticationError:
+        except openai.OpenAIError:
             handle_errors()
         except Exception:
             handle_errors()