you know how your teeth feel smooth after you brush? that smoothness helps things like coffee not stick to them, in turn helps avoid stains

brushing before coffee is legit the thing to do