Make "looking at entity" a higher priority than "Holding melee weapon", or add "Holding melee weapon and looking at the entity"
It would also be cool to add a color change depending on the cooldown of the attack, like the "attack indicator" in the vanilla version

Added in version 6.0