Replace microcontroller/bluetooth unit TX YouFS scooter

Hi All, I have a TX brand scooter, which is maybe supposed to be a Xiaomi copy, on which I have rebuilt the battery pack and it seems to work ok, however the scooter shuts down completely at times for no reason and I have to disconnect/reconnect the battery to get it to switch on again. This can leave you stranded. Also, I want to enable regenerative braking if possible but I cannot get the YouFS app to connect nor any other app except Xiaoflasher which asks for a firmware file.

Anyway, rather than keep trying to solve these problems I was wondering if there was a replacement chip or pcb and firmware option for these scooters which would hopefully solve all issues at once.

TIA