Можешь этим же способом сделать только нажимать на не show all comments а на .comment__branch, то есть по сути он будет нажимать ветку и оно будет скрываться. Но я не тратил время на тест