Update app.py

app.py

@@ -165,7 +165,31 @@ def upload_to_firebase(user_id, content):
         st.error(f"Error uploading to Firebase: {e}")
         return None
 
+def fetch_trustbuilders(user_id, keyword=None, brand=None, bucket=None):
+    try:
+        trustbuilders = db.child("users").child(user_id).child("TrustBuilder").get().val()
+        if trustbuilders:
+            filtered_trustbuilders = {
+                k: v for k, v in trustbuilders.items()
+                if (not keyword or keyword.lower() in v.get("message", "").lower())
+                   and (not brand or brand.lower() in v.get("brand", "").lower())
+                   and (not bucket or bucket == v.get("bucket", ""))
+            }
+            return filtered_trustbuilders
+        return {}
+    except Exception as e:
+        st.error(f"Error fetching TrustBuilders: {e}")
+        return {}
 
+def delete_trustbuilder(user_id, trustbuilder_id):
+    try:
+        db.child("users").child(user_id).child("TrustBuilder").child(trustbuilder_id).remove()
+        st.success("TrustBuilder deleted successfully.")
+        st.experimental_rerun()
+    except Exception as e:
+        st.error(f"Error deleting TrustBuilder: {e}")
+
+        
 
 # Define and validate API keys
 openai_api_key = os.getenv("OPENAI_API_KEY")
@@ -503,6 +527,9 @@ def save_content(user_id, brand, trust_builder_text, trust_bucket=None):
     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"
 
         
 
@@ -534,17 +561,13 @@ def delete_document(user_id, doc_id):
         st.error(f"Error deleting document: {e}")
 
 # Fetch documents from Firebase
-def fetch_documents(user_id):
-    """
-    Fetches all saved documents from Firebase.
-    """
+def delete_document(user_id, doc_id):
     try:
-        docs = db.child("users").child(user_id).child("KnowledgeBase").get().val()
-        return docs if docs else {}
+        db.child("users").child(user_id).child("documents").child(doc_id).remove()
+        st.success("Document deleted successfully.")
+        st.experimental_rerun()
     except Exception as e:
-        st.error(f"Error fetching documents: {e}")
-        return {}
-
+        st.error(f"Error deleting document: {e}")
 
 
 def side():
@@ -682,22 +705,27 @@ def side():
         .scrollable-container {
             max-height: 300px;
             overflow-y: auto;
-            border: 1px solid #ddd;
+            border: 1px solid gray;
             padding: 10px;
             border-radius: 5px;
+            background-color: #f9f9f9;
         }
         </style>
     """, unsafe_allow_html=True)
     
-        st.markdown('<div class="scrollable-container">', unsafe_allow_html=True)
+        # Container for Saved Documents
+        with st.container():
+            st.subheader("Saved Documents")
+            st.write("Save documents like your brand tonality, key phrases, or segments here.")
+    
+            # Scrollable container with gray border
+            st.markdown('<div class="scrollable-container">', unsafe_allow_html=True)
     
-        # Saved Documents Section
-        st.subheader("Saved Documents")
-        docs = fetch_documents(st.session_state["wix_user_id"])
+            docs = fetch_documents(st.session_state["wix_user_id"])
     
-        if docs:
-            for doc_id, doc_data in docs.items():
-                with st.expander(f"Document ID: {doc_id[:8]}"):
+            if docs:
+                for doc_id, doc_data in docs.items():
+                    st.markdown(f"**Document ID:** {doc_id[:8]}")
                     st.text_area("Preview", doc_data.get("content", ""), height=80, disabled=True)
                     public_url = doc_data.get("public_url", "URL not available")
                     if public_url != "URL not available":
@@ -705,9 +733,10 @@ def side():
                     # Add delete button for each document
                     if st.button(f"Delete Document {doc_id[:8]}", key=f"del_doc_{doc_id}"):
                         delete_document(st.session_state["wix_user_id"], doc_id)
-                        st.experimental_rerun()  # Refresh the app to reflect deletion
-        else:
-            st.write("No saved documents found.")
+            else:
+                st.write("No saved documents found.")
+    
+            st.markdown('</div>', unsafe_allow_html=True)
     
         # File Upload Section
         st.subheader("Upload New Documents")
@@ -734,14 +763,12 @@ def side():
                         })
                         st.success("Files uploaded and saved successfully!")
                         st.markdown(f"[View your knowledge base here]({public_url})", unsafe_allow_html=True)
-                        st.experimental_rerun()  # Refresh to show the new document
+                        st.experimental_rerun()
                 else:
                     st.warning("No valid files uploaded.")
             else:
                 st.warning("Please select at least one file to upload.")
     
-        st.markdown('</div>', unsafe_allow_html=True)
-    
         # Show My TrustBuilders Section
         st.subheader("Show my TrustBuilders®")
         search_query = st.text_input("Search by keyword", key="keyword")
@@ -761,7 +788,7 @@ def side():
                     with st.expander(f"TrustBuilder ID: {result_id[:8]}"):
                         st.write(f"Brand: {result_data.get('brand', 'N/A')}")
                         st.write(f"Bucket: {result_data.get('bucket', 'N/A')}")
-                        st.write(f"Message: {result_data['message']}")
+                        st.write(f"Message: {result_data.get('message', '')}")
                         # Add delete button for each TrustBuilder
                         if st.button(f"Delete TrustBuilder {result_id[:8]}", key=f"del_tb_{result_id}"):
                             delete_trustbuilder(st.session_state["wix_user_id"], result_id)
@@ -769,13 +796,13 @@ def side():
                 st.write("No TrustBuilders found.")
     
         # Trust Bucket Selection
-        st.write("and choose the Trust Bucket®")
+        st.write("Choose the Trust Bucket®")
         trust_buckets = ["S", "D", "R", "B", "V", "C"]
         bucket_labels = ["Stability", "Development", "Relationship", "Benefit", "Vision", "Competence"]
         cols = st.columns(len(trust_buckets))
         for idx, (bucket, label) in enumerate(zip(trust_buckets, bucket_labels)):
             if cols[idx].button(f"{bucket}", key=f"bucket_{label}"):
-                st.write(f"{label} Trust clicked")
+                st.write(f"Selected Trust Bucket: {label}")
                 st.session_state["selected_bucket"] = label
     
         # Save TrustBuilders Section
@@ -797,6 +824,8 @@ def side():
                 st.success(f"AI allocated the Trust Bucket: {allocated_bucket} and saved your TrustBuilder.")
             else:
                 st.error("AI could not allocate the Trust Bucket.")
+
+    
 side()
 # Load knowledge base