Update app.py

app.py

@@ -10,7 +10,7 @@ from langchain.prompts import PromptTemplate
 from langchain_community.document_loaders import PyPDFLoader
 from langchain_chroma import Chroma
 import io
-import fitz
+from streamlit import session_state as ss
 
 st.set_page_config(page_title="Document Genie", layout="wide")
 
@@ -35,8 +35,9 @@ Follow these simple steps to interact with the chatbot:
 
 def get_pdf(pdf_file):
     # Extract file path from UploadedFile object
-    file_pdf = fitz.open(stream=pdf_file.read(), filetype="pdf")
-    docs = file_pdf.load()
+    with open(pdf_file, "rb") as f:
+        loader = PyPDFLoader(f)
+        docs = file_pdf.load()
     return docs
 
 def text_splitter(text):