Maybe you could make use of the automation i made for a Xiaomi switch
https://community.home-assistant.io/t/longpress-and-shortpress-script-automation-for-xiaomi-round-button-with-zha/229367