For the best AI features, I would recommend Samsung Galaxy S26 Ultra. It includes Photo Assist and Creative Studio through which you can create high quality images.