Are schools legally allowed to take away a students phone?

No, because it is not their personal property and is a violation of privacy. Students do have the right to not give up their phone to school officials