Pretty sure productivity will go down. When I WFH I just slack off.
I've worked in an office for years, and then transitioned to working from home for years. The initial transition was rough: snacking, lack of motivation, lack of structure.
But it only took about a month to get past that. You start building your own routine and structure, optimized to your life, not a mandated 9-5. Not only am I more efficient and less stressed at home, I'm pretty certain I actually "work" more hours than sitting at an office.
And I'm a pretty extroverted person, that thrives with other human contact. When working from home, I get to choose when to socialize/collaborate, it isn't hoisted on me.