Plot atom type contribution on band structure

Dear Exciting Community,

I am following a talk given at one of your official workshop[1], while reproducing the official exciton visualization tutorial[2]. I was able to reproduce the tutorial example and apply it to my own problem but I am looking to superimpose the atom type contribution on the bandstrucure with exciton weights as shown in the 1st slide of below talk. Can anybody please guide how I can reproduce same bandstructure shown in the 1st slide(at the given time-stamp) of the video.

Thank you.