My understanding is that the reason the brain gets rotted by AI is because you do less thinking per question/problem, leading to your mind thinking less in general and getting used to that. So the solution should be to get your brain thinking more to readjust back to where it was (and beyond!).
A day or two ago in another thread someone posted these two daily brain teaser websites:
https://www.minutecryptic.com/
You could try replacing some time spent scrolling each day with solving these. Minutecryptic especially is requiring me to flex my mind in new ways.
The Dungeons and Dragons idle game has a champion that is composed of Nerds candy pieces. It's actually a fun character concept because you can switch the nerds around to different colors, each one with a different class that changes how their mechanics work. But it's just so weird seeing candy you can eat alongside Drizzt Do'urden that I never use them.