Update app.py

app.py

@@ -15,6 +15,7 @@ from langchain.prompts import PromptTemplate
 from langchain.vectorstores import Chroma
 from utils import load_txt_documents, split_docs, load_uploaded_documents, retriever_from_chroma
 from langchain.text_splitter import TokenTextSplitter, RecursiveCharacterTextSplitter
+from langchain_community.document_loaders.directory import DirectoryLoader
 
 script_dir = os.path.dirname(os.path.abspath(__file__))
 data_path = os.path.join(script_dir, "data/")
@@ -45,12 +46,8 @@ def get_vectorstore(text_chunks):
     return vectorstore
 
 def get_pdf_text(pdf_docs):
-    text = ""
-    for pdf in pdf_docs:
-        pdf_reader = PdfReader(pdf)
-        for page in pdf_reader.pages:
-            text += page.extract_text()
-    return text
+    document_loader = DirectoryLoader(pdf_docs)
+    return document_loader.load()
 
 def get_text_chunks(text):
     text_splitter = RecursiveCharacterTextSplitter.from_tiktoken_encoder(