I want to produce some images related to graphene-like materials. I was able to draw them using ASE visualize module after constructing the atomistic model. However the result is only sketchy and I want to add some other fancy things like different colors for atoms, vector arrows, bond lines, words and so on. Is it possible to do so, maybe integrating it with some other tools?

To give some reference, I would like to produce something like the first image in this https://www.researchgate.net/profile/Jermaine-Lado/publication/342435484/figure/fig1/AS:906158508371969@1593056432796/Twisted-bilayer-graphene-a-Real-space-moire-pattern-formed-by-the-relative-rotation-of.png

If not, is there another software that can help me achieve this, without manually working with graphics primitives in a library like matplotlib?

Thanks