Update app.py

app.py

@@ -7,6 +7,10 @@ from Crypto.PublicKey import RSA
 from Crypto.Cipher import PKCS1_OAEP
 import binascii
 
+from Crypto.PublicKey import RSA
+from Crypto.Cipher import PKCS1_v1_5
+from base64 import b64decode
+
 
 import os
 import base64
@@ -17,6 +21,10 @@ import qr
 import stegan
 import json
 import re
+
+
+
+
 eth_wal = gr.Interface.load("spaces/Omnibus/decentral-test-mod")
 
 def rsa_encrypt_eth():
@@ -49,13 +57,24 @@ def rsa_generate():
     pubcipher = PKCS1_OAEP.new(pubKey)
     out1 = f'{pubKey.n}{keyPair.d}'
     out2 = f'{pubKey.n}{keyPair.e}'
-    return privcipher,pubcipher,output,"qr_wal_tmp.png",output
+    return privKeyPEM,pubKeyPEM,output,"qr_wal_tmp.png",output
+
+
+
+
     
 def rsa_encrypt(msg,pubKey):
-    pubKey=pubKey.encode("utf-8")
+
+    key = b64decode(pubKey)
+    private_key = RSA.importKey(key)
+    cipher = PKCS1_v1_5.new(private_key)
+    cipher.encrypt(msg)
+
+    
+    #pubKey=pubKey.encode("utf-8")
     print (pubKey)
     msg = b'A message for encryption'
-    encryptor = PKCS1_OAEP.new(pubKey)
+    encryptor = PKCS1_OAEP.new(cipher)
     encrypted = encryptor.encrypt(msg)
     print("Encrypted:", binascii.hexlify(encrypted))
     return binascii.hexlify(encrypted)