Can you first roll your dice in an ability check, and then decide to roll a bonus die to improve the result?

Yes, you can absolutely spend pools afterwards to get bonus dice. [1]