Dash AG Grid single cell styling after cell edit

Hello guys, Thank you so much for this solution;

I’m trying to implement this in a project I’m building and it’s correctly highlighting the changed cell but if I undo the change it’s not removing the highlight
highlight-issue


There’s a way to control the highlight in a callback instead of the way it’s being done?

If you need/want, here I have a MRE you can use:

My goal is to undo the highlight if the user:
1 - undo the change by setting the old value again ( I have a dcc.Store with the original data)
2 - if the user submit the change, it should also remove the highligh as a new original data will be set

Did you have tried this approach or similar?