Update app.py

app.py

@@ -10,7 +10,7 @@ from langchain.vectorstores.chroma import Chroma
 from langchain.prompts import ChatPromptTemplate
 from langchain_community.llms.ollama import Ollama
 from get_embedding_function import get_embedding_function
-from langchain.document_loaders.pdf import PyPDFLoader
+from langchain.document_loaders import PyPDFDirectoryLoader
 from langchain.text_splitter import RecursiveCharacterTextSplitter
 from langchain.schema import Document
 import tempfile
@@ -166,16 +166,18 @@ def query_rag(query_text: str):
 
 def save_pdf_and_update_database(pdf_filepath):
     try:
-        document_loader = PyPDFLoader(pdf_filepath)
+        # Assuming you're loading PDFs from a specific directory
+        document_loader = PyPDFDirectoryLoader(os.path.dirname(pdf_filepath))
         documents = document_loader.load()
-
+        
         text_splitter = RecursiveCharacterTextSplitter(
             chunk_size=800,
             chunk_overlap=80,
             length_function=len,
+            is_separator_regex=False,
         )
         chunks = text_splitter.split_documents(documents)
-
+        
         add_to_chroma(chunks)
         print(f"PDF processed and data updated in Chroma.")
     except Exception as e: