Update app.py

app.py

@@ -411,29 +411,41 @@ def load_main_data_source():
         st.error(f"Error loading main data source: {e}")
         return []
 
-def combine_data_sources():
-    main_data_source = load_main_data_source()
- 
-    return main_data_source 
+def combine_data_sources(user_id, include_main=True):
+    main_data_source = load_main_data_source() if include_main else []
+    user_documents = load_user_data_source(user_id)
+    return main_data_source + user_documents
 
 
-def refresh_faiss_index():
-    combined_sources = combine_data_sources()
-    if combined_sources:
+def refresh_faiss_index(documents=None):
+    """
+    Refresh the FAISS index with updated documents.
+    If documents are provided, only update those; otherwise, rebuild the index.
+    """
+    if documents:
+        # Add/Update specific documents in the index
         embeddings = OpenAIEmbeddings()
-        db_faiss = FAISS.from_documents(combined_sources, embeddings)
-        st.session_state["faiss_db"] = db_faiss
+        st.session_state["faiss_db"].add_documents(documents, embeddings)
+    else:
+        # Rebuild the entire index
+        combined_sources = combine_data_sources()
+        if combined_sources:
+            embeddings = OpenAIEmbeddings()
+            st.session_state["faiss_db"] = FAISS.from_documents(combined_sources, embeddings)
+
 
 def load_user_data_source(user_id):
-    """
-    Fetches user-uploaded data sources from Firebase.
-    """
-    docs = fetch_documents(user_id)
-    user_documents = []
+    try:
+        docs = db.child("users").child(user_id).child("KnowledgeBase").get().val()
+        if not docs:
+            return []
+        user_documents = [Document(page_content=doc["content"]) for doc in docs.values()]
+        return user_documents
+    except Exception as e:
+        st.error(f"Error loading user data source: {e}")
+        return []
+
 
-   
-       
-    return user_documents