Зашифровать послание с помощью шифра Цезаря с заданным шагом шифрования

step = int(input()) 
word = input() 
 
for i in word: 
    if chr(ord(i)) == ' ' or chr(ord(i)) == '!': 
        print(i, end='') 
    else: 
        print(chr(ord(i) + step), end='')

Ввод:                          Ожидаемый результат:                 Вывод:
4                                 Ъйлдфа еяп сй ичфдо!                  Ъйлдфѐ еяп сй ичфдо!
Цезарь был не дурак!

from tkinter import *
import sys
 
 
alpha = "abcdefghijklmnopqrstuvwxyz ";
 
def encr_click():
    key = int(key_entry.get());
    length = len(alpha);
 
    res = '';
 
    for c in text_entry.get():
 
        if c.lower() in alpha:
            s1 = alpha.find(c.lower());
            s1 = (s1 + key) % length;
            res += alpha[s1];
        else:
            res += c;
    
    res_entry.delete(0, END);
    res_entry.insert(0, res);
 
def dencr_click():
    key = int(key_entry.get());
    length = len(alpha);
 
    res = '';
 
    for c in text_entry.get():
 
        if c.lower() in alpha:
            s1 = alpha.find(c.lower());
            s1 = (s1 - key) % length;
            res += alpha[s1];
        else:
            res += c;
    
    res_entry.delete(0, END);
    res_entry.insert(0, res);
 
 
root = Tk();
root.title("Caesar cipher");
 
text_label = Label(text = "Enter your text: ", font = ("Verdana", 12));
key_label = Label(text = "Enter a key: ", font = ("Verdana", 12));
res_label = Label(text = "Your result is: ", font = ("Verdana", 12));
 
text_label.grid(row=0, column=0, sticky="w");
key_label.grid(row=1, column=0, sticky="w");
res_label.grid(row=2, column=0, sticky="w");
 
 
text_entry = Entry(root, font = ("Verdana", 12, "bold"));
key_entry = Entry(root, font = ("Verdana", 12, "bold"));
res_entry = Entry(root, font = ("Verdana", 12, "bold"));
 
text_entry.grid(row = 0, column = 1, padx = 5, pady = 5);
key_entry.grid(row = 1, column = 1, padx = 5, pady = 5);
res_entry.grid(row = 2, column = 1, padx = 5, pady = 5);
 
 
e_bt = Button(root, text = "Encrypt", command = encr_click, background = "#555", 
                foreground = "#ccc", font = ("Verdana", 16, "bold"));
d_bt = Button(root, text = "Decrypt", command = dencr_click, background = "#555", 
                foreground = "#ccc", font = ("Verdana", 16, "bold"));
 
e_bt.grid(row = 3, column = 1, padx = 5, pady = 5, sticky = "w");
d_bt.grid(row = 3, column = 1, padx = 5, pady = 5, sticky = "e");
 
 
root.mainloop();

step = int(input()) 
word = input() 
 
for i in word: 
    if chr(ord(i)) == ' ' or chr(ord(i)) == '!': 
        print(i, end='') 
    else: 
        base = ord('А') if i.isupper() else ord('а')
        print(chr((ord(i) + step - base) % 32 + base), end='')

def ceasar_encode(letter, shift):
    if letter.isalpha():
        number = ord(letter) + shift % 32
        if number  > 1103:
            number -= 32
        return chr(number)
    return letter
 
shift = int(input())
for l in input():
    print(ceasar_encode(l, shift), end='')