Update app.py

app.py

@@ -186,9 +186,12 @@ def fetch_trustbuilders(user_id):
         st.write(f"Debugging: Fetching TrustBuilders for user: {user_id}")
         trustbuilders = db.child("users").child(user_id).child("TrustBuilders").get().val()
         st.write(f"Debugging: Raw TrustBuilders fetched: {trustbuilders}")
+        
         if trustbuilders:
+            # Extract content from TrustBuilders
             return [tb["content"] for tb in trustbuilders.values()]
         else:
+            st.warning("No TrustBuilders found in Firebase.")
             return []
     except Exception as e:
         st.error(f"Error fetching TrustBuilders: {e}")
@@ -495,15 +498,20 @@ def save_content(user_id, content):
     Save a TrustBuilder as plain text under the user's TrustBuilders node in Firebase.
     """
     try:
+        # Prepare the TrustBuilder data
         trustbuilder_data = {
             "content": content
         }
-        # Push to TrustBuilders node
+        
+        # Push to TrustBuilders node under the user's ID
         db.child("users").child(user_id).child("TrustBuilders").push(trustbuilder_data)
+        
         st.success("TrustBuilder saved successfully!")
         st.write(f"Debugging: Saved TrustBuilder content: {trustbuilder_data}")
     except Exception as e:
         st.error(f"Error saving TrustBuilder: {e}")
+
+        
 def ai_allocate_trust_bucket(trust_builder_text):
     # Implement your AI allocation logic here
     return "Stability"
@@ -1503,20 +1511,30 @@ if "used_messages" not in st.session_state:
 
 def initialize_user_session():
     """
-    Initialize user data in Firebase and session state.
+    Initialize user session and ensure user data exists in Firebase.
     """
     try:
         user_id = st.session_state["wix_user_id"]
         email = st.session_state["email"]
-        user_data = {
-            "user_name": user_id,
-            "email": email,
-            "message_limit": 1000,
-            "used_messages": 0,
-            "TrustBuilders": {}
-        }
-        db.child("users").child(user_id).update(user_data)
-        st.success("User session initialized successfully.")
+
+        # Check if user already exists in Firebase
+        user_data = db.child("users").child(user_id).get().val()
+        
+        if not user_data:
+            # Create user data in Firebase if it doesn't exist
+            user_data = {
+                "user_name": user_id,
+                "email": email,
+                "message_limit": 1000,
+                "used_messages": 0
+            }
+            db.child("users").child(user_id).set(user_data)
+            st.success("User session initialized and user data created in Firebase.")
+        else:
+            st.success("User data loaded from Firebase.")
+        
+        # Update session state with user data
+        st.session_state.update(user_data)
     except Exception as e:
         st.error(f"Error initializing user session: {e}")