[tudorw]

Very interesting. Is there a way to retrieve the segment information, then surface that in a UI so I can select and regenerate single elements?

[jbhuang0604]

Yes, if you go to the huggingface demo: https://huggingface.co/spaces/songweig/rich-text-to-image

You can find the segmentation information on the bottom-right of the rich-text generation result.