Update app.py

app.py

@@ -719,10 +719,13 @@ def side():
                         st.markdown("For detailed descriptions, visit [Academy](https://www.trustifier.ai/account/academy)")
                 
               
-                
+        trust_buckets = ["S", "D", "R", "B", "V", "C"]
+        bucket_labels = ["Stability", "Development", "Relationship", "Benefit", "Vision", "Competence"]
+        
         # Image paths and labels (pre-resized images in your folder)
         st.sidebar.header("TrustVault®")
-            # CSS for scrollable container with gray border
+        
+        # CSS for scrollable container with gray border
         st.markdown("""
             <style>
             .scrollable-container {
@@ -735,15 +738,15 @@ def side():
             }
             </style>
         """, 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.")
             st.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.markdown(f"**Document ID:** {doc_id[:8]}")
@@ -756,9 +759,9 @@ def side():
                         delete_document(st.session_state["wix_user_id"], doc_id)
             else:
                 st.write("No saved documents found.")
-    
+        
             st.markdown('</div>', unsafe_allow_html=True)
-    
+        
         # File Upload Section
         st.subheader("Upload New Documents")
         with st.form(key='upload_form'):
@@ -768,7 +771,7 @@ def side():
                 accept_multiple_files=True
             )
             submit_upload = st.form_submit_button(label='Upload')
-    
+        
         if submit_upload:
             if uploaded_files:
                 md_contents = []
@@ -793,17 +796,17 @@ def side():
                     st.warning("No valid files uploaded.")
             else:
                 st.warning("Please select at least one file to upload.")
-    
+        
         # Show My TrustBuilders Section
         st.subheader("Show my TrustBuilders®")
         search_query = st.text_input("Search by keyword", key="search_query")
         brand_input = st.text_input("Brand/Product/Person", key="brand_input")
-    
-        # Trust Bucket Selection
+        
+        # Trust Bucket Selection for Filtering
         st.write("Choose the Trust Bucket®")
         bucket_options = ["All"] + bucket_labels
         selected_bucket = st.selectbox("Select Trust Bucket", options=bucket_options, key="selected_bucket")
-    
+        
         if st.button("Show TrustBuilders"):
             search_results = fetch_trustbuilders(
                 st.session_state["wix_user_id"],
@@ -819,24 +822,21 @@ def side():
                         st.write(f"Message: {result_data.get('message', '')}")
             else:
                 st.write("No TrustBuilders found.")
-          
-    
-        # Trust Bucket Selection
+        
+        # Trust Bucket Selection for Saving
         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)):
             with cols[idx]:
                 if st.button(f"{bucket}", key=f"bucket_{label}"):
                     st.write(f"Selected Trust Bucket: {label}")
-                    st.session_state["selected_bucket"] = label
-    
+                    st.session_state["selected_bucket_to_save"] = label
+        
         # Save TrustBuilders Section
         st.subheader("Save TrustBuilders®")
         brand = st.text_input("Brand/Product/Person", key="brand_product")
         trust_builder_text = st.text_area("Type/paste Trust Builder®", height=100)
-    
+        
         # Allocate TrustBuilders to Buckets
         st.write("Allocate to")
         cols_allocate = st.columns(len(trust_buckets))
@@ -844,44 +844,12 @@ def side():
             with cols_allocate[idx]:
                 if st.button(f"{bucket}", key=f"allocate_{label}"):
                     save_content(st.session_state["wix_user_id"], brand, trust_builder_text, trust_bucket=label)
-    
-        if st.button("Let AI allocate"):
-            allocated_bucket = ai_allocate_trust_bucket(trust_builder_text)
-            if allocated_bucket:
-                save_content(st.session_state["wix_user_id"], brand, trust_builder_text, trust_bucket=allocated_bucket)
-                st.success(f"AI allocated the Trust Bucket: {allocated_bucket} and saved your TrustBuilder.")
-            else:
-                st.error("AI could not allocate the Trust Bucket.")
-    
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
 
 
+# Initialize embeddings and FAISS index only when wix_user_id is set
 
 
-        
 side()
-# Load knowledge base
-
-
-# Initialize embeddings and FAISS index only when wix_user_id is set
-
 if st.session_state.get("wix_user_id") and "faiss_db" not in st.session_state:
     refresh_faiss_index()