Tests langchain

app.py

@@ -15,11 +15,18 @@ import asyncio
 import nest_asyncio
 nest_asyncio.apply()
 
+from langchain_community.document_loaders import PyMuPDFLoader
 
-pdf_loader_NIST = PDFFileLoader("data/NIST.AI.600-1.pdf")
-pdf_loader_Blueprint = PDFFileLoader("data/Blueprint-for-an-AI-Bill-of-Rights.pdf")
-documents_NIST = pdf_loader_NIST.load_documents()
-documents_Blueprint = pdf_loader_Blueprint.load_documents()
+filepath_NIST = "data/NIST.AI.600-1.pdf"
+filepath_Blueprint = "data/Blueprint-for-an-AI-Bill-of-Rights.pdf"
+
+documents_NIST = PyMuPDFLoader(filepath_NIST).load()
+documents_Blueprint = PyMuPDFLoader(filepath_Blueprint).load()
+
+# pdf_loader_NIST = PDFFileLoader("data/NIST.AI.600-1.pdf")
+# pdf_loader_Blueprint = PDFFileLoader("data/Blueprint-for-an-AI-Bill-of-Rights.pdf")
+# documents_NIST = pdf_loader_NIST.load_documents()
+# documents_Blueprint = pdf_loader_Blueprint.load_documents()
 
 text_splitter = CharacterTextSplitter()
 split_documents_NIST = text_splitter.split_texts(documents_NIST)