Update app.py

app.py

@@ -145,9 +145,9 @@ def decrypt_text1(im,in2):
     print(f'mes_bytes3::{mes_bytes3}')
     print(f'mes_bytes4::{mes_bytes4}')
 
-    #file_out = open("encrypted_data.bin", "w")
-    #[ file_out.write(x) for x in (mes_bytes1, mes_bytes2, mes_bytes3, mes_bytes4) ]
-    #file_out.close()
+    file_out = open("encrypted_data.bin", "w")
+    [ file_out.write(x) for x in (mes_bytes1, mes_bytes2, mes_bytes3, mes_bytes4) ]
+    file_out.close()
 
 
     
@@ -224,10 +224,11 @@ def encrypt(passw,trans_s=None,trans_r=None,trans_a=None):
     if trans_s != None and trans_r != None and trans_a != None and trans_s != "" and trans_r != "" and trans_a != "":
         trans = trans_block(trans_s,trans_r,trans_a)
         bytes_j = trans.encode()
-
-        json_bytes = fernet.encrypt(bytes_j)
+        cipher_rsa = PKCS1_OAEP.new(recipient_key)
+        json_bytes = cipher_rsa.encrypt(bytes_j)
+        #json_bytes = fernet.encrypt(bytes_j)
         enc_mes = f'{json_bytes}+ccc+'
-
+        
     #enc_qr = qr.make_qr(txt=(enc_mes.strip('"')))
     m = hashlib.sha256()
     bytes_enc_mes = enc_mes.encode()
@@ -254,6 +255,8 @@ def decrypt(passw,enc_in=None):
         mes1=enc_in.split("+aaa+",1)[0]
         mes1=mes1.strip("b'").strip("'")
         mes_bytes = bytes(mes1,'utf-8')
+
+        
         mes_dec = fernet.decrypt(mes_bytes).decode()
         if "+bbb+" in enc_in:
             mes12=enc_in.split("+aaa+",1)[1]
@@ -313,7 +316,13 @@ def decrypt(passw,enc_in=None):
                 mes14=mes4.strip("b'").strip("'")
                 json_bytes = bytes(mes14,'utf-8')
                 print(f'im_bytes::{json_bytes}')
-                dec_json = fernet.decrypt(json_bytes).decode()       
+                private_key = RSA.import_key(open("private.pem").read())
+
+                cipher_rsa = PKCS1_OAEP.new(private_key)
+                dec_json = cipher_rsa.decrypt(json_bytes).decode()
+                
+                
+                #dec_json = fernet.decrypt(json_bytes).decode()       
 
     
     return(dec_im, mes_dec, dec_json)