Wajahat698 commited on
Commit
7b082d8
·
verified ·
1 Parent(s): 1c9bea3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -9
app.py CHANGED
@@ -180,13 +180,13 @@ def upload_to_firebase(user_id, file):
180
 
181
  def fetch_trustbuilders(user_id):
182
  """
183
- Fetch all TrustBuilders for a user from Firebase.
184
  """
185
  try:
186
- trustbuilders = db.child("users").child(user_id).child("TrustBuilders").get().val()
187
- if trustbuilders:
188
- # Return TrustBuilders in dictionary format
189
- return trustbuilders
190
  else:
191
  st.warning("No TrustBuilders found.")
192
  return {}
@@ -495,17 +495,18 @@ def save_content(user_id, content, bucket):
495
  Save TrustBuilder in Firebase under the correct structure.
496
  """
497
  try:
498
- trustbuilder_id = str(uuid.uuid4()) # Generate unique ID for each TrustBuilder
499
  trustbuilder_data = {
500
  "content": f"{bucket}: {content}"
501
  }
502
  # Save under the user's TrustBuilders path
503
  db.child("users").child(user_id).child("TrustBuilders").child(trustbuilder_id).set(trustbuilder_data)
504
- st.success(f"TrustBuilder '{content}' under '{bucket}' saved successfully!")
505
- st.write(f"Debugging: Saved Data - {trustbuilder_data}") # Log the saved data
506
  except Exception as e:
507
  st.error(f"Error saving TrustBuilder: {e}")
508
-
 
 
509
  def ai_allocate_trust_bucket(trust_builder_text):
510
  # Implement your AI allocation logic here
511
  return "Stability"
 
180
 
181
  def fetch_trustbuilders(user_id):
182
  """
183
+ Fetch all TrustBuilders for a user from Firebase using Pyrebase.
184
  """
185
  try:
186
+ # Fetch all TrustBuilders for the given user ID
187
+ trustbuilders = db.child("users").child(user_id).child("TrustBuilders").get()
188
+ if trustbuilders.each():
189
+ return {item.key(): item.val() for item in trustbuilders.each()}
190
  else:
191
  st.warning("No TrustBuilders found.")
192
  return {}
 
495
  Save TrustBuilder in Firebase under the correct structure.
496
  """
497
  try:
498
+ trustbuilder_id = str(uuid.uuid4()) # Generate a unique ID for the TrustBuilder
499
  trustbuilder_data = {
500
  "content": f"{bucket}: {content}"
501
  }
502
  # Save under the user's TrustBuilders path
503
  db.child("users").child(user_id).child("TrustBuilders").child(trustbuilder_id).set(trustbuilder_data)
504
+ st.success(f"TrustBuilder saved successfully under '{bucket}'!")
 
505
  except Exception as e:
506
  st.error(f"Error saving TrustBuilder: {e}")
507
+
508
+
509
+
510
  def ai_allocate_trust_bucket(trust_builder_text):
511
  # Implement your AI allocation logic here
512
  return "Stability"