Update app.py

app.py

@@ -417,23 +417,14 @@ def refresh_faiss_index():
         st.session_state["faiss_db"] = db_faiss
 
 def load_user_data_source(user_id):
-    """
-    Load user-uploaded data source from Firebase.
-    """
-    try:
-        file_path = f"knowledge_bases/{user_id}/knowledge_base.md"
-        storage_url = storage.child(file_path).get_url(None)
-        response = requests.get(storage_url)
-
+    docs = fetch_documents(user_id)
+    user_documents = []
+    for doc in docs.values():
+        response = requests.get(doc["content_url"])
         if response.status_code == 200:
-            user_content = response.text
-            st.session_state["user_data_source"] = user_content
-            return [Document(page_content=user_content)]
-        else:
-            return []
-    except Exception as e:
-        st.error(f"Error loading user data source: {e}")
-        return []
+            user_documents.append(Document(page_content=response.text))
+    return user_documents
+
 
 def update_message_counter():
     remaining_messages = st.session_state["message_limit"] - st.session_state["used_messages"]
@@ -1449,7 +1440,7 @@ def handle_memory_queries(prompt):
     if "save this as trustbuilder" in prompt:
         content_to_save = prompt.replace("save this as trustbuilder", "").strip()
         if content_to_save:  # Only save if there's actual content
-            store_trustbuilder(st.session_state["wix_user_id"], content_to_save)
+            save_content(st.session_state["wix_user_id"], content_to_save)
             return "TrustBuilder saved successfully!"
         else:
             return "Please provide content to save as TrustBuilder."
@@ -1458,7 +1449,7 @@ def handle_memory_queries(prompt):
     elif "save this as brand tonality" in prompt:
         content_to_save = prompt.replace("save this as brand tonality", "").strip()
         if content_to_save:  # Only save if there's actual content
-            store_brand_tonality(st.session_state["wix_user_id"], content_to_save)
+            save_content(st.session_state["wix_user_id"], content_to_save)
             return "Brand Tonality saved successfully!"
         else:
             return "Please provide content to save as Brand Tonality."