update to work on one pdf

app.py

@@ -12,12 +12,6 @@ from pypdf import PdfReader, PdfWriter
 from pathlib import Path
 
 
-data_root = './data/pdf/'
-
-def load_pdf_paths(data_root):
-    return [data_root+path for path in os.listdir(data_root)]
-
-
 def build_rag_chain(pdf_paths):
     loaders = [PyPDFLoader(path) for path in pdf_paths]
 
@@ -60,25 +54,25 @@ def build_rag_chain(pdf_paths):
 def predict(query, pdf_file):
     print(type(pdf_file))
     if pdf_file:
-        pdf_path = Path(pdf_file)
-        pdf_reader = PdfReader(pdf_path)
-        pdf_writer = PdfWriter()
+        # pdf_path = Path(pdf_file)
+        # pdf_reader = PdfReader(pdf_path)
+        # pdf_writer = PdfWriter()
 
 
-        pdf_name = pdf_file.split('/')[-1]
-        pdf_path = data_root + pdf_name
+        # pdf_name = pdf_file.split('/')[-1]
+        # pdf_path = data_root + pdf_name
 
-        if pdf_path not in load_pdf_paths(data_root):
-            print('Saving file...')
-            for page in pdf_reader.pages:
-                pdf_writer.add_page(page)
+        # if pdf_path not in load_pdf_paths(data_root):
+        #     print('Saving file...')
+        #     for page in pdf_reader.pages:
+        #         pdf_writer.add_page(page)
             
-            with open(pdf_path, 'wb') as f:
-                pdf_writer.write(f)
-        os.system("ls data/pdf")
+        #     with open(pdf_path, 'wb') as f:
+        #         pdf_writer.write(f)
+        # os.system("ls data/pdf")
             
-    pdf_paths = load_pdf_paths(data_root)
-    rag_chain = build_rag_chain(pdf_paths)
+    # pdf_paths = load_pdf_paths(data_root)
+        rag_chain = build_rag_chain([pdf_file])
     return rag_chain.invoke(query)
 
 # examples = [