Update app.py

app.py

@@ -22,7 +22,7 @@ def calculate_hash(data, hash_function: str = "sha256") -> str:
         return h.hexdigest()
 
 
-def generate_keys1():
+def generate_keys():
     key = RSA.generate(2048)
     private_key = key.export_key('PEM')
     file_out = open("private.pem", "wb")
@@ -41,7 +41,7 @@ def generate_keys1():
     
     return public_key,private_key,address_im,address
 
-def encrypt_text1(data,in2,address):
+def encrypt_text(data,in2,address):
     data = data.encode("utf-8")
     #data = "I met aliens in UFO. Here is the map.".encode("utf-8")
     file_out = open("encrypted_data.bin", "wb")
@@ -68,7 +68,7 @@ def encrypt_text1(data,in2,address):
     file.close()
     return str(file_data),enc_qr
 
-def decrypt_text1(im,in2):
+def decrypt_text(im,in2):
     enc_in = stegan2.decode(im)
     private_key = RSA.import_key(open("private.pem").read())
     enc_session_key = enc_in[:private_key.size_in_bytes()]
@@ -96,14 +96,15 @@ def decrypt_text1(im,in2):
 
 
 with gr.Blocks() as app:
-    gen_wal_btn=gr.Button()
-    seed = gr.Textbox(label='Seed Phrase')
+
     with gr.Row():
         gr.Column()
         with gr.Column():
             with gr.Box():
                 with gr.Row():
                     with gr.Column():
+                        gen_wal_btn=gr.Button()
+                        seed = gr.Textbox(label='Seed Phrase')
                         out1 = gr.Textbox(label='Private Key')
                         out2 = gr.Textbox(label='Public Key')
                         img3=gr.Pil()
@@ -115,7 +116,9 @@ with gr.Blocks() as app:
             rsa_dec_btn = gr.Button("RSA Decrypt")
             rsa_dec_mes = gr.Textbox(label="decoded")
         gr.Column()
-    rsa_enc_btn.click(encrypt_text1,[rsa_to_enc,out2,out3],[rsa_enc_mes,qr_enc_mes])
-    rsa_dec_btn.click(decrypt_text1,[qr_enc_mes,out1],rsa_dec_mes)
+
+    gen_wal_btn.click(generate_keys,None,[out2,out1, img3,out3])
+    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()