Update app.py

app.py

@@ -570,30 +570,41 @@ def compare_and_delete_files(files):
 # Function to get file size
 def get_file_size(file_path):
     return os.path.getsize(file_path)
-    
+
 def FileSidebar():
-    # File Sidebar for files 🌐View, 📂Edit, and 🗑Delete per file
-    all_files = glob.glob("*.md") + glob.glob("*_abstract.html") + glob.glob("*_abstract.pdf")
-    all_files = [file for file in all_files if os.path.basename(file) and len(os.path.splitext(file)[0]) >= 10]
-    all_files.sort(key=lambda x: (os.path.splitext(x)[1], x), reverse=True)
+    # Initialize session state variables if they don't exist
+    if 'files_to_display' not in st.session_state:
+        st.session_state.files_to_display = []
+    if 'search_result' not in st.session_state:
+        st.session_state.search_result = None
+    if 'sidebar_needs_update' not in st.session_state:
+        st.session_state.sidebar_needs_update = False
+
+    # Check if sidebar needs update
+    if st.session_state.sidebar_needs_update:
+        st.session_state.files_to_display = glob.glob("*.md") + glob.glob("*_abstract.html") + glob.glob("*_abstract.pdf")
+        st.session_state.files_to_display = [file for file in st.session_state.files_to_display if os.path.basename(file) and len(os.path.splitext(file)[0]) >= 10]
+        st.session_state.files_to_display.sort(key=lambda x: (os.path.splitext(x)[1], x), reverse=True)
+        st.session_state.sidebar_needs_update = False
 
     # ⬇️ Download
     Files1, Files2 = st.sidebar.columns(2)
     with Files1:
         if st.sidebar.button("🗑 Delete All"):
-            for file in all_files:
+            for file in st.session_state.files_to_display:
                 try:
                     os.remove(file)
                 except Exception as e:
                     st.sidebar.error(f"Error deleting {file}: {str(e)}")
+            st.session_state.sidebar_needs_update = True
             st.rerun()
     with Files2:
         if st.sidebar.button("⬇️ Download"):
-            zip_file = create_zip_of_files(all_files)
+            zip_file = create_zip_of_files(st.session_state.files_to_display)
             st.sidebar.markdown(get_zip_download_link(zip_file), unsafe_allow_html=True)
 
     # Add files 🌐View, 📂Edit, and 🗑Delete per file
-    for file in all_files:
+    for file in st.session_state.files_to_display:
         timestamp = time.strftime("%H%M%S%f")[:-4]  # HHMMSSNN format
         
         col1, col2, col3, col4 = st.sidebar.columns([3,1,1,1])
@@ -612,9 +623,10 @@ def FileSidebar():
                 try:
                     os.remove(file)
                     st.sidebar.success(f"Deleted {file}")
+                    st.session_state.sidebar_needs_update = True
+                    st.rerun()
                 except Exception as e:
                     st.sidebar.error(f"Error deleting {file}: {str(e)}")
-                st.rerun()
 
     # Display content in main area
     if 'current_file' in st.session_state and 'action' in st.session_state:
@@ -624,10 +636,16 @@ def FileSidebar():
             edit_file(st.session_state.current_file)
 
     # Show the result MD from the search
-    if 'search_result' in st.session_state:
+    if st.session_state.search_result:
         st.markdown("## Search Result")
         st.markdown(st.session_state.search_result)
+        st.session_state.search_result = None  # Clear the search result after displaying
 
+# Function to be called after search completes
+def update_sidebar_after_search(search_result):
+    st.session_state.search_result = search_result
+    st.session_state.sidebar_needs_update = True
+    
 # Function to be called after search completes
 def redraw_sidebar_after_search(search_result):
     st.session_state.search_result = search_result