Update app.py

app.py

@@ -74,6 +74,7 @@ def encrypt_text(data,in2,address):
 def decrypt_text(im,in2):
     enc_in = stegan2.decode(im)
     private_key = RSA.import_key(open("private.pem").read())
+    private_key = stegan2.decode(in2)
     enc_session_key = enc_in[:private_key.size_in_bytes()]
     end1 = private_key.size_in_bytes()+16
     nonce = enc_in[private_key.size_in_bytes():end1]
@@ -103,8 +104,8 @@ with gr.Blocks() as app:
     with gr.Row():
         gr.Column()
         with gr.Column():
-            with gr.Box():
-                with gr.Row():
+            with gr.Tab("Gen Wal"):
+                with gr.Box():
                     with gr.Column():
                         gen_wal_btn=gr.Button()
                         seed = gr.Textbox(label='Seed Phrase')
@@ -114,16 +115,20 @@ with gr.Blocks() as app:
                         out2 = gr.Textbox(label='Public Key')
                         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")
+                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("Decrypt"):
+            mes_in = gr.Pil("Message")
+            priv_key_in = gr.Pil("Private Key")
+            
             rsa_dec_btn = gr.Button("RSA Decrypt")
             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,out2,out3],[rsa_enc_mes,qr_enc_mes])
-    rsa_dec_btn.click(decrypt_text,[qr_enc_mes,out1],rsa_dec_mes)
-app.launch()
-
+    rsa_dec_btn.click(decrypt_text,[qr_enc_mes,img1],rsa_dec_mes)
+app.launch()