Yes, a belt would definitely be simpler, although, despite the manufacturer's claims, they are less efficient than chains and sprockets.
However, on an e-bike where there's motor assistance, a small loss of efficiency doesn't matter. And anyway, the NuVinci hub I suggested is not particularly...