Does this work with CSS in JS stuff and CSS frameworks - like if I was using Chakra would this be able to edit the site elements and have the agent reverse map to where the style attributes need to go ?

Yes - if your agent can edit your codebase it will put the changes in the right place.