Update app.py

app.py

@@ -6,26 +6,18 @@ import streamlit as st
 import re
 
 def fix_bartpho_output(text: str) -> str:
-    """
-    BARTpho syllable tách từng âm tiết riêng lẻ nên hay bị dính.
-    Dùng underthesea để word tokenize rồi ghép lại đúng cách.
-    """
-    import re
-    
-    # Fix các pattern dính từ phổ biến nhất của BARTpho
-    # Thêm space giữa: chữ thường + chữ hoa
+
     text = re.sub(
-        r'([a-záàảãạăắằẳẵặâấầẩẫậéèẻẽẹêếềểễệíìỉĩịóòỏõọôốồổỗộơớờởỡợúùủũụưứừửữựýỳỷỹỵđ])'
-        r'([A-ZÁÀẢÃẠĂẮẰẲẴẶÂẤẦẨẪẬÉÈẺẼẸÊẾỀỂỄỆÍÌỈĨỊÓÒỎÕỌÔỐỒỔỖỘƠỚỜỞỠỢÚÙỦŨỤƯỨỪỬỮỰÝỲỶỸỴĐ])',
-        r'\1 \2', text
+        r'([a-zđàáâãèéêìíòóôõùúýăắặấầẩẫậ])'
+        r'([A-ZĐÀÁÂÃÈÉÊÌÍÒÓÔÕÙÚÝĂẮẶẤẦẨẪẬ])',
+        r'\1 \2',
+        text
     )
-    
-    # Fix dấu câu dính chữ: "vong.Cơ" → "vong. Cơ"
-    text = re.sub(r'([.!?,;:])([^\s\d])', r'\1 \2', text)
-    
-    # Xóa space thừa
+
+    text = re.sub(r'([.!?,;:])([^\s])', r'\1 \2', text)
+
     text = re.sub(r' +', ' ', text).strip()
-    
+
     return text