I have had a similar experience before, but I didn’t even get the courtesy of a warning before I got sacked off because I worked for a tiny company with no HR and no care for their legal duties.
The problem with the equality act is that it is a pretty toothless and vague piece of legislation that favours companies more because they can afford to use the law, but you still have a decent chance of using it in your favour if you have had an assessment that says you are likely covered by the act. It might be worth going to your doctor simply to show you are being proactive about looking after your mental health, and then maybe I would try take a bit of control (if you feel able and that the job is worth it) and talk to someone to say that you know that you have had these problems at work, but maybe there is a way small adjustments (which the equality act states you are entitled to) could help make things better for both parties? As perhaps there might be some small things that could help you a little… it is hard to say not knowing you or your workplace, but they could be things like written instructions, flexibility, small changes to the work environment. I guess it is hard because you said you are not sure what is causing the crying, but perhaps there are small things at work causing anxiety that can be dealt with? Anyway, a conversation like that with them can show that you are being reasonable and maybe shift the onus more to them that they should be looking after you as an employee they have a duty of care towards.
This probably isn’t helpful at all, if it is a load of rubbish please substitute all that up there for a simple message to say I hope it works out ok and that if not your employers are completely insensitive fucks who don’t deserve someone as good as you x