Update app.py

app.py

@@ -47,11 +47,14 @@ def generate_keys():
     return public_key,private_key,address_im,address,priv_key,pub_key
 
 def encrypt_text(data,in2,address):
+    pub_key = stegan2.decode(in2)
+    
     data = data.encode("utf-8")
     #data = "I met aliens in UFO. Here is the map.".encode("utf-8")
-    file_out = open("encrypted_data.bin", "wb")
     
-    recipient_key = RSA.import_key(open("receiver.pem").read())
+    #recipient_key = RSA.import_key(open("receiver.pem").read())
+    recipient_key = RSA.import_key(pub_key)
+
     session_key = get_random_bytes(16)
     
     # Encrypt the session key with the public RSA key
@@ -61,6 +64,9 @@ def encrypt_text(data,in2,address):
     # Encrypt the data with the AES session key
     cipher_aes = AES.new(session_key, AES.MODE_EAX)
     ciphertext, tag = cipher_aes.encrypt_and_digest(data)
+
+    file_out = open("encrypted_data.bin", "wb")
+    
     [ file_out.write(x) for x in (enc_session_key, cipher_aes.nonce, tag, ciphertext) ]
     file_out.close()
     doc_name = "encrypted_data.bin"
@@ -116,16 +122,22 @@ with gr.Blocks() as app:
                         gen_wal_btn=gr.Button()
                         seed = gr.Textbox(label='Seed Phrase')
                         img1=gr.Pil(label='Private Key')
-                        out1 = gr.Textbox(label='Private Key')
+                        out1 = gr.Textbox(label='Private Key',max_lines=4)
                         img2=gr.Pil(label='Public Key')
-                        out2 = gr.Textbox(label='Public Key')
+                        out2 = gr.Textbox(label='Public Key',max_lines=4)
                         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")
-                rsa_enc_mes = gr.Textbox(label="encoded")
-                qr_enc_mes = gr.Image(type="filepath")
 
+
+        with gr.Tab("Encrypt"):
+            
+            rsa_to_enc = gr.Textbox(label="txt to encrypt")
+            pub_key_in = gr.Image(label="Public Key", type="filepath")
+            rsa_enc_btn = gr.Button("RSA Encrypt")
+            rsa_enc_mes = gr.Textbox(label="encoded", max_lines=4)
+            qr_enc_mes = gr.Image(type="filepath")
+            
+           
         with gr.Tab("Decrypt"):
             mes_in = gr.Image(label="Message", type="filepath")
             priv_key_in = gr.Image(label="Private Key", type="filepath")
@@ -135,6 +147,6 @@ with gr.Blocks() as app:
         gr.Column()
 
     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_enc_btn.click(encrypt_text,[rsa_to_enc,pub_key_in,out3],[rsa_enc_mes,qr_enc_mes])
     rsa_dec_btn.click(decrypt_text,[mes_in,priv_key_in],rsa_dec_mes)
 app.launch()