How can I add a returnable item when an item is broken?

I am so close to finishing my painted blocks mod, but there is one hurdle in my way. Is there any possible way to add a returnable item to the players inventory once a certain item is broken. For example I have made a paint tool in which when the paint runs out I want the bucket to be reusable. I haven't really found a suitable procedure for this. I tested the 2020.4 snapshot and tried the 'when item destroyed' trigger, but that doesn't work for me as it's global. So please if anyone knows a method (e.g. ideally a procedure, or custom code) please let me know. Thanks :)