Update app.py

app.py

@@ -101,6 +101,39 @@ def display_chat_history(chat_history):
         background_color = "#FFA07A" if sender == "User" else "#caf"
         st.markdown(f"<div style='background-color: {background_color}; padding: 10px; border-radius: 10px; margin: 10px;'>{sender}: {msg}</div>", unsafe_allow_html=True)
 
+
+def load_pdf(file_path):
+    pdf_reader = PdfReader(file_path)
+    text = ""
+    for page in pdf_reader.pages:
+        text += page.extract_text()
+
+    text_splitter = RecursiveCharacterTextSplitter(
+        chunk_size=1000,
+        chunk_overlap=200,
+        length_function=len
+    )
+    chunks = text_splitter.split_text(text=text)
+
+    store_name, _ = os.path.splitext(os.path.basename(file_path))
+
+    if os.path.exists(f"{store_name}.pkl"):
+        with open(f"{store_name}.pkl", "rb") as f:
+            VectorStore = pickle.load(f)
+    else:
+        embeddings = OpenAIEmbeddings()
+        VectorStore = FAISS.from_texts(chunks, embedding=embeddings)
+        with open(f"{store_name}.pkl", "wb") as f:
+            pickle.dump(VectorStore, f)
+
+    return VectorStore
+
+
+
+def load_chatbot():
+    return load_qa_chain(llm=OpenAI(), chain_type="stuff")
+
+    
 def main():
     hide_streamlit_style = """
         <style>