Update app.py

app.py

@@ -10,6 +10,7 @@ from langchain.prompts import PromptTemplate
 from langchain_community.document_loaders import PyPDFLoader
 from langchain_chroma import Chroma
 import io
+import fitz
 
 st.set_page_config(page_title="Document Genie", layout="wide")
 
@@ -34,14 +35,8 @@ Follow these simple steps to interact with the chatbot:
 
 def get_pdf(pdf_file):
     # Extract file path from UploadedFile object
-    file_path = pdf_file.name
-    # Read the content of the uploaded PDF file
-    pdf_content = pdf_file.read()
-    # Close the file object
-    pdf_file.close()
-    # Load the PDF content using PyPDFLoader
-    loader = PyPDFLoader(io.BytesIO(pdf_content))
-    docs = loader.load()
+    file_pdf = fitz.open(stream=pdf_file.read(), filetype="pdf")
+    docs = file_pdf.load()
     return docs
 
 def text_splitter(text):