OK, some progress. It looks like that controller doesn’t fit. It's pretty crappy anyway and it has the old three speed switch, that's useless, instead of a control panel. That picture of the bike seems to show an LCD. What do you actually have.
If got it working during self leaning, you should congratulate yourself because most people wouldn't figure out how to switch it on. It should be OK for testing, but I wouldn't use it for riding unless you're really strapped for cash. That's if you can figure out where to put it.
I'm still finding it difficult to recommend anything because I don't understand what you're trying to achieve. Do you just want the bike the same as it was before or are you wanting more power or speed, or do you want a better control system that makes the bike better to ride.