You could try to make an alias for chroot to run your umount command or script after it completes, such as:
alias chroot='chroot; umount my/drive'
Or maybe use a different name for it.
Another option would be to have script running in the background watching for chroot processes and then umount the drive when the process is terminated.