I Was Diagnosed With Gastritis 1 Week Ago. Since Then,

i was diagnosed with gastritis 1 week ago. since then, i have been on a clean diet; no fatty food, no fried food, and have been drinking lots of coconut water and eating lots of peaches as well as rice, spinach and broccoli. i feel like my stomach is almost back to normal. my question is, can i start taking my protein shake again now that i feel okay? i just came back from the gym