Update app.py

app.py

@@ -17,51 +17,55 @@ with gr.Blocks() as app:
         pass_box = gr.Textbox()
         pass_btn = gr.Button()
         
-    with gr.Box(visible=False) as valida:    
-        with gr.Tab("Messages"):
-            with gr.Accordion("Key"):
-                with gr.Row():
-                    input_address = gr.Image(label="Wallet",type="filepath")
-                    input_key = gr.Image(label="Key",type="filepath")
-                    
+    with gr.Box(visible=False) as valida: 
+        gr.Column()
+        with gr.Column():
             with gr.Row():
-                with gr.Column():
-                    rec=gr.Dropdown(label="Recipient", choices=[f for f in rec_list], allow_custom_value=True)
-                    send_mes=gr.Textbox(label="Message", lines=6)
-                    send_mes_btn=gr.Button()
-                rec_mes = gr.Textbox(lines=6)
+                with gr.Tab("Messages"):
+                    with gr.Accordion("Key"):
+                        input_key = gr.Image(label="Key",type="filepath")
+                            
+                    with gr.Row():
+                        with gr.Column():
+                            rec=gr.Dropdown(label="Recipient", choices=[f for f in rec_list], allow_custom_value=True)
+                            send_mes=gr.Textbox(label="Message", lines=6)
+                            send_mes_btn=gr.Button()
+                        rec_mes = gr.Textbox(lines=6)
+            
+            
+                        
+                with gr.Tab("BC"):
+                    with gr.Row():
+                        with gr.Tab("Gen Wal"):
+                            gen_wal_btn=gr.Button()
+                            seed = gr.Textbox(label='Seed Phrase')
+                            img1=gr.Pil(label='Private Key')
+                            out1 = gr.Textbox(label='Private Key',max_lines=4)
+                            img2=gr.Pil(label='Public Key')
+                            out2 = gr.Textbox(label='Public Key',max_lines=4)
+                            img3=gr.Pil(label='Address')
+                            out3 = gr.Textbox(label='Address')
+                        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")
+                        with gr.Tab("Decrypt"):
+                            mes_in = gr.Image(label="Message", type="filepath")
+                            priv_key_in = gr.Image(label="Private Key", type="filepath")
+                            rsa_dec_btn = gr.Button("RSA Decrypt")
+                            rsa_dec_mes = gr.Textbox(label="decoded")  
+        gr.Column()
+    def test_fn(im):
+        return (im)
     
-    
-                
-        with gr.Tab("BC"):
-            with gr.Row():
-                with gr.Tab("Gen Wal"):
-                    gen_wal_btn=gr.Button()
-                    seed = gr.Textbox(label='Seed Phrase')
-                    img1=gr.Pil(label='Private Key')
-                    out1 = gr.Textbox(label='Private Key',max_lines=4)
-                    img2=gr.Pil(label='Public Key')
-                    out2 = gr.Textbox(label='Public Key',max_lines=4)
-                    img3=gr.Pil(label='Address')
-                    out3 = gr.Textbox(label='Address')
-                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")
-                with gr.Tab("Decrypt"):
-                    mes_in = gr.Image(label="Message", type="filepath")
-                    priv_key_in = gr.Image(label="Private Key", type="filepath")
-                    rsa_dec_btn = gr.Button("RSA Decrypt")
-                    rsa_dec_mes = gr.Textbox(label="decoded")  
-
     input_key.change(crypt.address,input_key,None)
     
     pass_btn.click(checkp,pass_box,[invalid,valida])
 
-    gen_wal_btn.click(crypt.generate_keys,None,[out2,out1,img3,out3,img1,img2]).then(crypt.test_fn,[img1,img3],[input_key,input_address])
+    gen_wal_btn.click(crypt.generate_keys,None,[out2,out1,img3,out3,img1,img2]).then(test_fn,[img1],[input_key])
     rsa_enc_btn.click(crypt.encrypt_text,[rsa_to_enc,pub_key_in,priv_key_in1,out3],[rsa_enc_mes,qr_enc_mes])
     rsa_dec_btn.click(crypt.decrypt_text,[mes_in,priv_key_in],rsa_dec_mes)