Mushroom-template-badge - change icon based on state

Hi.

I have cobbled this together. Its working in the sense that its displaying the number of lights on. What i would like to do is change the colour depending on if the lights are on or off.

Im a little unsure as everything i have tried has just not worked.

Thanks in advance

type: custom:mushroom-template-badge
content: |-
  {{states.light| rejectattr('entity_id','in',label_entities('LIGHT-GROUP'))
                | rejectattr('entity_id','in',label_entities('LED')) 
                | selectattr('state','eq','on')
                |list | count }} 
icon: mdi:lightbulb
color: Yellow