yes you can apply wax over water based paint. But it has risk to the wax to resolve some of your water based paint.
It is better if you apply a thin coat of clear coat on your paint, You can apply a thin shellac, lacquer or varnish to seal and protect your paint.
Good luck
wisno