Add directive to disable buttons after click