Update app

app.py

@@ -47,7 +47,7 @@ def load_and_process_file(file_data):
         chunk_overlap=200,
     )
     chunks = text_splitter.split_documents(documents)
-    embeddings = OpenAIEmbeddings(openai_api_key=st.secrets.openai_api_key)
+    embeddings = OpenAIEmbeddings(openai_api_key=st.secrets.OPENAI_API_KEY)
     vector_store = Chroma.from_documents(chunks, embeddings)
     return vector_store
 
@@ -60,7 +60,7 @@ def initialize_chat_model(vector_store):
     llm = ChatOpenAI(
         model="gpt-3.5-turbo",
         temperature=0,
-        openai_api_key=st.secrets.openai_api_key,
+        openai_api_key=st.secrets.OPENAI_API_KEY,
     )
     retriever = vector_store.as_retriever()
     return ConversationalRetrievalChain.from_llm(llm, retriever)
@@ -72,12 +72,12 @@ def main():
     """
 
     if "openai_api_key" in st.secrets:
-        openai_api_key = st.secrets.openai_api_key
+        openai_api_key = st.secrets.OPENAI_API_KEY
     else:
         openai_api_key = st.sidebar.text_input("OpenAI API Key", type="password")
-        st.secrets.openai_api_key = openai_api_key
+        st.secrets.OPENAI_API_KEY = openai_api_key
 
-        if not st.secrets.openai_api_key:
+        if not st.secrets.OPENAI_API_KEY:
             st.info("Please add your OpenAI API key to continue.")
 
     assistant_message = "Hello, you can upload a document and chat with me to ask questions related to its content. Start by adding OpenAI API Key in the sidebar."
@@ -89,7 +89,7 @@ def main():
 
     if prompt := st.chat_input(
         placeholder="Chat with your document",
-        disabled=(not st.secrets.openai_api_key),
+        disabled=(not st.secrets.OPENAI_API_KEY),
     ):
         st.session_state.messages.append(User(message=prompt).build_message())
         st.chat_message(ChatProfileRoleEnum.User).write(prompt)
@@ -121,7 +121,7 @@ def handle_question(question):
     with st.chat_message(ChatProfileRoleEnum.Assistant):
         stream_handler = StreamHandler(st.empty())
         llm = ChatOpenAI(
-            openai_api_key=st.secrets.openai_api_key,
+            openai_api_key=st.secrets.OPENAI_API_KEY,
             streaming=True,
             callbacks=[stream_handler],
         )
@@ -161,9 +161,9 @@ def build_sidebar():
 
         add_file = st.button(
             "Process File",
-            disabled=(not uploaded_file and not st.secrets.openai_api_key),
+            disabled=(not uploaded_file and not st.secrets.OPENAI_API_KEY),
         )
-        if add_file and uploaded_file and st.secrets.openai_api_key.startswith("sk-"):
+        if add_file and uploaded_file and st.secrets.OPENAI_API_KEY.startswith("sk-"):
             with st.spinner("💭 Thinking..."):
                 vector_store = load_and_process_file(uploaded_file)