Why does Socrates choose to stay in prison?

He didn't choose - he was held in prison during his trial, and on being found guilty, was given the humane choice of suicide, otherwise having his throat cut by a state slave and being thrown into a garbage pit.