AI technology could help protect water supplies