Improved direct diabatization and coupled potential energy surfaces for the photodissociation of ammonia