I've got ADHD and actually found the opposite for me. Work from home was disastrous because there are so many potential distractions at home, and little accountability. I would see some dust and go get a broom to sweep, or go brush my teeth in case I forgot, or get food, or take "just five minutes" (soon 30) to check social media, or decide I need to make my bed. Any frustration or roadblock meant I would find something to derail me, and this all contributed to depression that made it even harder to focus on work.

Going back to the office was a huge boon for me, even before starting medication. I'm currently the only person on my team working 100% from office, and I would walk barefoot over broken glass to my open plan office before I would ever willing return to working from home.

I'm not sure if you've got ADHD; I know there's a lot of variation in personal experience. I've heard similar sentiments from multiple people with ADHD regarding working from home, although I think I'm a bit of a freak for tolerating open plan offices well.