Update app.py

app.py

@@ -541,8 +541,8 @@ with gr.Blocks() as bc:
             return e,None,None,e
 
 
-    gen_wal_btn.click(app_og.generate_keys,None,[out2,out1, img3,out3,img1,img2]).then(test_fn,[img1,img2],[priv_key_in,pub_key_in])
-    rsa_enc_btn.click(app_og.encrypt_text,[rsa_to_enc,pub_key_in,out3],[rsa_enc_mes,qr_enc_mes]).then(test_fn2,qr_enc_mes,mes_in)
+    gen_wal_btn.click(app_og.generate_keys,None,[out2,out1, img3,out3,img1,img2]).then(app_og.test_fn,[img1,img2],[priv_key_in,pub_key_in])
+    rsa_enc_btn.click(app_og.encrypt_text,[rsa_to_enc,pub_key_in,out3],[rsa_enc_mes,qr_enc_mes]).then(app_og.test_fn2,qr_enc_mes,mes_in)
     rsa_dec_btn.click(app_og.decrypt_text,[mes_in,priv_key_in],rsa_dec_mes)
             
     check_trans.click(deep_trans,None,[block_text,out_box_bool])