Enhancing water supply predictions through improved AI processes