Update app.py

app.py

@@ -663,11 +663,26 @@ def side():
         # Image paths and labels (pre-resized images in your folder)
         st.sidebar.header("TrustVault®")
 
-    # Saved Documents Section
+            # Saved Documents Section
+        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)
+        
+        # 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)
         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,43 +690,56 @@ 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)
+                        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
+        # File Upload Section inside the same container
         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)
-            if uploaded_files:
-                md_contents = [convert_file_to_md(file) for file in uploaded_files if convert_file_to_md(file)]
-                if md_contents:
-                    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.")
+        uploaded_files = st.sidebar.file_uploader("Upload PDF, DOCX, or TXT files", type=["pdf", "docx", "txt"], accept_multiple_files=True)
+        if uploaded_files:
+            md_contents = [convert_file_to_md(file) for file in uploaded_files if convert_file_to_md(file)]
+            if md_contents:
+                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.")
+        
+        st.sidebar.markdown('</div>', unsafe_allow_html=True)
         
         # 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()}
-            for result_id, result_data in search_results.items():
-                st.sidebar.write(f"- {result_data['message'][:50]}")
+        brand_input = st.sidebar.text_input("Brand/Product/Person", key="brand_input")
+        bucket = st.session_state.get("selected_bucket")
+        
+        if st.sidebar.button("Show TrustBuilders"):
+            # Fetch TrustBuilders from Firebase
+            search_results = fetch_trustbuilders(
+                st.session_state["wix_user_id"],
+                keyword=search_query,
+                brand=brand_input,
+                bucket=bucket
+            )
+            if search_results:
+                for result_id, result_data in search_results.items():
+                    st.sidebar.write(f"- {result_data['message'][:50]}")
+                    # Optionally, add a delete button for each TrustBuilder
+                    if st.sidebar.button(f"Delete TrustBuilder {result_id[:8]}", key=f"del_tb_{result_id}"):
+                        delete_trustbuilder(st.session_state["wix_user_id"], result_id)
+            else:
+                st.sidebar.write("No TrustBuilders found.")
         
         # Trust Bucket Selection
         st.sidebar.write("and choose the Trust Bucket®")
@@ -721,24 +749,28 @@ 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.sidebar.write(f"{label} Trust clicked")
+                    st.session_state["selected_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", key="brand_product")
         trust_builder_text = st.sidebar.text_area("Type/paste Trust Builder®", height=100)
         
         # Allocate TrustBuilders to Buckets
         st.sidebar.write("Allocate to")
         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)
+            if st.sidebar.button(f"{bucket}", key=f"allocate_{label}"):
+                save_content(st.session_state["wix_user_id"], trust_builder_text, trust_bucket=label)
+                st.sidebar.success(f"TrustBuilder saved to {label} bucket.")
         
         if st.sidebar.button("Let AI allocate"):
-            st.write("AI allocated the Trust Bucket")
-
-
+            allocated_bucket = ai_allocate_trust_bucket(trust_builder_text)
+            if allocated_bucket:
+                save_content(st.session_state["wix_user_id"], trust_builder_text, trust_bucket=allocated_bucket)
+                st.sidebar.success(f"AI allocated the Trust Bucket: {allocated_bucket} and saved your TrustBuilder.")
+            else:
+                st.sidebar.error("AI could not allocate the Trust Bucket.")
 
 side()
 # Load knowledge base