Update app.py

app.py

@@ -36,7 +36,7 @@ with st.sidebar:
 
 openai_api_key = st.text_input("Enter your OpenAI API key:")
 
-def load_pdf(file_path):
+def load_pdf(file_path, openai_api_key):
     pdf_reader = PdfReader(file_path)
     text = ""
     for page in pdf_reader.pages:
@@ -55,7 +55,7 @@ def load_pdf(file_path):
         with open(f"{store_name}.pkl", "rb") as f:
             VectorStore = pickle.load(f)
     else:
-        embeddings = OpenAIEmbeddings()
+        embeddings = OpenAIEmbeddings(openai_api_key=openai_api_key)  # Pass the API key here
         VectorStore = FAISS.from_texts(chunks, embedding=embeddings)
         with open(f"{store_name}.pkl", "wb") as f:
             pickle.dump(VectorStore, f)
@@ -82,6 +82,10 @@ def main():
     if "processing_input" not in st.session_state:
         st.session_state['processing_input'] = ""
 
+    # 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)