Excellent choice @leahlei!
Coffee is a very rich drink and really helps a lot who works late into the night, but the stomach is first.
Green tea is an excellent choice for the amount of properties it has and, at this time, you are still under treatment, it is a good ally.
I wish you a complete recovery !!