Update app.py

app.py

@@ -152,23 +152,21 @@ def merge_markdown_contents(contents):
     merged_content = "\n\n---\n\n".join(contents)
     return merged_content
 
-def upload_to_firebase(user_id, merged_md_content):
-    """
-    Upload the merged Markdown content to Firebase Storage.
-    Returns the public URL of the uploaded file.
-    """
+def upload_to_firebase(user_id, content):
     try:
-        file_path = f"knowledge_bases/{user_id}/knowledge_base.md"
-        # Upload the content as bytes
-        storage.child(file_path).put(io.BytesIO(merged_md_content.encode('utf-8')), content_type="text/markdown")
-        # Get the download URL
-        public_url = storage.child(file_path).get_url(None)
+        # Create a unique filename
+        filename = f"knowledge_bases/{user_id}/knowledge_base_{uuid.uuid4()}.md"
+        # Upload content to Firebase Storage
+        storage.child(filename).put(content.encode('utf-8'))
+        # Get the public URL
+        public_url = storage.child(filename).get_url(None)
         return public_url
     except Exception as e:
-        logger.error(f"Error uploading to Firebase Storage: {e}")
-        st.sidebar.error(f"Error uploading files: {e}")
+        st.error(f"Error uploading to Firebase: {e}")
         return None
 
+
+
 # Define and validate API keys
 openai_api_key = os.getenv("OPENAI_API_KEY")
 serper_api_key = os.getenv("SERPER_API_KEY")
@@ -408,6 +406,30 @@ def update_message_counter():
 
 
 
+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.get("brand", "").lower())
+                   and (not bucket or bucket == v.get("bucket", ""))
+            }
+            return filtered_trustbuilders
+        return {}
+    except Exception as e:
+        st.error(f"Error fetching TrustBuilders: {e}")
+        return {}
+
+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:
         tonality_id = str(uuid.uuid4())
@@ -468,25 +490,18 @@ def load_user_content(user_id):
 
 
 
-def save_content(user_id, content, trust_bucket, content_type="TrustBuilder"):
+def save_content(user_id, brand, trust_builder_text, trust_bucket=None):
     try:
-        entry_id = str(uuid.uuid4())
-        firebase_key = "TrustBuilder" if content_type.lower() == "trustbuilder" else "BrandTonality"
-
-        # Save with metadata (trust_bucket for TrustBuilders)
-        data = {"message": content}
-        if content_type == "TrustBuilder":
-            data["trust_bucket"] = trust_bucket
-
-        db.child("users").child(user_id).child(firebase_key).child(entry_id).set(data)
-
-        if firebase_key not in st.session_state:
-            st.session_state[firebase_key] = {}
-        st.session_state[firebase_key][entry_id] = data
-
-        st.success(f"{content_type.capitalize()} saved successfully!")
+        trustbuilder_id = str(uuid.uuid4())
+        data = {
+            "brand": brand,
+            "message": trust_builder_text,
+            "bucket": trust_bucket
+        }
+        db.child("users").child(user_id).child("TrustBuilder").child(trustbuilder_id).set(data)
+        st.success(f"TrustBuilder saved to {trust_bucket} bucket.")
     except Exception as e:
-        st.error(f"Error saving {content_type}: {e}")
+        st.error(f"Error saving TrustBuilder: {e}")
 
 
         
@@ -662,69 +677,78 @@ def side():
                 
         # Image paths and labels (pre-resized images in your folder)
         st.sidebar.header("TrustVault®")
-
-            # 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)
-        
+        st.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.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)
+        st.subheader("Saved Documents")
         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]}")
-                st.sidebar.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.sidebar.markdown(f"[View Full Document]({public_url})", unsafe_allow_html=True)
+                with st.expander(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)
                     # Add delete button for each document
