Update app.py

app.py

@@ -511,6 +511,7 @@ with gr.Blocks() as bc:
                     with gr.Tab("Encrypt"):
                         rsa_to_enc = gr.Textbox(label="txt to encrypt")
                         pub_key_in = gr.Image(label="Public Key", type="filepath")
+                        priv_key_in1 = gr.Image(label="Private Key(sig)", 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")
@@ -552,8 +553,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(create_new_chain,out3,[json_out,block_text,send,rec]).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)
+    gen_wal_btn.click(app_og.generate_keys,None,[out2,out1, img3,out3,img1,img2]).then(create_new_chain,out3,[json_out,block_text,send,rec]).then(app_og.test_fn,[img1,img2],[priv_key_in,pub_key_in,priv_key_in1])
+    rsa_enc_btn.click(app_og.encrypt_text,[rsa_to_enc,pub_key_in,priv_key_in1,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])