Kim Zolciak praised son KJ for his ‘discipline’ and ‘dedication’ two months before sexual assault arrest