Update app.py

app.py

@@ -432,13 +432,6 @@ def update_message_counter():
 
 
 
-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()  # Refresh the app to reflect deletion
-    except Exception as e:
-        st.error(f"Error deleting TrustBuilder: {e}")
 
 def store_brand_tonality(user_id, message):
     try:
@@ -548,14 +541,6 @@ def delete_document(user_id, doc_id):
     except Exception as e:
         st.error(f"Error deleting document: {e}")
 
-# Fetch documents from Firebase
-def delete_document(user_id, doc_id):
-    try:
-        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 deleting document: {e}")
 
 
 def side():
@@ -687,44 +672,8 @@ def side():
 
 
         st.sidebar.header("TrustVault®")
-        image_files = [
-                            ("Bucket_STABILITY.png"),
-                            ("Bucket_DEVELOPMENT.png"),
-                            ("Bucket_RELATIONSHIP.png"),
-                            ("Bucket_BENEFIT.png"),
-                            ("Bucket_VISION.png"),
-                            ("Bucket_COMPETENCE.png"),
-                        ]
-        
-        # CSS for Layout and Hover Effect
-        st.markdown(
-            """
-            <style>
-                .image-container {
-                    display: flex;
-                    justify-content: center;
-                    gap: 20px;
-                }
-                .image-wrapper {
-                    text-align: center;
-                }
-                .image-wrapper img {
-                    width: 80px;
-                    height: 80px;
-                    object-fit: contain;
-                    cursor: pointer;
-                    transition: transform 0.3s;
-                }
-                .image-wrapper img:hover {
-                    transform: scale(1.1);
-                }
-            </style>
-            """,
-            unsafe_allow_html=True,
-        )
-
-                
-        # CSS for scrollable container
+       
+        st.subheader("Saved Documents")
         st.markdown("""
             <style>
             .scrollable-container {
@@ -737,91 +686,72 @@ def side():
             }
             </style>
         """, unsafe_allow_html=True)
-    
-        # Saved Documents Section
-        with st.container():
-            st.subheader("Saved Documents")
-            st.write("Save documents like your brand tonality, key phrases, or segments here.")
-            st.markdown('<div class="scrollable-container">', unsafe_allow_html=True)
-    
-            # Replace with your document fetching logic
-            docs = {}  # Example: fetch_documents(st.session_state["user_id"])
-    
+        st.markdown('<div class="scrollable-container">', unsafe_allow_html=True)
+        
+        # Fetch and display saved documents
+        try:
+            docs = db.child("users").child(user_id).child("KnowledgeBase").get().val()
             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":
-                        st.markdown(f"[View Full Document]({public_url})", unsafe_allow_html=True)
+                    col1, col2 = st.columns([4, 1])
+                    with col1:
+                        st.write(doc_data.get("content", ""))
+                    with col2:
+                        if st.button("Delete", key=f"delete_{doc_id}"):
+                            delete_document(user_id, doc_id)
             else:
                 st.write("No saved documents found.")
-    
-            st.markdown('</div>', unsafe_allow_html=True)
-    
+        except Exception as e:
+            st.error(f"Error fetching documents: {e}")
+        
+        st.markdown('</div>', unsafe_allow_html=True)
+        
         # File Upload Section
         st.subheader("Upload New Documents")
-        with st.form(key='upload_form'):
-            uploaded_files = st.file_uploader(
-                "Select PDF, DOCX, or TXT files to upload",
-                type=["pdf", "docx", "txt"],
-                accept_multiple_files=True
-            )
-            submit_upload = st.form_submit_button(label='Upload')
-    
-        if submit_upload:
-            if uploaded_files:
-                for file in uploaded_files:
-                    # Add your file upload logic here
-                    st.success(f"File {file.name} uploaded successfully!")
-            else:
-                st.warning("Please select at least one file to upload.")
-    
-        
+        uploaded_file = st.file_uploader("Choose a file to upload (PDF, DOCX, TXT):", type=["pdf", "docx", "txt"])
+        if uploaded_file:
+            if st.button("Upload"):
+                try:
+                    doc_id = str(uuid.uuid4())
+                    db.child("users").child(user_id).child("KnowledgeBase").child(doc_id).set({
+                        "content": uploaded_file.name
+                    })
+                    st.success(f"File '{uploaded_file.name}' uploaded successfully!")
+                except Exception as e:
+                    st.error(f"Error uploading file: {e}")
         
-                
+        # Show Trust Builders Section
         st.subheader("Show My TrustBuilders®")
-        search_query = st.text_input("Search by keyword", key="search_query")
-        
-        # Dropdown for selecting Trust Bucket
+        search_query = st.text_input("Search by keyword")
         trust_buckets = ["Stability", "Development", "Relationship", "Benefit", "Vision", "Competence"]
-        selected_bucket = st.selectbox("Select Trust Bucket", trust_buckets, key="selected_bucket")
+        selected_bucket = st.selectbox("Select Trust Bucket", trust_buckets)
         
-        if st.button("Show", key="show_button"):
+        if st.button("Show"):
             st.write(f"Filtering TrustBuilders for keyword='{search_query}' and bucket='{selected_bucket}'")
-            try:
-                # Fetch TrustBuilders
-                trustbuilders = fetch_trustbuilders("user_id_placeholder")
-                results = []
-                for bucket, contents in trustbuilders.items():
-                    if selected_bucket == "All" or selected_bucket == bucket:
-                        for content in contents:
-                            if not search_query or search_query.lower() in content.lower():
-                                results.append(f"{bucket}: {content}")
-                if results:
-                    for result in results:
-                        st.markdown(f"- {result}")
-                else:
-                    st.write("No TrustBuilders found.")
-            except Exception as e:
-                st.error(f"Error fetching TrustBuilders: {e}")
+            trustbuilders = fetch_trustbuilders(user_id)
+            results = []
+            for trustbuilder_id, data in trustbuilders.items():
+                content = data.get("content", "")
+                bucket = data.get("bucket", "")
+                if (selected_bucket == bucket or not selected_bucket) and (not search_query or search_query.lower() in content.lower()):
+                    results.append(f"{bucket}: {content}")
+            if results:
+                for result in results:
+                    st.markdown(f"- {result}")
+            else:
+                st.write("No TrustBuilders found.")
         
-    
-        # Save TrustBuilders Section
+        # Save Trust Builders Section
         st.subheader("Save TrustBuilders®")
-        brand = st.text_input("Brand/Product/Person", key="brand_input_save")
-        trust_builder_text = st.text_area("Type/paste Trust Builder®", key="trust_builder_text_save", height=100)
+        brand = st.text_input("Brand/Product/Person", key="brand_input")
+        trust_builder_text = st.text_area("Type/paste Trust Builder®")
+        selected_save_bucket = st.selectbox("Choose Trust Bucket to Save", trust_buckets, key="save_bucket")
         
-        # Dropdown for selecting Trust Bucket to save into
-        selected_save_bucket = st.selectbox("Choose Trust Bucket to Save", trust_buckets[1:], key="save_bucket")
-        
-        # Save button
-        if st.button("Save TrustBuilder", key="save_button"):
-            if selected_save_bucket and trust_builder_text.strip():
-                save_trustbuilder("user_id_placeholder", trust_builder_text.strip(), selected_save_bucket)
+        if st.button("Save"):
+            if trust_builder_text.strip() and selected_save_bucket:
+                save_trustbuilder(user_id, trust_builder_text.strip(), selected_save_bucket)
             else:
-                st.warning("Please fill in all fields before saving.")
-
+                st.warning("Please fill all fields before saving.")