Update app.py

app.py

@@ -554,7 +554,6 @@ def delete_document(user_id, doc_id):
         st.error(f"Error deleting document: {e}")
 
 
-
 def side():
     with st.sidebar:
 
@@ -681,7 +680,7 @@ def side():
                         )
                         st.markdown("For detailed descriptions, visit [Academy](https://www.trustifier.ai/account/academy)")
 
-
+        
 
         st.header("TrustVault®")
         st.sidebar.markdown("""
@@ -704,61 +703,75 @@ def side():
         """, unsafe_allow_html=True)
         
         # Saved Documents Section
+        if "documents" not in st.session_state:
+            try:
+                docs = db.child("users").child(st.session_state["wix_user_id"]).child("KnowledgeBase").get().val()
+                st.session_state["documents"] = docs if docs else {}
+            except Exception as e:
+                st.sidebar.error(f"Error fetching documents: {e}")
+                st.session_state["documents"] = {}
+
         st.sidebar.subheader("Saved Documents")
+        saved_docs = ""
+        if st.session_state["documents"]:
+            for doc_id, doc_data in st.session_state["documents"].items():
+                doc_name = doc_data.get("content", f"Document {doc_id[:8]}")
+                saved_docs += f"- {doc_name}\n"
+        else:
+            saved_docs = "Save documents like your brand tonality, key phrases, or segments here and they will show here."
+        # Scrollable text area
+        st.sidebar.markdown(
+            f"""
+            <style>
+            .scrollable-text-area {{
+                max-height: 150px;
+                overflow-y: auto;
+                border: 1px solid gray;
+                border-radius: 5px;
+                background-color: #f9f9f9;
+                padding: 10px;
+                font-size: 14px;
+                color: #555;
+            }}
+            </style>
+            <div class="scrollable-text-area">{saved_docs}</div>
+            """,
+            unsafe_allow_html=True,)
         
-        # Text area for displaying saved documents
-        doc_texts = ""
-        try:
-            docs = db.child("users").child(st.session_state["wix_user_id"]).child("KnowledgeBase").get().val()
-            if docs:
-                for doc_id, doc_data in docs.items():
-                    doc_name = doc_data.get("content", f"Document {doc_id[:8]}")
-                    doc_texts += f"{doc_name}\n"
-            else:
-                doc_texts = "No saved documents found."
-        except Exception as e:
-            st.sidebar.error(f"Error fetching documents: {e}")
-        
-        savedDoc = st.sidebar.text_area("", doc_texts, height=200)
-        
-        # Button container for Upload and Delete actions
-        st.sidebar.markdown('<div class="button-container">', unsafe_allow_html=True)
-        
-        # File uploader
+        # File uploader and delete buttons
         uploaded_file = st.sidebar.file_uploader("", key="file_uploader", label_visibility="collapsed")
-        
-        # Upload button
         if st.sidebar.button("Upload", key="upload_button"):
             if uploaded_file:
                 try:
-                    # Convert and upload the file
                     file_content = convert_file_to_md(uploaded_file)
                     if file_content:
                         doc_id = str(uuid.uuid4())
-                        db.child("users").child(st.session_state["wix_user_id"]).child("KnowledgeBase").child(doc_id).set({"content": file_content})
-                        st.sidebar.success(f"Document uploaded successfully!")
-                        st.rerun()  # Refresh the display
+                        db.child("users").child(st.session_state["wix_user_id"]).child("KnowledgeBase").child(doc_id).set(
+                            {"content": file_content}
+                        )
+                        st.session_state["documents"][doc_id] = {"content": file_content}
+                        st.sidebar.success(f"Document '{uploaded_file.name}' uploaded successfully!")
+                        st.experimental_rerun()
                     else:
                         st.sidebar.warning("Failed to process the uploaded file.")
                 except Exception as e:
                     st.sidebar.error(f"Error uploading document: {e}")
-            else:
-                st.sidebar.warning("Please select a file to upload.")
         
-        # Delete selected documents logic
-        if docs:
-            delete_options = [doc_data.get("content", f"Document {doc_id[:8]}") for doc_id, doc_data in docs.items()]
-            selected_docs = st.sidebar.multiselect("Select documents to delete", options=delete_options)
-            if st.sidebar.button("Delete Selected", key="delete_button"):
+        # Delete documents functionality
+        if st.session_state["documents"]:
+            selected_doc_to_delete = st.sidebar.selectbox(
+                "Select document to delete", 
+                options=list(st.session_state["documents"].keys()), 
+                format_func=lambda x: st.session_state["documents"][x].get("content", f"Document {x}")
+            )
+            if st.sidebar.button("Delete", key="delete_button"):
                 try:
-                    for doc_name in selected_docs:
-                        for doc_id, doc_data in docs.items():
-                            if doc_data.get("content") == doc_name:
-                                db.child("users").child(st.session_state["wix_user_id"]).child("KnowledgeBase").child(doc_id).remove()
-                    st.sidebar.success("Selected documents deleted successfully!")
-                    st.rerun()  # Refresh the display
+                    db.child("users").child(st.session_state["wix_user_id"]).child("KnowledgeBase").child(selected_doc_to_delete).remove()
+                    st.session_state["documents"].pop(selected_doc_to_delete)
+                    st.sidebar.success("Document deleted successfully!")
+                    st.experimental_rerun()
                 except Exception as e:
-                    st.sidebar.error(f"Error deleting documents: {e}")
+                    st.sidebar.error(f"Error deleting document: {e}")
         
         st.sidebar.markdown('</div>', unsafe_allow_html=True)