People should work in the way that suits the job they're paid to do best. At some workplaces, you need to be present in the office because the company, for various business reasons, has not invested in developing the necessary structures to make effective and well-directed remote work possible. At some workplaces, they have done so.
I prefer working from home, I'm happier doing so, and my boss says I'm even more productive than usual (and he hasn't been concerned about my productivity, and my boss's boss expressed concerns about me working too hard and hitting burnout in our last in-person one-on-one). But my company, organizationally, is not set up this way and we were dragged into it unprepared in March. I'm doing very well on the things where I know how to work autonomously, and I have a good-sized backlog, but at some point I'll run out, and once we're back in the office, I'm going to want to be around for all of the "distractions" and "socialization" that help me develop a good idea of what I should be working on. It does the company no good - and it does me no good, come performance review season - to be highly productive at the wrong thing.
And the "diversity" argument doesn't work here. You need to pick one context where things happen, unless you want to effectively run multiple companies. Either you go remote-first and you adopt the Apache-style policy of "if it didn't happen on the mailing list, it didn't happen," or the folks who are seeing each other in person will chat in person, come up with ideas in person, make decisions in person, consult each other for design reviews in person, etc. and cut the remote workers out of the loop.
Now, yes, I think my company probably should develop the structures needed to work remotely, but I'm aware that's a self-serving belief. I genuinely don't know if my company has enough people who have the "discipline" (in your words; I might also add "lack of roommates, spouse, or children") that this switch is the right thing for the company. It's rather like my position on whether my company should pay me more: I'm not going to argue that I don't want the money, since it would certainly be good for me, but there is certainly some finite salary level where it's no longer in the business's interest to do so.