Training Allays to Specific Instruments

I think that the Allays are an amazing new feature in the game, however I feel that they are rather limited when it comes to using noteblocks to set the location where they drop stuff off at. If we could tell the Allays specifically which noteblock they should go to, I think it would greatly improve the functionality of the creature. And there shouldn't be some kind of GUI for this, it would be pretty weird to open up how an entity works like that. I think it would be cool if we could train the Allay to only respond to specific instruments.
The way that this could work is we put the Allay in a room with a noteblock periodically playing a specific instrument. After a couple minutes, the Allay makes a small sound to let us know that it is trained, and from then on it would only deposit items in the noteblock making that specific noise. This could be extended to different pitches as well, but I think that would make this idea too complicated. But I believe that the ability to train the Allay to go to a specific instrument would greatly increase it's functionality.