Update app.py

app.py

@@ -4,7 +4,7 @@ from Crypto.Random import get_random_bytes
 from Crypto.Cipher import AES, PKCS1_OAEP
 from Crypto.Hash import RIPEMD160, SHA256
 import base58
-
+import stegan
 import stegan2
 import qr
 
@@ -25,21 +25,24 @@ def calculate_hash(data, hash_function: str = "sha256") -> str:
 def generate_keys():
     key = RSA.generate(2048)
     private_key = key.export_key('PEM')
-    file_out = open("private.pem", "wb")
-    file_out.write(private_key)
-    file_out.close()
+    file_out_priv = open("private.pem", "wb")
+    file_out_priv.write(private_key)
+    file_out_priv.close()
     
     public_key = key.publickey().export_key('PEM')
-    file_out = open("receiver.pem", "wb")
-    file_out.write(public_key)
-    file_out.close()
+    file_out_pub = open("receiver.pem", "wb")
+    file_out_pub.write(public_key)
+    file_out_pub.close()
 
+    qr_link="test"
+    priv_key = stegan.conv_im("private_key.png",data=file_out_priv)
+    pub_key = stegan.conv_im("public_key.png",data=file_out_pub)
     hash_1 = calculate_hash(public_key, hash_function="sha256")
     hash_2 = calculate_hash(hash_1, hash_function="ripemd160")
     address = base58.b58encode(hash_2)
     address_im=qr.make_qr(txt=address)
     
-    return public_key,private_key,address_im,address
+    return public_key,private_key,address_im,address,priv_key,pub_key
 
 def encrypt_text(data,in2,address):
     data = data.encode("utf-8")
@@ -105,9 +108,11 @@ with gr.Blocks() as app:
                     with gr.Column():
                         gen_wal_btn=gr.Button()
                         seed = gr.Textbox(label='Seed Phrase')
+                        img1=gr.Pil(label='Private Key')
                         out1 = gr.Textbox(label='Private Key')
+                        img2=gr.Pil(label='Public Key')
                         out2 = gr.Textbox(label='Public Key')
-                        img3=gr.Pil()
+                        img3=gr.Pil(label='Address')
                         out3 = gr.Textbox(label='Address')
             rsa_to_enc = gr.Textbox(label="txt to encrypt")
             rsa_enc_btn = gr.Button("RSA Encrypt")
@@ -117,7 +122,7 @@ with gr.Blocks() as app:
             rsa_dec_mes = gr.Textbox(label="decoded")
         gr.Column()
 
-    gen_wal_btn.click(generate_keys,None,[out2,out1, img3,out3])
+    gen_wal_btn.click(generate_keys,None,[out2,out1, img3,out3,img1,img2])
     rsa_enc_btn.click(encrypt_text,[rsa_to_enc,out2,out3],[rsa_enc_mes,qr_enc_mes])
     rsa_dec_btn.click(decrypt_text,[qr_enc_mes,out1],rsa_dec_mes)
 app.launch()