Update app.py

app.py

@@ -34,6 +34,7 @@ with st.sidebar:
     add_vertical_space(3)  # Add more vertical space between text blocks
     st.write('Made with ❤️ by Anne')
 
+# User input for the OpenAI API key
 openai_api_key = st.text_input("Enter your OpenAI API key:")
 
 def load_pdf(file_path, openai_api_key):
@@ -85,7 +86,6 @@ def main():
     # Pass the openai_api_key parameter here
     VectorStore = load_pdf(pdf, openai_api_key)
 
-
     display_chat_history(st.session_state['chat_history'])
 
     st.write("<!-- Start Spacer -->", unsafe_allow_html=True)
@@ -105,8 +105,8 @@ def main():
             loading_message = st.empty()
             loading_message.text('Bot is thinking...')
 
-            VectorStore = load_pdf(pdf)
-            chain = load_chatbot()
+            VectorStore = load_pdf(pdf, openai_api_key)  # Pass the API key here
+            chain = load_chatbot(openai_api_key)  # Pass the API key here
             docs = VectorStore.similarity_search(query=st.session_state['processing_input'], k=3)
             with get_openai_callback() as cb:
                 response = chain.run(input_documents=docs, question=st.session_state['processing_input'])
@@ -120,11 +120,5 @@ def main():
         # Mark all messages as old after displaying
         st.session_state['chat_history'] = [(sender, msg, "old") for sender, msg, _ in st.session_state['chat_history']]
 
-
-def display_chat_history(chat_history):
-    for chat in chat_history:
-        background_color = "#FFA07A" if chat[2] == "new" else "#acf" if chat[0] == "User" else "#caf"
-        st.markdown(f"<div style='background-color: {background_color}; padding: 10px; border-radius: 10px; margin: 10px;'>{chat[0]}: {chat[1]}</div>", unsafe_allow_html=True)
-
 if __name__ == "__main__":
     main()