Удалил лишний код, использовал функцию restore_word_case из pymorphy2, чтоб не дублировать код