Use light color as indicator - temporarily

I want to use my lights as color coded status lights. However, after that it should return to the light color and state of before.

What is the best way to do this without making it too complicated?