Update app.py

app.py

@@ -504,50 +504,34 @@ 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 {}
 
-def delete_document(user_id, doc_id):
-    try:
-        db.child("users").child(user_id).child("documents").child(doc_id).remove()
-        st.sidebar.success("Document deleted successfully.")
-    except Exception as e:
-        st.sidebar.error(f"Error deleting document: {e}")
 
-def save_trustbuilder(user_id, brand, trust_builder, trust_bucket):
+# Function to delete a document from Firebase
+def delete_document(user_id, doc_id):
+    """
+    Deletes a document from Firebase.
+    """
     try:
-        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.")
+        db.child("users").child(user_id).child("KnowledgeBase").child(doc_id).remove()
+        st.success("Document deleted successfully!")
+        st.rerun()  # Refresh the list after deletion
     except Exception as e:
-        st.sidebar.error(f"Error saving TrustBuilder: {e}")
+        st.error(f"Error deleting document: {e}")
 
-def fetch_trustbuilders(user_id, keyword=None, brand=None, bucket=None):
+# Fetch documents from Firebase
+def fetch_documents(user_id):
+    """
+    Fetches all saved documents from Firebase.
+    """
     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 {}
+        docs = db.child("users").child(user_id).child("KnowledgeBase").get().val()
+        return docs if docs else {}
     except Exception as e:
-        st.sidebar.error(f"Error fetching TrustBuilders: {e}")
+        st.error(f"Error fetching documents: {e}")
         return {}
 
 
+
 def side():
     with st.sidebar:
 
@@ -678,27 +662,12 @@ 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]}")
@@ -706,13 +675,16 @@ 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)
-                    if st.button(f"Delete {doc_id[:8]}", key=f"del_{doc_id}"):
-                        delete_document(st.session_state["wix_user_id"], doc_id)
+        
+                    # 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)
         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)
@@ -722,27 +694,25 @@ 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!")
-        with col2:
-            if st.button("Delete"):
-                st.warning("Select a document to delete from the list above.")
-    
+                        st.sidebar.markdown(f"[View your knowledge base here]({public_url})", unsafe_allow_html=True)
+                else:
+                    st.sidebar.warning("No valid files uploaded.")
+        
         # Show My TrustBuilders Section
         st.sidebar.subheader("Show my TrustBuilders®")
         search_query = st.sidebar.text_input("Search by keyword", key="keyword")
-        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)
+        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()}
             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"]
@@ -751,12 +721,11 @@ 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.session_state["selected_bucket"] = label
-                    st.write(f"Selected Trust Bucket: {label}")
-    
+                    st.write(f"{label} Trust clicked")
+        
         # Save TrustBuilders Section
         st.sidebar.subheader("Save TrustBuilders®")
-        brand = st.sidebar.text_input("Brand/Product/Person")
+        st.sidebar.text_input("Brand/Product/Person", key="brand_product")
         trust_builder_text = st.sidebar.text_area("Type/paste Trust Builder®", height=100)
         
         # Allocate TrustBuilders to Buckets
@@ -764,7 +733,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_trustbuilder(st.session_state["wix_user_id"], brand, trust_builder_text, trust_bucket=label)
+                    save_content(st.session_state["wix_user_id"], trust_builder_text, trust_bucket=label)
         
         if st.sidebar.button("Let AI allocate"):
             st.write("AI allocated the Trust Bucket")