I think AGI/ASI could help quite a bit. However, it's worth noting that some percentage of mental disorders are environmentally-driven (e.g. it's no surprise that someone who was abused as a child is more likely to suffer from mental disease).

To some extent, I suspect that type of thing sticks with you unless your neural implant decides to kill the bad memories altogether. But that raises interesting questions and could lead to unintended consequences.