Add a button to allow user to choose to generate AI images.