Update app.py

app.py

@@ -121,8 +121,10 @@ def decrypt_text(im,in2):
     data = cipher_aes.decrypt_and_verify(ciphertext, tag)
     return(data.decode("utf-8"))
 
-
-
+def test_fn(im1,im2):
+    return im1,im2
+def test_fn2(im1):
+    return im1
 with gr.Blocks() as app:
 
     with gr.Row():
@@ -151,7 +153,9 @@ 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,img1,img2])
-    rsa_enc_btn.click(encrypt_text,[rsa_to_enc,pub_key_in,out3],[rsa_enc_mes,qr_enc_mes])
+    
+    
+    gen_wal_btn.click(generate_keys,None,[out2,out1, img3,out3,img1,img2]).then(test_fn,[img1,img2],[priv_key_in,pub_key_in])
+    rsa_enc_btn.click(encrypt_text,[rsa_to_enc,pub_key_in,out3],[rsa_enc_mes,qr_enc_mes]).then(test_fn2,qr_enc_mes,mes_in)
     rsa_dec_btn.click(decrypt_text,[mes_in,priv_key_in],rsa_dec_mes)
 app.launch()