Update app.py

app.py

@@ -1,4 +1,5 @@
 import os
+import gdown
 import gradio as gr
 import faiss
 import numpy as np
@@ -8,7 +9,6 @@ from groq import Groq
 
 # ==============================
 # 🔐 SET YOUR GROQ API KEY
-# In HuggingFace Spaces, go to Settings → Secrets → Add "GROQ_RAG_API"
 # ==============================
 api_key = os.environ.get("GROQ_RAG_API")
 if not api_key:
@@ -16,16 +16,29 @@ if not api_key:
 client = Groq(api_key=api_key)
 
 # ==============================
-# 📥 PDF DOCUMENTS
-# Put your PDFs directly in a folder called "documents" in your repo
+# 📥 GOOGLE DRIVE LINKS
+# Add as many as you want
 # ==============================
+drive_links = [
+    "https://drive.google.com/file/d/1pKlbI7XbhlDc-hOBJEkLpYteg6mkeY0X/view?usp=sharing"
+]
+
 pdf_folder = "documents"
-if not os.path.exists(pdf_folder):
-    os.makedirs(pdf_folder)
+os.makedirs(pdf_folder, exist_ok=True)
+
+def download_from_drive(link):
+    file_id = link.split("/d/")[1].split("/")[0]
+    output_path = os.path.join(pdf_folder, f"{file_id}.pdf")
+    if not os.path.exists(output_path):
+        download_url = f"https://drive.google.com/uc?id={file_id}"
+        gdown.download(download_url, output_path, quiet=False)
+    return output_path
 
-# Load PDFs from the folder
-pdf_files = [os.path.join(pdf_folder, f) for f in os.listdir(pdf_folder) if f.endswith(".pdf")]
+pdf_files = [download_from_drive(link) for link in drive_links]
 
+# ==============================
+# 📖 LOAD PDF TEXT
+# ==============================
 documents = []
 for pdf in pdf_files:
     reader = PdfReader(pdf)
@@ -53,7 +66,7 @@ print("Total chunks:", len(chunks))
 # ==============================
 model = SentenceTransformer("all-MiniLM-L6-v2")
 embeddings = model.encode(chunks, convert_to_numpy=True)
-embeddings = embeddings.astype("float32")  # important for FAISS
+embeddings = embeddings.astype("float32")  # for FAISS
 
 # ==============================
 # 🗄️ FAISS VECTOR INDEX
@@ -105,7 +118,7 @@ def chat_fn(message, history):
 
 with gr.Blocks() as demo:
     gr.Markdown("# 📚 RAG Assistant")
-    gr.Markdown("Ask questions from your uploaded PDFs.")
+    gr.Markdown("Ask questions from your Google Drive PDFs.")
     chatbot = gr.Chatbot()
     msg = gr.Textbox(placeholder="Ask your question here...")
     btn = gr.Button("Send")