I'm glad you're enjoying the game, and thank you for the kind words! :)
The way I had to code the rehab system, it couldn't detect what state the girls were in before you sent them to rehab, hence the weird jump in their personalities.
Any Boss Girl you beat after the rehab release, however, has their mental state recorded correctly and they will act accordingly in the first rehab session (which MC uses as a "deprogramming" session).
Aside from that, I can't really add much more to the rehab system, I'm afraid, as I'm only one person and time spent on adding more and more content to the rehab side of things is 1) more content I need to add in the future whenever I add a new Boss Girl, and 2) time taken away from creating story events.
The weekly updates put me on a very limited schedule, too ;P