As for substituting the chaos bags with the dice, I think it would be possible without having to customize them. Just create an equivalence sheet/card for each scenario/difficulty and roll the dice that would fit the number of tokens best. You can even change dice mid game if the number of tokens is altered somehow.
Sounds like it would work - unless there are 17 tokens in the bag or something like that.