Update app.py

app.py

@@ -674,51 +674,51 @@ def side():
         st.sidebar.header("TrustVault®")
        
         st.subheader("Saved Documents")
-        st.markdown("""
+        st.sidebar.markdown("""
             <style>
             .scrollable-container {
-                max-height: 300px;
+                max-height: 200px;
                 overflow-y: auto;
                 border: 1px solid gray;
                 padding: 10px;
                 border-radius: 5px;
                 background-color: #f9f9f9;
             }
+            .button-container {
+                display: flex;
+                justify-content: flex-end;
+                margin-top: 10px;
+            }
             </style>
         """, unsafe_allow_html=True)
-        st.markdown('<div class="scrollable-container">', unsafe_allow_html=True)
         
-        # Fetch and display saved documents
+        # Scrollable container for saved documents
+        st.sidebar.markdown('<div class="scrollable-container">', unsafe_allow_html=True)
         try:
             docs = db.child("users").child(user_id).child("KnowledgeBase").get().val()
             if docs:
                 for doc_id, doc_data in docs.items():
-                    col1, col2 = st.columns([4, 1])
-                    with col1:
-                        st.write(doc_data.get("content", ""))
-                    with col2:
-                        if st.button("Delete", key=f"delete_{doc_id}"):
-                            delete_document(user_id, doc_id)
+                    st.sidebar.markdown(f"**{doc_data.get('content', 'Document')}**")
+                    if st.sidebar.button("Delete", key=f"delete_{doc_id}"):
+                        delete_document(user_id, doc_id)
             else:
-                st.write("No saved documents found.")
+                st.sidebar.write("No saved documents found.")
         except Exception as e:
-            st.error(f"Error fetching documents: {e}")
+            st.sidebar.error(f"Error fetching documents: {e}")
+        st.sidebar.markdown('</div>', unsafe_allow_html=True)
         
-        st.markdown('</div>', unsafe_allow_html=True)
+        # Buttons aligned beneath the text area
+        st.sidebar.markdown('<div class="button-container">', unsafe_allow_html=True)
+        uploaded_file = st.sidebar.file_uploader("", label_visibility="collapsed", type=["pdf", "docx", "txt"])
         
-        # File Upload Section
-        st.subheader("Upload New Documents")
-        uploaded_file = st.file_uploader("Choose a file to upload (PDF, DOCX, TXT):", type=["pdf", "docx", "txt"])
-        if uploaded_file:
-            if st.button("Upload"):
-                try:
-                    doc_id = str(uuid.uuid4())
-                    db.child("users").child(user_id).child("KnowledgeBase").child(doc_id).set({
-                        "content": uploaded_file.name
-                    })
-                    st.success(f"File '{uploaded_file.name}' uploaded successfully!")
-                except Exception as e:
-                    st.error(f"Error uploading file: {e}")
+        # Compact Upload Button
+        if st.sidebar.button("Upload"):
+            if uploaded_file:
+                save_document(user_id, uploaded_file.name)
+            else:
+                st.sidebar.warning("Please select a file to upload.")
+        st.sidebar.markdown('</div>', unsafe_allow_html=True)
+
         
         # Show Trust Builders Section
         st.subheader("Show My TrustBuilders®")