Need some advice about the hair loss

Okay I met my surgeon in June and told I should lose about fifteen to 20 pounds. However I been told that around four month out you can experience hair loss. I was thinking about getting a weave since I am worried about using heat and chemicals. I was thinking about getting it prior to surgery. I got keratin treatment done in now but I won't get it back in at least maybe next year since it a chemical. I thinking telling my stylist not make the braids too tight. What do you think or do any suggestion? I won't look right with a afro. I was thinking getting a weave anyway