Шифр Виженера "Python"

Задание

Написать программу , которая будет просить пользователя написать на выбор два варианта: расшифровать или шифровать. Далее программа просит написать ключ и она будет шифровать или расшифровывать сообщение.

Функционал программы

Программа просит написать пользователя два варианта:расшифровать или шифровать. Далее она попросит написать ключ и она будет шифровать или расшифровывать сообщения.

Фрагмент программного кода

def main():
    print('''Vigener Cipher. The Vigener cipher is a polyalphabetic substitution cipher that was
      powerful enough remain unbroken for centuries.''')

    # Спрашиваем у пользователя, хочет ли он шифровать или расшифровывать
    while True:  # Повторяем вопрос, пока пользователь не введет e или d
        print('Do you want to (e)ncrypt or (d)ecrypt?')
        response = input('>  ').lower()
        if response.startswith('e'):
            myMode = 'encrypt'
            break
        elif response.startswith('d'):
            myMode = 'decrypt'
            break
        print('Please enter the letter e or d')

    # Просим пользователя ввести ключ шифрования
    while True:
        print('Please specify the key to use.')
        print('It can be a word or any combination of letters: ')
        response = input('>  ').upper()
        if response.isalpha():
            myKey = response
            break

    print('Enter the message to {}.'.format(myMode))
    myMessage = input('>  ')

    if myMode == 'encrypt':
        translated = encryptMessage(myMessage, myKey)
    elif myMode == 'decrypt':
        translated = decryptMessage(myMessage, myKey)

    print('%sed message: ' % (myMode.title()))
    print(translated)

    try:
        pyperclip.copy(translated)
        print('Full %sed text copied to clipboard.' % myMode)
    except:
        pass

Скриншот архива с проектом

Пояснения по запуску программы

Запустить PyCharm, открыть проект и запустить программу, нажав на вкладку вверху и run Main.