Update app.py

app.py

@@ -50,7 +50,8 @@ if uploaded:
     st.session_state.all_fields = get_all_fields(st.session_state.data)
     save_to_file()
     st.session_state.editor_key = str(uuid4())
-    st.rerun()
+    st.session_state.just_added = True
+    st.rerun()  # Force rerun to update editor
 
 # --- Add New Fields fallback ---
 if not st.session_state.all_fields:
@@ -70,7 +71,7 @@ if submit_add:
     save_to_file()
     st.session_state.editor_key = str(uuid4())
     st.session_state.just_added = True
-    st.rerun()
+    st.rerun()  # Force rerun to update editor
 
 # --- Wait for rerun before rendering editor ---
 if st.session_state.just_added:
@@ -100,10 +101,10 @@ edited_df = st.data_editor(
     column_config=column_configs,
 )
 
-# --- Save edits ---
+# --- Save edits immediately after editing ---
 if edited_df is not None:
     new_data = edited_df.fillna("").to_dict(orient="records")
-    if new_data != st.session_state.data:
+    if new_data != st.session_state.data:  # Ensure the data has changed before saving
         st.session_state.data = new_data
         save_to_file()
         st.toast("✅ Changes auto-saved!", icon="💾")
@@ -114,7 +115,7 @@ with st.expander("➕ Add New Field"):
     if st.button("Add Field"):
         if new_field and new_field not in st.session_state.all_fields:
             st.session_state.all_fields.append(new_field)
-            st.rerun()
+            st.rerun()  # Rerun after adding field
 
 # --- Export Dataset ---
 st.markdown("### 📤 Export")
@@ -135,7 +136,7 @@ with col1:
         st.session_state.clear()
         if os.path.exists(TMP_FILE):
             os.remove(TMP_FILE)
-        st.rerun()
+        st.rerun()  # Force rerun after export
 
 with col2:
     if os.path.exists(TMP_FILE):
@@ -148,4 +149,4 @@ with col3:
         if os.path.exists(TMP_FILE):
             os.remove(TMP_FILE)
         st.session_state.clear()
-        st.rerun()
+        st.rerun()  # Rerun after clearing session