Remaking the UI for AI photo