Update app.py

app.py

@@ -683,6 +683,9 @@ def side():
         
 
         st.header("TrustVault®")
+        st.sidebar.subheader("Saved Documents")
+    
+        # Add custom styles for buttons and scrollable container
         st.sidebar.markdown("""
             <style>
             .scrollable-container {
@@ -697,30 +700,39 @@ def side():
             .button-container {
                 display: flex;
                 justify-content: space-between;
-                margin-top: 10px;
             }
             .custom-upload-button {
                 display: inline-block;
                 font-size: 14px;
-                font-weight: 600;
+                font-weight: bold;
                 color: white;
                 background-color: #007bff;
                 padding: 8px 16px;
                 border-radius: 5px;
                 cursor: pointer;
                 text-align: center;
-                margin-right: 10px;
+                border: none;
             }
             .custom-upload-button:hover {
                 background-color: #0056b3;
             }
-            .hidden-file-uploader {
-                display: none;
+            .custom-delete-button {
+                background-color: #dc3545;
+                color: white;
+                padding: 8px 16px;
+                font-size: 14px;
+                border: none;
+                border-radius: 5px;
+                cursor: pointer;
+                font-weight: bold;
+            }
+            .custom-delete-button:hover {
+                background-color: #a71d2a;
             }
             </style>
         """, unsafe_allow_html=True)
-        
-        # Saved Documents Section
+    
+        # Fetch documents from Firebase
         if "documents" not in st.session_state:
             try:
                 docs = db.child("users").child(st.session_state["wix_user_id"]).child("KnowledgeBase").get().val()
@@ -728,9 +740,8 @@ def side():
             except Exception as e:
                 st.sidebar.error(f"Error fetching documents: {e}")
                 st.session_state["documents"] = {}
-        
-        st.sidebar.subheader("Saved Documents")
-        
+    
+        # Display saved documents
         saved_docs = ""
         if st.session_state["documents"]:
             for doc_id, doc_data in st.session_state["documents"].items():
@@ -738,63 +749,46 @@ def side():
                 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."
-        
-        # Display documents inside a text area (scrollable)
+    
+        # Scrollable text area
         saved_docs_display = st.sidebar.text_area(
             label="",
             value=saved_docs,
             height=150,
             key="saved_documents_text_area",
-            disabled=True  # To prevent editing of the text area
+            disabled=True
         )
-        
-        # Upload and Delete Buttons
+    
+        # File uploader and delete button in the same row
         st.sidebar.markdown('<div class="button-container">', unsafe_allow_html=True)
-        
-        # Custom File Upload Button
-        uploaded_file = st.file_uploader("", key="hidden_file_uploader", label_visibility="collapsed")
+    
+        # File uploader button (customized)
+        uploaded_file = st.file_uploader("", type=["pdf", "docx", "txt"], key="hidden_file_uploader", label_visibility="collapsed")
         upload_button_html = f"""
-            <label for="hidden_file_uploader" class="custom-upload-button">Upload</label>
+            <button class="custom-upload-button" onclick="document.querySelector('input[type=file]').click();">Upload</button>
         """
         st.sidebar.markdown(upload_button_html, unsafe_allow_html=True)
-        
-        if uploaded_file:
-            try:
-                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.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}")
-        
-        # Delete Button and Selector
+    
+        # Delete Button
         if st.session_state["documents"]:
-            with st.sidebar:
-                selected_doc_to_delete = st.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}"),
-                    key="delete_doc_selector"
-                )
-                if st.button("Delete", key="delete_button"):
-                    try:
-                        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 document: {e}")
-        
-        st.sidebar.markdown('</div>', unsafe_allow_html=True)
-
-
+            selected_doc_to_delete = st.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}"),
+                key="delete_doc_selector"
+            )
+            if st.sidebar.button("Delete", key="delete_button"):
+                try:
+                    delete_document(st.session_state["wix_user_id"], selected_doc_to_delete)
+                    st.session_state["documents"].pop(selected_doc_to_delete)
+                    st.rerun()
+                except Exception as e:
+                    st.sidebar.error(f"Error deleting document: {e}")
+    
+        st.sidebar.markdown("</div>", unsafe_allow_html=True)
+    
+    
+