Interesting work! There's NeuroMechFly (https://github.com/NeLy-EPFL/flygym/) available, you might try using that to simulate the fly body in real time as well.