Hi OP.
Sounds like I went through something similar. I was section 2 a number of years ago from a stress induced psychosis. I was also on Aripiprazole and lorazepam whilst in the hospital, but later found olanzapine was better for me.
I went through ups and downs so dont panic if its not a smooth line of recovery. You will get there :)
I found a notebook very helpful. I would draw thoughts which came into my head because writing it was hard to get down.
I also liked to do yoga and some meditation when things kicked off. I appreciate its a difficult environment to be in and other residents can trigger or escalate your thoughts. Try not to get too close to any one person because everyone will come and go at different times; and try not to take any notice of what others are saying.
Here for you if you need to talk 