app updated

app.py

@@ -1,12 +1,23 @@
-import streamlit as st
+import os
+
 import openai
+import streamlit as st
+from llama_index.core import Settings, SimpleDirectoryReader, VectorStoreIndex
 from llama_index.llms.openai import OpenAI
-from llama_index.core import VectorStoreIndex, SimpleDirectoryReader, Settings
 
-st.set_page_config(page_title="Chat with the Streamlit docs, powered by LlamaIndex", page_icon="🦙", layout="centered", initial_sidebar_state="auto", menu_items=None)
-openai.api_key = st.secrets.openai_key
+st.set_page_config(
+    page_title="Chat with the Streamlit docs, powered by LlamaIndex",
+    page_icon="🦙",
+    layout="centered",
+    initial_sidebar_state="auto",
+    menu_items=None,
+)
+openai.api_key = os.environ["OPENAI_API_KEY"]
 st.title("Chat with the Streamlit docs, powered by LlamaIndex 💬🦙")
-st.info("Check out the full tutorial to build this app in our [blog post](https://blog.streamlit.io/build-a-chatbot-with-custom-data-sources-powered-by-llamaindex/)", icon="📃")
+st.info(
+    "Check out the full tutorial to build this app in our [blog post](https://blog.streamlit.io/build-a-chatbot-with-custom-data-sources-powered-by-llamaindex/)",
+    icon="📃",
+)
 
 if "messages" not in st.session_state.keys():  # Initialize the chat messages history
     st.session_state.messages = [
@@ -16,6 +27,7 @@ if "messages" not in st.session_state.keys():  # Initialize the chat messages hi
         }
     ]
 
+
 @st.cache_resource(show_spinner=False)
 def load_data():
     reader = SimpleDirectoryReader(input_dir="./data", recursive=True)
@@ -58,4 +70,4 @@ if st.session_state.messages[-1]["role"] != "assistant":
         st.write_stream(response_stream.response_gen)
         message = {"role": "assistant", "content": response_stream.response}
         # Add response to message history
-        st.session_state.messages.append(message)
+        st.session_state.messages.append(message)