Not my cup of tea, just want some kind of drive that could adapt to any bike.
I think upsizing the gearbox is the right idea. I'm OK to leave out regen and use a freewheel also.
Maybe if you could make a sandwich adapter of sorts and cut holes in the motor casing etc.
A 3210 + planetary gear reduction is pushing it as far as width goes for a bike with 135mm dropouts. The weight may need to be slightly offset at that point.. I'd explore liquid cooling first.
I adapted a Neugart gearbox to an Astro motor for a chain puller drive. It worked pretty well. I suspect the Neugart gearbox is somewhat more robust though.