Update flask_app.py

flask_app.py

@@ -55,32 +55,19 @@ def create_faiss_index(docs):
     vector_store = FAISS.from_documents(texts, embedding_model)
     return vector_store
 
-# Function to extract files from .rar archive
-def extract_rar_files(rar_path, extract_to="extracted_files"):
-    with rarfile.RarFile(rar_path) as rf:
-        rf.extractall(extract_to)
-    return extract_to
-
 # Function to fetch and process PDF files
-def fetch_and_process_pdfs(folder_path):
-    pdf_files = [
-        os.path.join(folder_path, file) for file in os.listdir(folder_path) if file.endswith(".pdf")
-    ]
-    
+def fetch_and_process_pdfs(pdf_paths):
     all_documents = []
-    for pdf_file in pdf_files:
+    for pdf_file in pdf_paths:
         loader = PyPDFLoader(pdf_file)
         documents = loader.load()
         all_documents.extend(documents)
     return all_documents
 
-# Function to create RAG pipeline
-def create_rag_pipeline(rar_path):
-    # Extract .rar file
-    extracted_folder = extract_rar_files(rar_path)
-
+# Function to create RAG pipeline with provided PDF paths
+def create_rag_pipeline(pdf_paths):
     # Fetch and process PDF files
-    documents = fetch_and_process_pdfs(extracted_folder)
+    documents = fetch_and_process_pdfs(pdf_paths)
 
     # Create vector store
     vector_store = create_faiss_index(documents)
@@ -88,9 +75,17 @@ def create_rag_pipeline(rar_path):
 
     return retriever
 
+# Specify the PDF paths
+pdfs_paths = [
+    "Esta_Code_2024_.pdf", 
+    "KP_Policy_Agriculture1.pdf", 
+    "THE_KHYBER_PAKHTUNKHWA_SALES_TAX_ON_SERVICES_ACT_2022.pdf", 
+    "The-Khyber-Pakhtunkhwa-Police-Amendment-Act-2024.pdf"
+]
+
 # Usage example
-rar_path = "Rules folder.rar"
-retriever = create_rag_pipeline(rar_path)
+retriever = create_rag_pipeline(pdfs_paths)
+