Update app.py

app.py

@@ -46,6 +46,10 @@ def read_docx(file):
 
 @st.cache_resource(show_spinner=False)
 def load_data(uploaded_files):
+    if not st.session_state.openai_api_key:
+        st.error("No OpenAI API key provided. Please enter a valid API key.")
+        return None
+
     with st.spinner("Loading and indexing the documents – hang tight! This should take 1-2 minutes."):
         docs = []
         for uploaded_file in uploaded_files:
@@ -56,7 +60,9 @@ def load_data(uploaded_files):
                 text = read_docx(uploaded_file)
                 docs.append(Document(text=text))
 
+        # Set the OpenAI API key
         Settings.llm = OpenAI(model="gpt-3.5-turbo", temperature=0.5, 
+                              api_key=st.session_state.openai_api_key,
                               system_prompt="You are an expert on the Streamlit Python library and your job is to answer technical questions. Assume that all questions are related to the Streamlit Python library. Keep your answers technical and based on facts – do not hallucinate features.")
         
         index = VectorStoreIndex.from_documents(docs, settings=Settings.llm)
@@ -90,43 +96,44 @@ uploaded_files = st.file_uploader("Upload PDF or DOCX files", type=["pdf", "docx
 
 if uploaded_files and st.session_state.openai_api_key:
     index = load_data(uploaded_files)
-    chat_engine = index.as_chat_engine(chat_mode="condense_question", verbose=True)
-
-    # User input for questions
-    if prompt := st.chat_input("Your question"):
-        st.session_state.messages.append({"role": "user", "content": prompt})
-
-    for message in st.session_state.messages:
-        with st.chat_message(message["role"]):
-            st.write(message["content"])
-
-    if len(st.session_state.messages) > 0 and st.session_state.messages[-1]["role"] != "assistant":
-        with st.chat_message("assistant"):
-            with st.spinner("Thinking..."):
-                response = chat_engine.chat(prompt)
-                st.write(response.response)
-                message = {"role": "assistant", "content": response.response}
-                st.session_state.messages.append(message)
-
-    if st.button("Save Conversation"):
-        if st.session_state.messages:
-            st.session_state.confirm_save = True
-
-    if st.session_state.get('confirm_save', False):
-        st.warning("Do you want to save the conversation?")
-        col1, col2 = st.columns(2)
-        with col1:
-            if st.button("Yes"):
-                save_conversation()
-                st.success("Conversation saved!")
-                st.session_state.confirm_save = False
-        with col2:
-            if st.button("No"):
-                st.session_state.confirm_save = False
-
-    if st.button("End Conversation"):
-        st.session_state.messages = []
-        st.success("Conversation ended. You can start a new one!")
+    if index:  # Ensure index is not None
+        chat_engine = index.as_chat_engine(chat_mode="condense_question", verbose=True)
+
+        # User input for questions
+        if prompt := st.chat_input("Your question"):
+            st.session_state.messages.append({"role": "user", "content": prompt})
+
+        for message in st.session_state.messages:
+            with st.chat_message(message["role"]):
+                st.write(message["content"])
+
+        if len(st.session_state.messages) > 0 and st.session_state.messages[-1]["role"] != "assistant":
+            with st.chat_message("assistant"):
+                with st.spinner("Thinking..."):
+                    response = chat_engine.chat(prompt)
+                    st.write(response.response)
+                    message = {"role": "assistant", "content": response.response}
+                    st.session_state.messages.append(message)
+
+        if st.button("Save Conversation"):
+            if st.session_state.messages:
+                st.session_state.confirm_save = True
+
+        if st.session_state.get('confirm_save', False):
+            st.warning("Do you want to save the conversation?")
+            col1, col2 = st.columns(2)
+            with col1:
+                if st.button("Yes"):
+                    save_conversation()
+                    st.success("Conversation saved!")
+                    st.session_state.confirm_save = False
+            with col2:
+                if st.button("No"):
+                    st.session_state.confirm_save = False
+
+        if st.button("End Conversation"):
+            st.session_state.messages = []
+            st.success("Conversation ended. You can start a new one!")
 
 else:
     st.sidebar.warning("Please enter your OpenAI API key and upload PDF or DOCX files to proceed.")
@@ -162,4 +169,4 @@ if st.session_state.show_conversations:
     else:
         st.sidebar.write("No previous conversations found.")
 else:
-    st.sidebar.write("Previous conversations are hidden. Click 'Toggle Previous Conversations' to show.")
+    st.sidebar.write("Previous conversations are hidden. Click 'Toggle Previous Conversations' to show.")