Can I not drink coffee right away after teeth whitening?

I've been a little self-conscious about the color of my teeth lately, so I'm thinking about getting them whitened.

However, I tend to drink at least one cup of coffee a day.
I am worried about how careful I need to be with food and drinks after whitening.

I heard that you should avoid not only coffee but also dark-colored foods like kimchi and curry.
I felt like there might be more things to manage than I thought.

For those of you who have received whitening treatments, how did you eat after the procedure?