I don't see anything wrong with setting boundaries with the student. you can let him know that things have gotten out of hand and these chats are taking too much time away from work so you need to minimize the chit chat and get back to work. good luck and take care.