Badges with different icons based on state

In my LoveLace UI, I have badges… I like the badges up the top for specific things we use often.

Some of these badges are actually input_booleans we use. If the input boolean is on, it shows on… if it is off… it shows off…

I would like to change the on/off words for icons/something pretty.

Is there anyway to do this? I have tried a bunch of things (customize and create template sensors with a template icon) but the same issue remains.

look for the custom-button-card https://github.com/kuuji/button-card