from os import system
system("")
tekstDoZaszyfrowania=input("Wpisz tekst do zaszyfrowania: ")
zaszyfrowanyTekst=''
for kazdaLitera in tekstDoZaszyfrowania:
    if not kazdaLitera.isalpha():
        continue
    kazdaLitera=kazdaLitera.upper()
    kodASCII=ord(kazdaLitera)+1
    if kodASCII>ord('Z'):
        kodASCII=ord('A')
    zaszyfrowanyTekst+=chr(kodASCII)
print('\n\033[1;33;40m'+ zaszyfrowanyTekst +'\033[0m\n')
tekstDoOdszyfrowania=input("Wpisz tekst do odszyfrowania lub wciśnij ENTER,\naby odszyfrować powyżej zaszyfrowany tekst:\n")
if tekstDoOdszyfrowania=='':
    tekstDoOdszyfrowania=zaszyfrowanyTekst
odszyfrowanyTekst=''
for kazdaLitera in tekstDoOdszyfrowania:
    if not kazdaLitera.isalpha():
        continue
    kazdaLitera=kazdaLitera.upper()
    kodASCII=ord(kazdaLitera)-1
    if kodASCII<ord('A'):
        kodASCII=ord('Z')
    odszyfrowanyTekst+=chr(kodASCII)
print('\n\033[1;33;40m'+ odszyfrowanyTekst +'\033[0m\n')
input('\n\nNaciśnij ENTER, aby zakończyć...\n')