THE PRETTIEST CAESAR CIPHER CODE YOU'LL SEE!!!!

import string

alph_val = dict(zip(range(1,27), string.ascii_uppercase))
alph_key = dict(zip(string.ascii_uppercase, range(1,27)))

plaintext = str(input("Enter plain text: ")).upper().split()
shift = int(input("Enter desired shift: "))

cipher = ""
for word in plaintext:
    for letter in word:
        new_letter_key = shift + alph_key[letter]

        if (new_letter_key > 26):
            new_letter_key -= 26

        new_letter = alph_val[new_letter_key]
        cipher += new_letter

    cipher += " "

print("Cipher text: ", cipher)