Створення програми «Квадрат Полібія» (криптографія)

Повний код шифратора та дешифратора:

# Таблиця Полібія 6x6 для 33 літер української абетки + розділові знаки:
grid = [
    ['А', 'Б', 'В', 'Г', 'Ґ', 'Д'],
    ['Е', 'Є', 'Ж', 'З', 'И', 'І'],
    ['Ї', 'Й', 'К', 'Л', 'М', 'Н'],
    ['О', 'П', 'Р', 'С', 'Т', 'У'],
    ['Ф', 'Х', 'Ц', 'Ч', 'Ш', 'Щ'],
    ['Ь', 'Ю', 'Я', ' ', '.', '!']
]

def encrypt(text):
    result = []
    for char in text.upper():
        for r in range(6):
            for c in range(6):
                if grid[r][c] == char:
                    result.append(f"{r + 1}{c + 1}")
    return " ".join(result)

def decrypt(cipher):
    chars = []
    pairs = cipher.split()
    for p in pairs:
        r = int(p[0]) - 1
        c = int(p[1]) - 1
        chars.append(grid[r][c])
    return "".join(chars)

# Тест:
msg = "СЛАВА УКРАЇНІ"
cipher = encrypt(msg)
print(f"Шифротекст: {cipher}")
print(f"Розшифровано: {decrypt(cipher)}")