My biggest mistake I reckon was disclosing my illness to them last year. I should never have do e that because it's as if they are going out of their way to look for signs when right now I am not even having an episode.

I agree with this. But the mistake has been made. Now you need to move forward and consider your options. Not necessarily the ones they gave you, but the ones you can create for yourself. I do believe one thing. Keep in mind I am no expert on this. So you may want to consult a lawyer. But I believe that as long as you are able to keep your performance up, they will not fire you. They know you can then sue them. They will do everything they can do to protect themselves from this. I think I would now approach them for some concessions. Negotiate with them. You need to show them that you are cooperating with them. Meanwhile, I would also be looking for another job. This is what I would do.

Just FWIW. I hope this helps!