Fix bug thanks to claude-3.5-sonnet

app.py

@@ -58,11 +58,10 @@ def show_data_editor(filtered_df: pd.DataFrame, key: str):
                 key=key)
     
     if edited_df is not None and not edited_df.equals(filtered_df):
-        # update the df of the session state with the affected rows
-        # TODO there seems to be a bug in here
-        original_df = st.session_state.df
-        original_df.update(edited_df)
-        st.session_state.df = original_df
+        main_df = st.session_state.df
+        edited_indices = edited_df.index
+        main_df.loc[edited_indices] = edited_df.loc[edited_indices]
+        st.session_state.df = main_df
 
 
 def display_data(filtered_df: pd.DataFrame):