Any job that can be WFH should be WFH.
Any job that can't be WFH that requires sitting at a desk all day should give each person an individual office. The open office plan has been an absolute nightmare, and only benefits micromanagers. It's a productivity disaster, and makes for a miserable experience, and only exists for the sake of surveillance. However, I doubt there are many jobs that can't be WFH that require such a situation.
The real issue here is an intentional mis-framing, imo. Why must people get back to a traditional office setting? The only people who want this are employers who think that Butts In Seats = Productivity, and the only way to ensure it is to intensely surveil your employees. I also don't give two shits if some real estate company goes bankrupt because business tenants stop renting their properties. Boo fucking hoo.
I've been working for a remote-first company now for over a year, and I won't ever got back to working in an office. There is literally nothing about what I do that needs me to be physically present in any specific place. The problem isn't "productivity" or "collaboration", the problem is entirely based around a work culture that is fundamentally punitive, puritanical, and antithetical to life balance.