Is it ok to drink coffee after whitening teeth? If not, how long do I need to wait before consuming it again?

Are there any other foods or drinks that I should avoid? Thank you!
--------------------
You can, but be aware that coffee and tea will form surface stains on the teeth.
Source