I'm also a bit confused. Does the ballroom scene occur before she was arrested ? She seemed to be struggling with something then so if it's before she was arrested then she was struggling with something else before being exiled. Is that it ?
Yes, the ballroom scene happens before the arrest! Some might understand her struggle, and some won't, but she has anxiety and panic attacks, and this is visualized by the flickering scene at the ball ^^