Hi again @Dom_Peou!
I see your point here. The patient address is not the same as the OU (health facility).
If I understand correctly, you want the same / similar format as the OU selection process but in the capture form itself:

It is possible to have the same exact hierarchy as an Organization Unit data element or tracked entity attribute which is then added to a program stage or the enrollment attributes. See screenshot below:

If it is possible to create an OU structure in your instance (or use the existing one) then you have saved yourself from a lot of complicated program rules.
The other option would be to use multi-select optionsets and control their appearance using program rules. See here for example on how to use program rules with multi-select optionsets: Defining program rules for multiselect questions in dhis2 - #3 by Gassim
Whatever your next steps would be, I hope this helps. It’d be great if you provide an update on your next steps too.
Thank you!