St.multiselect not populating with selection after callback

Hi all, running into some issues where st.multiselect is not populating selected values on click. The st.multiselect has a callback function that updates a dataframe (which feeds the multiselect options dropdown - this may be the issue here). Any idea how to retain the selected values upon selection, while still updating the multiselect options?

Can you show a demo code ?