Update app.py

app.py

@@ -12,6 +12,8 @@ if 'selected_file' not in st.session_state:
     st.session_state.selected_file = None
 if 'view_mode' not in st.session_state:
     st.session_state.view_mode = 'view'
+if 'files' not in st.session_state:
+    st.session_state.files = []
 
 # Define the markdown variables
 Boxing_and_MMA_Commentary_and_Knowledge = """
@@ -171,6 +173,7 @@ def file_management_sidebar():
     # Get list of .md files excluding README.md
     md_files = [file for file in glob.glob("*.md") if os.path.basename(file).lower() != 'readme.md']
     md_files.sort()
+    st.session_state.files = md_files  # Update session state
 
     if md_files:
         st.sidebar.markdown("### Markdown Files")
@@ -184,12 +187,10 @@ def file_management_sidebar():
                 if st.sidebar.button("📄", key=f"view_{key_base}"):
                     st.session_state.selected_file = file
                     st.session_state.view_mode = 'view'
-                    st.experimental_rerun()
             with col3:
                 if st.sidebar.button("✏️", key=f"edit_{key_base}"):
                     st.session_state.selected_file = file
                     st.session_state.view_mode = 'edit'
-                    st.experimental_rerun()
         # Option to create a new markdown file
         if st.sidebar.button("Create New Markdown File"):
             # Generate automatic filename
@@ -200,7 +201,6 @@ def file_management_sidebar():
             st.sidebar.success(f"Created new file: {new_filename}")
             st.session_state.selected_file = new_filename
             st.session_state.view_mode = 'edit'
-            st.experimental_rerun()
     else:
         st.sidebar.write("No markdown files found.")
         if st.sidebar.button("Create New Markdown File"):
@@ -212,7 +212,6 @@ def file_management_sidebar():
             st.sidebar.success(f"Created new file: {new_filename}")
             st.session_state.selected_file = new_filename
             st.session_state.view_mode = 'edit'
-            st.experimental_rerun()
 
 # Main application logic
 def main():
@@ -251,8 +250,9 @@ def main():
                 st.write(f"Generated file **{filename}** with AI lookup results.")
                 # Clear the query parameters
                 st.experimental_set_query_params()
-                # Update file management sidebar
-                st.experimental_rerun()
+                # Update session state
+                st.session_state.selected_file = filename
+                st.session_state.view_mode = 'view'
     except Exception as e:
         st.write(f"An error occurred while processing query parameters: {e}")
 
@@ -267,7 +267,6 @@ def main():
             elif action == 'clear':
                 # Clear query parameters
                 st.experimental_set_query_params()
-                st.experimental_rerun()
     if 'query' in params:
         query_list = params['query']
         if query_list:
@@ -298,7 +297,8 @@ def main():
                     with open(selected_file, 'w', encoding='utf-8') as f:
                         f.write(edited_content)
                     st.success(f"Changes saved to {selected_file}")
-                    st.experimental_rerun()
+                    # Update the file list in session state
+                    st.session_state.files = [file for file in glob.glob("*.md") if os.path.basename(file).lower() != 'readme.md']
         else:
             st.error("Selected file does not exist.")
     else: