Artificial intelligence technology could help protect water supplies