How do I make a button play a certain disc when the certain disc is in a slot?

That procedure should work- though I recommend using Else-If blocks instead to streamline it. Make sure you have block entity turned on for your custom jukebox- also bear in mind that with this method, the music will continue playing even if the disc is removed, or if another disc is played. (You might want to add some sort of delay with the block.)