Update app.py

app.py

@@ -350,7 +350,6 @@ def edit_all_documents(container, search_keyword=None):
     st.markdown("### 📑 All Documents" + (f" (Filtered: '{search_keyword}')" if search_keyword else ""))
     documents = get_documents(container)
     if search_keyword:
-        # Filter documents by search keyword
         documents = [doc for doc in documents if vector_keyword_search(search_keyword, doc)]
     if not documents:
         st.info("No documents match the current filter." if search_keyword else "No documents in this container.")
@@ -473,7 +472,6 @@ def new_links_record(container):
         st.error(f"Error creating links record: {message}")
 
 def vector_keyword_search(keyword, doc):
-    # Check if keyword exists in any string field of the document
     keyword = keyword.lower()
     for key, value in doc.items():
         if isinstance(value, str) and keyword in value.lower():
@@ -489,11 +487,11 @@ def search_documents_ui(container):
         with col2:
             clear_submitted = st.form_submit_button("🗑️ Clear")
         if search_submitted and keyword:
-            st.session_state.search_keyword = keyword
+            st.session_state.active_search = keyword  # Use a separate key
             st.rerun()
         if clear_submitted:
-            if 'search_keyword' in st.session_state:
-                del st.session_state.search_keyword
+            if 'active_search' in st.session_state:
+                del st.session_state.active_search
             st.rerun()
 
 def validate_and_preprocess_image(file_data, target_size=(576, 1024)):
@@ -623,8 +621,8 @@ def main():
         st.session_state.selected_database = selected_db
         st.session_state.selected_container = None
         st.session_state.current_container = None
-        if 'search_keyword' in st.session_state:
-            del st.session_state.search_keyword
+        if 'active_search' in st.session_state:
+            del st.session_state.active_search
         st.rerun()
 
     # Containers Section
@@ -645,8 +643,8 @@ def main():
         if selected_container != st.session_state.get("selected_container"):
             st.session_state.selected_container = selected_container
             st.session_state.current_container = database.get_container_client(selected_container)
-            if 'search_keyword' in st.session_state:
-                del st.session_state.search_keyword
+            if 'active_search' in st.session_state:
+                del st.session_state.active_search
             st.rerun()
 
         # Actions Section
@@ -673,7 +671,7 @@ def main():
 
     # Central Area: Editable Documents with Search Filter
     if st.session_state.current_container:
-        search_keyword = st.session_state.get('search_keyword', None)
+        search_keyword = st.session_state.get('active_search', None)
         edit_all_documents(st.session_state.current_container, search_keyword)
     else:
         st.info("Select a database and container to view and edit documents.")
@@ -689,8 +687,8 @@ def main():
         st.session_state.selected_database = None
         st.session_state.selected_container = None
         st.session_state.current_container = None
-        if 'search_keyword' in st.session_state:
-            del st.session_state.search_keyword
+        if 'active_search' in st.session_state:
+            del st.session_state.active_search
         st.rerun()
 
 if __name__ == "__main__":