Update app.py

app.py

@@ -504,32 +504,48 @@ def download_link(content, filename):
 
 
 
+def fetch_documents(user_id):
+    try:
+        docs = db.child("users").child(user_id).child("documents").get().val()
+        return docs if docs else {}
+    except Exception as e:
+        st.sidebar.error(f"Error fetching documents: {e}")
+        return {}
 
-
-# Function to delete a document from Firebase
 def delete_document(user_id, doc_id):
-    """
-    Deletes a document from Firebase.
-    """
     try:
-        db.child("users").child(user_id).child("KnowledgeBase").child(doc_id).remove()
-        st.success("Document deleted successfully!")
-        st.rerun()  # Refresh the list after deletion
+        db.child("users").child(user_id).child("documents").child(doc_id).remove()
+        st.sidebar.success("Document deleted successfully.")
     except Exception as e:
-        st.error(f"Error deleting document: {e}")
+        st.sidebar.error(f"Error deleting document: {e}")
 
-# Fetch documents from Firebase
-def fetch_documents(user_id):
-    """
-    Fetches all saved documents from Firebase.
-    """
+def save_trustbuilder(user_id, brand, trust_builder, trust_bucket):
     try:
-        docs = db.child("users").child(user_id).child("KnowledgeBase").get().val()
-        return docs if docs else {}
+        trust_id = str(uuid.uuid4())
+        db.child("users").child(user_id).child("TrustBuilder").child(trust_id).set({
+            "brand": brand,
+            "message": trust_builder,
+            "bucket": trust_bucket
+        })
+        st.sidebar.success("TrustBuilder saved successfully.")
     except Exception as e:
-        st.error(f"Error fetching documents: {e}")
-        return {}
+        st.sidebar.error(f"Error saving TrustBuilder: {e}")
 
+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["message"].lower()) and
+                   (not brand or brand.lower() in v["brand"].lower()) and
+                   (not bucket or bucket == v["bucket"])
+            }
+            return filtered_trustbuilders
+        return {}
+    except Exception as e:
+        st.sidebar.error(f"Error fetching TrustBuilders: {e}")
+        return {}
 
 
 def side():
@@ -662,12 +678,27 @@ def side():
                 
         # Image paths and labels (pre-resized images in your folder)
         st.sidebar.header("TrustVault®")
-
-    # Saved Documents Section
+    
+        # Saved Documents Section
         st.sidebar.subheader("Saved documents")
         st.sidebar.text_area("Save documents like your brand tonality, key phrases, or segments here", height=80, disabled=True)
+    
+        # Scrollable container for displaying saved documents
+        st.sidebar.markdown("""
+            <style>
+            .scrollable-container {
+                max-height: 300px;
+                overflow-y: auto;
+                border: 1px solid #ddd;
+                padding: 10px;
+                border-radius: 5px;
+            }
+            </style>
+        """, unsafe_allow_html=True)
+    
+        st.sidebar.markdown('<div class="scrollable-container">', unsafe_allow_html=True)
         docs = fetch_documents(st.session_state["wix_user_id"])
-
+    
         if docs:
             for doc_id, doc_data in docs.items():
                 st.sidebar.markdown(f"**Document ID:** {doc_id[:8]}")
@@ -675,16 +706,13 @@ def side():
                 public_url = doc_data.get("public_url", "URL not available")
                 if public_url != "URL not available":
                     st.sidebar.markdown(f"[View Full Document]({public_url})", unsafe_allow_html=True)
-        
-                    # Add delete button for each document
-                    if st.button(f"Delete: {doc_id[:8]}", key=f"del_{doc_id}"):
-                        delete_document(st.session_state["wix_user_id"], public_url)
+                    if st.button(f"Delete {doc_id[:8]}", key=f"del_{doc_id}"):
+                        delete_document(st.session_state["wix_user_id"], doc_id)
         else:
             st.sidebar.write("No saved documents found.")
         st.sidebar.markdown('</div>', unsafe_allow_html=True)
-        
+    
         # File Upload Section
-        st.sidebar.subheader("Upload New Documents")
         col1, col2 = st.sidebar.columns([0.5, 0.5])
         with col1:
             uploaded_files = st.file_uploader("Upload PDF, DOCX, or TXT files", type=["pdf", "docx", "txt"], accept_multiple_files=True)
@@ -694,25 +722,27 @@ def side():
                     merged_md_content = merge_markdown_contents(md_contents)
                     public_url = upload_to_firebase(st.session_state["wix_user_id"], merged_md_content)
                     if public_url:
-                        # Save document reference and public URL to Firebase
                         doc_id = str(uuid.uuid4())
                         db.child(f"/users/{st.session_state['wix_user_id']}/documents/{doc_id}").set({
                             "content": merged_md_content,
                             "public_url": public_url
                         })
                         st.sidebar.success("Files uploaded and saved successfully!")
-                        st.sidebar.markdown(f"[View your knowledge base here]({public_url})", unsafe_allow_html=True)
-                else:
-                    st.sidebar.warning("No valid files uploaded.")
-        
+        with col2:
+            if st.button("Delete"):
+                st.warning("Select a document to delete from the list above.")
+    
         # Show My TrustBuilders Section
         st.sidebar.subheader("Show my TrustBuilders®")
         search_query = st.sidebar.text_input("Search by keyword", key="keyword")
-        if search_query:
-            search_results = {k: v for k, v in st.session_state.get("TrustBuilder", {}).items() if search_query.lower() in v["message"].lower()}
+        brand_input = st.sidebar.text_input("Brand/Product/Person", key="brand")
+        bucket = None
+        if st.sidebar.button("Show"):
+            bucket = st.session_state.get("selected_bucket")
+            search_results = fetch_trustbuilders(st.session_state["wix_user_id"], keyword=search_query, brand=brand_input, bucket=bucket)
             for result_id, result_data in search_results.items():
                 st.sidebar.write(f"- {result_data['message'][:50]}")
-        
+    
         # Trust Bucket Selection
         st.sidebar.write("and choose the Trust Bucket®")
         trust_buckets = ["S", "D", "R", "B", "V", "C"]
@@ -721,11 +751,12 @@ def side():
         for idx, (bucket, label) in enumerate(zip(trust_buckets, bucket_labels)):
             with cols[idx]:
                 if st.button(f"{bucket}", key=f"bucket_{label}"):
-                    st.write(f"{label} Trust clicked")
-        
+                    st.session_state["selected_bucket"] = label
+                    st.write(f"Selected Trust Bucket: {label}")
+    
         # Save TrustBuilders Section
         st.sidebar.subheader("Save TrustBuilders®")
-        st.sidebar.text_input("Brand/Product/Person", key="brand_product")
+        brand = st.sidebar.text_input("Brand/Product/Person")
         trust_builder_text = st.sidebar.text_area("Type/paste Trust Builder®", height=100)
         
         # Allocate TrustBuilders to Buckets
@@ -733,7 +764,7 @@ def side():
         for idx, (bucket, label) in enumerate(zip(trust_buckets, bucket_labels)):
             with st.sidebar:
                 if st.button(f"{bucket}", key=f"allocate_{label}"):
-                    save_content(st.session_state["wix_user_id"], trust_builder_text, trust_bucket=label)
+                    save_trustbuilder(st.session_state["wix_user_id"], brand, trust_builder_text, trust_bucket=label)
         
         if st.sidebar.button("Let AI allocate"):
             st.write("AI allocated the Trust Bucket")