Я не вижу вашего кода.это пользовательский диалог ??Вы можете разместить кнопку в AlerDialog
AlertDialog alertDialog = new AlertDialog.Builder(a).create();
alertDialog.setTitle(title);
alertDialog.setMessage(message);
alertDialog.setButton2("OK", new DialogInterface.OnClickListener() {
public void onClick(DialogInterface dialog, int which) {
//place what you want to do here
return ;
} });
alertDialog.show();