import sys
import re

def fix_myanmar_content(text):
    # 1. AI tự ý viết tắt (Vi phạm Literal Transcription)
    text = text.replace("၎", "လည်းကောင်း")
    
    # 2. AI tự đổi số thành chữ hoặc ngược lại (Nếu phát hiện mẫu cụ thể)
    # Theo MEMORY: tự đổi 'သုံး' thành '၃'
    text = text.replace("၃", "သုံး") # Thận trọng với quy tắc này
    
    # 3. Lỗi ký tự OCR phổ biến (Dựa trên phân tích vừa thực hiện)
    # Lưu ý: Chỉ sửa những gì chắc chắn là lỗi nhận diện sai hình thể 
    # nhầm 'ပ' thành 'န' (thường gặp trong các từ phổ biến)
    # Ở đây em sẽ dùng regex để sửa một số từ vựng Myanmar phổ biến bị sai
    
    # Các quy tắc từ MEMORY.md
    # Nhầm ဖ thành ဘ
    # Nhầm ျ thành ြ
    # Nhầm ထ thành ဌ
    # Nhầm ပ thành ဘ
    
    # Tự động hóa một số thay thế dựa trên thống kê lỗi (Substitutions)
    # Ví dụ: nếu 'ဘို့' (cổ/sai) -> 'ဖို့' (chuẩn) nhưng anh muốn literal, 
    # nên em sẽ xem bản standard để quyết định.
    
    return text

if __name__ == "__main__":
    if len(sys.argv) < 2:
        print("Usage: python3 fix_myanmar_v2.py <input_file>")
        sys.exit(1)
        
    file_path = sys.argv[1]
    with open(file_path, 'r', encoding='utf-8') as f:
        content = f.read()
    
    fixed_content = fix_myanmar_content(content)
    
    with open(file_path, 'w', encoding='utf-8') as f:
        f.write(fixed_content)
    print(f"Fixed {file_path}")