i haven't used it yet but https://c4model.com/ might be relevant to what you're looking for.

Personally I start with exploratory code and wiki for documentation (these days notion)