I really enjoyed reading this Diego!
I thought a recent post of mine might interest you, specifically in terms of schematic diagrams of compartmental models. Robert Nachbar actually made a CompartmentalModeling package that makes producing these types of diagrams very easy. More can be found here:
https://community.wolfram.com/groups/-/m/t/2078640
I hope you find it useful!