The upcoming flagship is expected to get a new Tensor chip, improved cameras and a more helpful AI assistant