Text_encryption.py

#encryption programming

import random
import string

chars = " " + string.punctuation + string.digits + string.ascii_letters
chars=list(chars)
key = chars.copy()

random.shuffle(key)

#print(chars)
#print(key)

#ENCRYPT

plain_text = input("Enter a message to encrypt: ")
if not plain_text.isalpha():
    print("Error: Only alphabets are allowed!")
    exit()
cipher_text = ""

for letter in plain_text:
    index = chars.index(letter)
    cipher_text += key[index]
    
print(f"original message : {plain_text}")
print(f"encrypted message : {cipher_text}")

#decrypt 
cipher_text = input("Enter a message to decrypt: ")
if not cipher_text.isalpha():
    print("Error: Only alphabets are allowed!")
    exit()
plain_text = ""

for letter in cipher_text:
    index = key.index(letter)
    plain_text += chars[index]
    
print(f"encrypted message : {cipher_text}")
print(f"original message : {plain_text}")