Yes, the patient and the guardian are allowed to stay in the country after the treatment, for as long as it becomes safe for the patient to travel again. The visa is extended accordingly after the doctor gives the minimum time period to rest before it is safe to travel.