revise

def button(bot, update):
    query = update.callback_query
    print(query.data)
    if 'yes' == query.data:
        query.edit_message_text('ok')
    else:
        query.edit_message_text('next step')

        print(update.message.text)

    return CHOICE