Former world number one Iga Swiatek needed the help of a psychologist to regain her form after a doping ban last year.