I would absolutely and totally upgrade the bike to Di2. I've had electronic shifting on my TT bike since 2014 or thereabouts, but i only recently put it on my road bike. And after that, i cant go back to mechanical.

If you are buying the Di2 locally, perhaps the shop where you are buying it from can throw in an install for free - worth asking.