Update app.py

app.py

@@ -67,27 +67,37 @@ def get_download_link(file):
     return href
 
 def show_file_operations(file_path):
-    st.write(f"File: {os.path.basename(file_path)}")
+    # Unique keys for each file based on their path
+    edit_key = f"edit_{file_path}"
+    content_key = f"content_{file_path}"
 
-    # Edit button and text area for file content
-    if st.button(f"✏️ Edit {os.path.basename(file_path)}"):
-        with open(file_path, "r") as f:
-            file_content = f.read()
-        file_content = st.text_area("Edit the file content:", value=file_content, height=250)
+    # Start Edit operation
+    if st.button(f"✏️ Edit {os.path.basename(file_path)}", key=edit_key):
+        if edit_key not in st.session_state:
+            with open(file_path, "r") as f:
+                st.session_state[content_key] = f.read()
+        st.session_state[edit_key] = True
+
+    # Display text area for editing if in edit mode
+    if st.session_state.get(edit_key, False):
+        edited_content = st.text_area("Edit the file content:", value=st.session_state.get(content_key, ""), height=250, key=content_key)
 
         # Save button
         if st.button(f"💾 Save {os.path.basename(file_path)}"):
             with open(file_path, "w") as f:
-                f.write(file_content)
+                f.write(edited_content)
             new_file_size = os.path.getsize(file_path)
             download_link = get_download_link(file_path)
             st.markdown(f"✅ File **{os.path.basename(file_path)}** saved! ([{download_link}]) - New size: {new_file_size} bytes", unsafe_allow_html=True)
+            st.session_state[edit_key] = False  # Exit edit mode
 
     # Delete button
     if st.button(f"🗑️ Delete {os.path.basename(file_path)}"):
         os.remove(file_path)
         st.markdown(f"🎉 File {os.path.basename(file_path)} deleted!")
-
+        # Remove state variables related to the deleted file
+        st.session_state.pop(edit_key, None)
+        st.session_state.pop(content_key, None)
 
 def main():
     st.sidebar.title('Web Datasets Bulk Downloader')