Update app.py

app.py

@@ -512,9 +512,6 @@ def preprocess_text(text):
 
 
 
-
-
-
 def load_file_content(file_path):
     """Load and return file content with error handling"""
     try:
@@ -560,42 +557,58 @@ def display_file_viewer(file_path):
 
 def display_file_editor(file_path):
     """Display file content in editor with save functionality"""
-    content = load_file_content(file_path)
-    if content:
-        st.markdown("### ✏️ File Editor")
-        st.markdown(f"**Editing:** {file_path}")
-        
-        # Create a unique key for the text area
-        editor_key = f"editor_{hash(file_path)}"
-        
-        # Editor with syntax highlighting for markdown
-        new_content = st.text_area(
-            "Edit content below:",
-            value=content,
-            height=400,
-            key=editor_key
+    # Initialize file content in session state if not already present
+    if 'file_content' not in st.session_state:
+        st.session_state.file_content = {}
+    
+    # Load content if not in session state or if it's a different file
+    if file_path not in st.session_state.file_content:
+        content = load_file_content(file_path)
+        if content is not None:
+            st.session_state.file_content[file_path] = content
+        else:
+            return
+    
+    st.markdown("### ✏️ File Editor")
+    st.markdown(f"**Editing:** {file_path}")
+    
+    # Create a unique key for the text area
+    editor_key = f"editor_{hash(file_path)}"
+    
+    # Editor with syntax highlighting for markdown
+    new_content = st.text_area(
+        "Edit content below:",
+        value=st.session_state.file_content[file_path],
+        height=400,
+        key=editor_key
+    )
+    
+    col1, col2 = st.columns([1, 5])
+    with col1:
+        if st.button("💾 Save Changes"):
+            if save_file_content(file_path, new_content):
+                st.session_state.file_content[file_path] = new_content
+                st.success("File saved successfully! 🎉")
+                time.sleep(1)
+                st.rerun()
+    
+    with col2:
+        st.download_button(
+            label="⬇️ Download File",
+            data=new_content,
+            file_name=os.path.basename(file_path),
+            mime="text/markdown"
         )
-        
-        col1, col2 = st.columns([1, 5])
-        with col1:
-            if st.button("💾 Save Changes"):
-                if save_file_content(file_path, new_content):
-                    st.success("File saved successfully! 🎉")
-                    # Update the file content in session state
-                    st.session_state.file_content = new_content
-                    time.sleep(1)  # Give user time to see success message
-                    st.rerun()
-        
-        with col2:
-            st.download_button(
-                label="⬇️ Download File",
-                data=new_content,
-                file_name=os.path.basename(file_path),
-                mime="text/markdown"
-            )
 
-# Update the file management section in main()
 def update_file_management_section():
+    # Initialize session state variables
+    if 'file_view_mode' not in st.session_state:
+        st.session_state.file_view_mode = None
+    if 'current_file' not in st.session_state:
+        st.session_state.current_file = None
+    if 'file_content' not in st.session_state:
+        st.session_state.file_content = {}
+
     all_files = glob.glob("*.md")
     all_files.sort(reverse=True)
 
@@ -605,18 +618,15 @@ def update_file_management_section():
     if st.sidebar.button("🗑 Delete All Files"):
         for file in all_files:
             os.remove(file)
+        st.session_state.file_content = {}  # Clear the file content cache
+        st.session_state.current_file = None
+        st.session_state.file_view_mode = None
         st.rerun()
 
     if st.sidebar.button("⬇️ Download All Files"):
         zip_file = create_zip_of_files(all_files)
         st.sidebar.markdown(get_download_link(zip_file), unsafe_allow_html=True)
 
-    # Initialize session state for view/edit mode if not exists
-    if 'file_view_mode' not in st.session_state:
-        st.session_state.file_view_mode = None
-    if 'current_file' not in st.session_state:
-        st.session_state.current_file = None
-
     # Display files in sidebar with action buttons
     for file in all_files:
         col1, col2, col3, col4 = st.sidebar.columns([1,3,1,1])
@@ -625,6 +635,10 @@ def update_file_management_section():
             if st.button("🌐", key=f"view_{file}"):
                 st.session_state.current_file = file
                 st.session_state.file_view_mode = 'view'
+                if file not in st.session_state.file_content:
+                    content = load_file_content(file)
+                    if content is not None:
+                        st.session_state.file_content[file] = content
                 st.rerun()
                 
         with col2:
@@ -634,11 +648,20 @@ def update_file_management_section():
             if st.button("📂", key=f"edit_{file}"):
                 st.session_state.current_file = file
                 st.session_state.file_view_mode = 'edit'
+                if file not in st.session_state.file_content:
+                    content = load_file_content(file)
+                    if content is not None:
+                        st.session_state.file_content[file] = content
                 st.rerun()
                 
         with col4:
             if st.button("🗑", key=f"delete_{file}"):
                 os.remove(file)
+                if file in st.session_state.file_content:
+                    del st.session_state.file_content[file]
+                if st.session_state.current_file == file:
+                    st.session_state.current_file = None
+                    st.session_state.file_view_mode = None
                 st.rerun()
 
     # Display viewer or editor in main area based on mode
@@ -650,8 +673,6 @@ def update_file_management_section():
 
 
 
-
-
                 
 # 🎭 Main function - "All the world's a stage, and all the code merely players" -Shakespeare, probably
 def main():