Updated app to use fire base

app.py

@@ -7,6 +7,21 @@ import torch
 from sentence_transformers import SentenceTransformer
 from transformers import pipeline, RagTokenizer, RagRetriever, RagSequenceForGeneration
 
+# -----------------------------
+# Firebase Initialization
+# -----------------------------
+import firebase_admin
+from firebase_admin import credentials, firestore
+
+@st.cache_resource
+def init_firestore():
+    # Replace with the path to your Firebase service account key
+    cred = credentials.Certificate("path/to/serviceAccountKey.json")
+    firebase_admin.initialize_app(cred)
+    return firestore.client()
+
+db = init_firestore()
+
 # ========================
 # File Names & Model Names
 # ========================
@@ -90,9 +105,18 @@ if user_query:
         if user_query.lower() in pair["question"].lower():
             answer = pair["answer"]
             break
-    
+
+    # If no curated answer found, save the question to Firebase and generate an answer.
     if not answer:
-        # Retrieve relevant context
+        try:
+            db.collection('llmquestions').add({
+                "question": user_query,
+                "timestamp": firestore.SERVER_TIMESTAMP
+            })
+        except Exception as e:
+            st.error(f"Error saving question to Firebase: {e}")
+
+        # Retrieve relevant context from FAISS index
         query_embedding = embed_model.encode([user_query]).astype("float32")
         distances, indices = index.search(query_embedding, 3)
         pdf_context = "\n".join(chunks[idx] for idx in indices[0])

curated_qa_pairs.json

@@ -19,6 +19,7 @@
       "question": "How does Adagrasib compare to immune checkpoint inhibitors?",
       "answer": "The analysis suggests that comparisons between Adagrasib and immune checkpoint inhibitors (or Docetaxel) should account for patients’ prior treatment lines. Direct comparisons may not be relevant in certain subgroups, especially in patients with only one prior line of systemic therapy."
     },
+    
     {
       "question": "Why were the PICO exercises carried out by the JCA subgroup in Spring 2024?",
       "answer": "They were conducted to test and improve the content of the Guidance on the scoping process by applying it to real cases, ensuring that the resulting guidance would be practical and refined for future assessments."