I have more energy and can work more hours since I switched to a Keto diet.

I think it's probably the other way around: you fix your diet first, and mental health will improve as a result.