Update app.py

app.py

@@ -52,7 +52,6 @@ import re
 # Set up logging to suppress Streamlit warnings about experimental functions
 logging.getLogger('streamlit').setLevel(logging.ERROR)
 
-st.session_state["documents"] = {}
 if "chat_history" not in st.session_state:
     st.session_state["chat_history"] = []
 if "documents" not in st.session_state:
@@ -720,7 +719,7 @@ def side():
         if "documents" not in st.session_state:
             try:
                 docs = db.child("users").child(st.session_state["wix_user_id"]).child("KnowledgeBase").get().val()
-                st.write("Fetched documents:", docs)  # Debugging output
+                st.sidebar.write("Fetched documents:", docs)  # Debugging output
 
                 st.session_state["documents"] = docs if docs else {}
             except Exception as e:
@@ -743,24 +742,43 @@ def side():
             disabled=True
         )
         
+        # File uploader
         # File uploader
         uploaded_file = st.file_uploader("", type=["pdf", "docx", "txt"], key="file_uploader", label_visibility="collapsed")
         if st.sidebar.button("Upload", key="upload_button"):
             if uploaded_file:
                 try:
-                    content, _= upload_to_firebase(st.session_state["wix_user_id"], uploaded_file)
+                    content, _ = upload_to_firebase(st.session_state["wix_user_id"], uploaded_file)
                     if content:
                         doc_id = str(uuid.uuid4())
+                        
+                        # Save to Firebase
                         db.child("users").child(st.session_state["wix_user_id"]).child("KnowledgeBase").child(doc_id).set(
                             {"content": content, "name": uploaded_file.name}
                         )
+        
+                        # Update session state
                         if "documents" not in st.session_state:
                             st.session_state["documents"] = {}
                         st.session_state["documents"][doc_id] = {"content": content, "name": uploaded_file.name}
-                        
-                        refresh_faiss_index()
+        
+                        # Refresh the text area dynamically
+                        saved_docs = "\n".join(
+                            [
+                                f"{doc_data.get('content', f'Document {doc_id[:8]}')}"
+                                for doc_id, doc_data in st.session_state["documents"].items()
+                            ]
+                        )
+                        st.sidebar.text_area(
+                            label="",
+                            value=saved_docs,
+                            height=150,
+                            key="saved_documents_text_area",
+                            disabled=True
+                        )
+        
+                        # Show success message without rerunning the app
                         st.sidebar.success(f"Document '{uploaded_file.name}' uploaded successfully!")
-                        st.rerun()
                     else:
                         st.sidebar.warning("File upload failed.")
                 except Exception as e:
@@ -768,6 +786,7 @@ def side():
             else:
                 st.sidebar.warning("Please select a file to upload.")
         
+                
         # Delete Button
         if st.session_state["documents"]:
             selected_doc_to_delete = st.selectbox(
@@ -780,8 +799,23 @@ def side():
                 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, None)
+        
+                    # Update the text area dynamically after deletion
+                    saved_docs = "\n".join(
+                        [
+                            f"{doc_data.get('content', f'Document {doc_id[:8]}')}"
+                            for doc_id, doc_data in st.session_state["documents"].items()
+                        ]
+                    )
+                    st.sidebar.text_area(
+                        label="",
+                        value=saved_docs,
+                        height=150,
+                        key="saved_documents_text_area",
+                        disabled=True
+                    )
+        
                     st.sidebar.success("Document deleted successfully!")
-                    st.rerun()
                 except Exception as e:
                     st.sidebar.error(f"Error deleting document: {e}")