How can I keep screen on ONLY while music is playing?
If you're using Pandora, you need to uncheck "Conserve battery option" option under the Menu > Settings > Advanced. Now the screen will stay on while the music is playing on Pandora.

You can also download the apps like KeepScreen and Stay Alive to block screen timeout when predefined applications are active.