-                    if st.button(f"Delete: {doc_id[:8]}", key=f"del_{doc_id}"):
+                    if st.button(f"Delete Document {doc_id[:8]}", key=f"del_doc_{doc_id}"):
                         delete_document(st.session_state["wix_user_id"], doc_id)
+                        st.experimental_rerun()  # Refresh the app to reflect deletion
         else:
-            st.sidebar.write("No saved documents found.")
-        
-        # File Upload Section inside the same container
-        st.sidebar.subheader("Upload New Documents")
-        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)
+            st.write("No saved documents found.")
+    
+        # 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:
+                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("users").child(st.session_state["wix_user_id"]).child("documents").child(doc_id).set({
+                            "content": merged_md_content,
+                            "public_url": public_url
+                        })
+                        st.success("Files uploaded and saved successfully!")
+                        st.markdown(f"[View your knowledge base here]({public_url})", unsafe_allow_html=True)
+                        st.experimental_rerun()  # Refresh to show the new document
+                else:
+                    st.warning("No valid files uploaded.")
             else:
-                st.sidebar.warning("No valid files uploaded.")
-        
-        st.sidebar.markdown('</div>', unsafe_allow_html=True)
-        
+                st.warning("Please select at least one file to upload.")
+    
+        st.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")
-        brand_input = st.sidebar.text_input("Brand/Product/Person", key="brand_input")
+        st.subheader("Show my TrustBuilders®")
+        search_query = st.text_input("Search by keyword", key="keyword")
+        brand_input = st.text_input("Brand/Product/Person", key="brand_input")
         bucket = st.session_state.get("selected_bucket")
-        
-        if st.sidebar.button("Show TrustBuilders"):
+    
+        if st.button("Show TrustBuilders"):
             # Fetch TrustBuilders from Firebase
             search_results = fetch_trustbuilders(
                 st.session_state["wix_user_id"],
@@ -734,44 +758,45 @@ def side():
             )
             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)
+                    with st.expander(f"TrustBuilder ID: {result_id[:8]}"):
+                        st.write(f"Brand: {result_data.get('brand', 'N/A')}")
+                        st.write(f"Bucket: {result_data.get('bucket', 'N/A')}")
+                        st.write(f"Message: {result_data['message']}")
+                        # Add delete button for each TrustBuilder
+                        if st.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.")
-        
+                st.write("No TrustBuilders found.")
+    
         # Trust Bucket Selection
-        st.sidebar.write("and choose the Trust Bucket®")
+        st.write("and choose the Trust Bucket®")
         trust_buckets = ["S", "D", "R", "B", "V", "C"]
         bucket_labels = ["Stability", "Development", "Relationship", "Benefit", "Vision", "Competence"]
-        cols = st.sidebar.columns(len(trust_buckets))
+        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.sidebar.write(f"{label} Trust clicked")
-                    st.session_state["selected_bucket"] = label
-        
+            if cols[idx].button(f"{bucket}", key=f"bucket_{label}"):
+                st.write(f"{label} Trust clicked")
+                st.session_state["selected_bucket"] = label
+    
         # Save TrustBuilders Section
-        st.sidebar.subheader("Save TrustBuilders®")
-        brand = st.sidebar.text_input("Brand/Product/Person", key="brand_product")
-        trust_builder_text = st.sidebar.text_area("Type/paste Trust Builder®", height=100)
-        
+        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.sidebar.write("Allocate to")
+        st.write("Allocate to")
+        cols_allocate = st.columns(len(trust_buckets))
         for idx, (bucket, label) in enumerate(zip(trust_buckets, bucket_labels)):
-            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"):
+            if cols_allocate[idx].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"], trust_builder_text, trust_bucket=allocated_bucket)
-                st.sidebar.success(f"AI allocated the Trust Bucket: {allocated_bucket} and saved your TrustBuilder.")
+                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.sidebar.error("AI could not allocate the Trust Bucket.")
-
+                st.error("AI could not allocate the Trust Bucket.")
 side()
 # Load knowledge base