Update app.py

app.py

@@ -62,6 +62,8 @@ def annotate_morphemes(word, prefixes=prefixes, suffixes=suffixes):
                     elif list(annotation[-1].values())[0] == 'interfix':
                         # print(73, stack[::-1])
                         annotation.append({stack[::-1]: 'root'})
+                    elif list(annotation[-1].values())[0] == 'postfix':
+                        annotation.append({stack[::-1]: 'root'})
                     else:
                         # print('1111111111', stack[::-1], annotation[::-1], annotation)
                         annotation.append({stack[::-1]: 'unk'})
@@ -87,7 +89,7 @@ def annotate_morphemes(word, prefixes=prefixes, suffixes=suffixes):
         if len(annotation) >=2 and list(annotation[-2].values())[0] == 'root' and list(annotation[-1].values())[0] == 'prefix':
             annotation[-1] = {list(annotation[-1].keys())[0]: 'interfix'}
             annotation.append({stack[::-1]: 'root'})
-        elif list(annotation[-1].values())[0] in ('interfix', 'suffix', 'root'):
+        elif list(annotation[-1].values())[0] in ('ending', 'suffix', 'interfix', 'root'):
             annotation.append({stack[::-1]: 'root'})
         else:
             annotation.append({stack[::-1]: 'unk'})