Update app.py

app.py

@@ -59,7 +59,29 @@ client = Groq(api_key="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"
 
 # Predefined list of Google Drive links
 drive_links = [
-    "https://drive.google.com/file/d/15ep4h17VFarb4coWAgqcEUDUTN7Ek6dr/view?usp=sharing"
+    "https://drive.google.com/file/d/15ep4h17VFarb4coWAgqcEUDUTN7Ek6dr/view?usp=sharing",
+    "https://drive.google.com/file/d/1ocxUsw4rVc9MNV_9wARvXg9D9SLe2vVi/view?usp=sharing",
+    "https://drive.google.com/file/d/1yrOTIlOYywjbfeMWcBG1kBK8-LLYGl3t/view?usp=sharing",
+    "https://drive.google.com/file/d/1otdel-JxaZBvitQQMiXtniTNAsmICrMT/view?usp=sharing",
+    "https://drive.google.com/file/d/16--HjKNn5Oz1cqX3b-8qvwnD0_yLvdy2/view?usp=sharing",
+    "https://drive.google.com/file/d/1lnBD-0H1mJJ8N-lNdrhomuzgm8nRFVxA/view?usp=sharing",
+    "https://drive.google.com/file/d/1ni7IxoI7j2wzCcWjQEEqrE1JO0HsRGrl/view?usp=sharing",
+    "https://drive.google.com/file/d/1s4QkcbTRsgT8Ju4O23gVpuh4Hy-dsHnb/view?usp=sharing",
+    "https://drive.google.com/file/d/1eRoAWgYqNsmYHVa90J2Tm0ejIMa_8qvZ/view?usp=sharing",
+    "https://drive.google.com/file/d/1r8SKClbsyAzYqOTfTEZXcQwf5YUEHPMG/view?usp=sharing",
+    "https://drive.google.com/file/d/1L-Lc_Yv14Tlu3waA3ufDwd4kzTkZfLHC/view?usp=sharing",
+    "https://drive.google.com/file/d/1bqz_eQkeo8Qc0OPaghIdQFMbDT9eFS77/view?usp=sharing",
+    "https://drive.google.com/file/d/1VSSNBc76yh_OtyBzuC_S43nrFMIc8Fw1/view?usp=sharing",
+    "https://drive.google.com/file/d/1NxQevabv6PPXxeXv5DZyeP8Q9GlHwD2r/view?usp=sharing",
+    "https://drive.google.com/file/d/1S_XKeDtPI_UBupnIURZTPiddmLbjqnRZ/view?usp=sharing",
+    "https://drive.google.com/file/d/1Bppf9VgUShBJWnYEw-4ZwDu5-A51nR9R/view?usp=sharing",
+    "https://drive.google.com/file/d/1s7z4TpFmMpvx1-PpYsaukACOSTrkbcOf/view?usp=sharing",
+    "https://drive.google.com/file/d/12Rrh0ixkQPvAAhExdnbMeol7c9aAzyTd/view?usp=sharing",
+    "https://drive.google.com/file/d/1BGq_aacZjmkEfAa2Oyzfw32piEQfBAmZ/view?usp=sharing",
+    "https://drive.google.com/file/d/1nxFqyFOa3C0TK8BLOYp-V6e_xjYGn0qV/view?usp=sharing",
+    "https://drive.google.com/file/d/1xsMYVFufaM2gzMPKADNCrDbGyq0rhSV2/view?usp=sharing"
+    
+
 ]
 
 # Function to download PDF from Google Drive
@@ -103,20 +125,20 @@ def create_embeddings_and_store(chunks):
     return vector_db
 
 # Function to query the vector database and interact with Groq
-def query_vector_db(query, vector_db):
-    # Retrieve relevant documents
-    docs = vector_db.similarity_search(query, k=3)
-    context = "\n".join([doc.page_content for doc in docs])
-
-    # Interact with Groq API
-    chat_completion = client.chat.completions.create(
-        messages=[
-            {"role": "system", "content": f"Use the following context:\n{context}"},
-            {"role": "user", "content": query},
-        ],
-        model="llama3-8b-8192",
-    )
-    return chat_completion.choices[0].message.content
+def query_vector_db(query, vector_db, selected_appliance):
+  # Filter documents based on appliance title
+  filtered_documents = [doc for doc in vector_db.documents if doc.title == selected_appliance]
+  context = "\n".join([doc.page_content for doc in filtered_documents])
+
+  # Interact with Groq API (unchanged)
+  chat_completion = client.chat.completions.create(
+      messages=[
+          {"role": "system", "content": f"Use the following context:\n{context}"},
+          {"role": "user", "content": query},
+      ],
+      model="llama3-8b-8192",
+  )
+  return chat_completion.choices[0].message.content
 
 # Streamlit app
 st.title("RAG-Based ChatBot (Already having Document)")