Can AI stop your mind from wandering?