Update app.py

app.py

@@ -509,6 +509,25 @@ def correct_dash_usage(text):
     # Join the sentences back
     text = ' '.join(sentences)
 
+    def restore_yo(text):
+        morph = pymorphy2.MorphAnalyzer()
+        words = text.split()
+        restored_words = []
+        
+        for word in words:
+            parsed = morph.parse(word)[0]
+            restored_word = parsed.word
+            
+            # Сохраняем оригинальный регистр первой буквы
+            if word and word[0].isupper():
+                restored_word = restored_word.capitalize()
+            
+            restored_words.append(restored_word)
+        
+        return ' '.join(restored_words)
+
+    text = restore_yo(text)
+
     return text