from cryptography.hazmat.primitives import hashes from cryptography.hazmat.primitives.asymmetric import padding, rsa # Generate rsa keys to encrypt and decrypt the messages private_key = rsa.generate_private_key(public_exponent=65537, key_size=2048) public_key = private_key.public_key() def encrypt_message(message) -> bytes: ciphertext = public_key.encrypt( message, padding.OAEP( mgf=padding.MGF1(algorithm=hashes.SHA256()), algorithm=hashes.SHA256(), label=None, ), ) return ciphertext def decrypt_message(ciphertext) -> str: decrypted_message = private_key.decrypt( ciphertext, padding.OAEP( mgf=padding.MGF1(algorithm=hashes.SHA256()), algorithm=hashes.SHA256(), label=None, ), ) return decrypted_message