Lemmatization - associate declined/ conjugated word forms with their initial form

Lemma: dictionary form of a word (undeclined/ unconjugated) Lemmatization: associating declined/ conjugated forms of a word with their "dictionary form". https://en.wikipedia.org/wiki/Lemmatization The goal is to be able to click any declined/ conjugated form of a word, then click one more button to automatically save its initial form as a dictionary entry and treat/ highlight all other forms of the word as "learning". All without needing to type anything or edit the entry other than selecting the translation(s). For example, I click the word прошёл, then click a button labeled "save initial form/ save lemma": -A dictionary entry is made for пройти. -All other conjugated forms пройду, пройдёш, пройдёт, пройдёте etc... are automatically highlighted as "learning". If "show dictionary translation in text" is turned on, then the initial form will be shown in parentheses with the translation. Or there could just be an additional menu option to show the initial form of "learning" words in the text. Of course, there would still be the option to make a normal entry for certain or unique declined forms the user may want to reinforce, but for the most part, the dictionary would contain 99% initial forms. This would be extremely useful for heavily inflected languages like Russian, French, Italian etc..., where a single word can have dozens of inflected forms. Currently, I have to constantly edit declined/ conjugated forms into their initial form because I hate saving conjugated/ declined forms. Also, because declined/ conjugated forms are treated as separate words in the program, it's impossible to be sure if I have already saved the initial form while reading, meaning I often end up looking up words whose initial form I've already saved. Lemmatization could bring the whole process down to two clicks and prevent repeated look ups. Finding the initial form is easy now with the AI functions but I guess associating the declined forms would need to be done with tagging or an API.