Update app_og.py

app_og.py

@@ -27,8 +27,11 @@ def calculate_hash(data, hash_function: str = "sha256") -> str:
 
 
 def generate_keys():
+    secret_code="password"
     key = RSA.generate(2048)
-    private_key = key.export_key('PEM')
+    #private_key = key.export_key('PEM')
+    private_key = key.export_key(passphrase=secret_code, pkcs=8,
+                          protection="scryptAndAES128-CBC")
     print(f'private_key:: {private_key}')
     
     file_out_priv = open("private.pem", "wb")
@@ -84,7 +87,7 @@ def encrypt_text(data,pub_im,priv_im,address):
     #print(f'priv_key:: {priv_key}')
     #private_key = RSA.import_key(priv_key)    
     data = data.encode("utf-8")
-    data = sign(data,priv_im)
+    #data = sign(data,priv_im)
     
     
     recipient_key = RSA.import_key(pub_key)
@@ -132,11 +135,11 @@ def encrypt_text(data,pub_im,priv_im,address):
 
 def decrypt_text(im,in2):
     enc_in = stegan2.decode(im)
-    
+    secret_code="password"
     #private_key = RSA.import_key(open("private.pem").read())
     priv_key = stegan2.decode(in2)
     print(f'priv_key:: {priv_key}')
-    private_key = RSA.import_key(priv_key)
+    private_key = RSA.import_key(priv_key,passphrase=secret_code)
     print(f'private_key:: {private_key}')
     enc_session_key = enc_in[:private_key.size_in_bytes()]
     end1 = private_key.size_in_bytes()+16
@@ -159,14 +162,17 @@ def decrypt_text(im,in2):
     cipher_aes = AES.new(session_key, AES.MODE_EAX, nonce)
     data = cipher_aes.decrypt_and_verify(ciphertext, tag)
     
-    public_key = private_key.publickey().export_key('PEM')
-    data_val = data[0]
-    signature = data[1]
-    transaction_hash = SHA256.new(data_val)
+    #public_key = private_key.publickey().export_key('PEM')
+    #data_val = data[0]
+    #print (data_val)
+    
+    #signature = data[1]
+    #print (signature)
+    #transaction_hash = SHA256.new(data)
 
     #transaction
     
-    print (pkcs1_15.new(public_key).verify(transaction_hash, signature))
+    #print (pkcs1_15.new(public_key).verify(transaction_hash, signature))
     return(data.decode("utf-8"))
 
 def test_fn(im1,im2):