Update app.py

app.py

@@ -38,6 +38,72 @@ def rsa_encrypt_eth():
     print("Address:", acct.address)
 
 
+'''
+from Crypto.PublicKey import RSA
+from Crypto.Cipher import PKCS1_OAEP
+from Crypto.Random import get_random_bytes
+
+def generate_keys():
+    key = RSA.generate(2048)
+    private_key = key.export_key()
+    public_key = key.publickey().export_key()
+    return public_key, private_key
+
+def encrypt_text(text, public_key):
+    rsa_public_key = RSA.import_key(public_key)
+    rsa_public_key = PKCS1_OAEP.new(rsa_public_key)
+    encrypted_text = rsa_public_key.encrypt(text)
+    return encrypted_text
+
+def decrypt_text(encrypted_text, private_key):
+    rsa_private_key = RSA.import_key(private_key)
+    rsa_private_key = PKCS1_OAEP.new(rsa_private_key)
+    decrypted_text = rsa_private_key.decrypt(encrypted_text)
+    return decrypted_text
+
+# Generate RSA keys
+public_key, private_key = generate_keys()
+
+# Encrypt text
+text = "Hello, World!".encode()
+encrypted_text = encrypt_text(text, public_key)
+print(f"Encrypted text: {encrypted_text}")
+
+# Decrypt text
+decrypted_text = decrypt_text(encrypted_text, private_key)
+print(f"Decrypted text: {decrypted_text.decode()}")'''
+
+def generate_keys():
+    key = RSA.generate(2048)
+    private_key = key.export_key()
+    public_key = key.publickey().export_key()
+    return public_key, private_key
+
+def save_keys(public_key, private_key, filename):
+    with open(filename, 'w') as f:
+        f.write(public_key)
+        f.write('\n')
+        f.write(private_key)
+
+def read_keys(filename):
+    with open(filename, 'r') as f:
+        return f.read()
+
+def encrypt_text(text, public_key):
+    rsa_public_key = RSA.import_key(public_key.decode())
+    rsa_cipher = PKCS1_OAEP.new(rsa_public_key)
+    encrypted_text = rsa_cipher.encrypt(text.encode())
+    return encrypted_text
+
+def decrypt_text(encrypted_text, private_key):
+    rsa_private_key = RSA.import_key(private_key.decode())
+    sequence = [0]*256
+    sequence[0] = 1
+    rsa_cipher = PKCS1_OAEP.new(rsa_private_key, mode='encrypt-decrypt')
+    decrypted_text = rsa_cipher.decrypt(encrypted_text, sequence)
+    return decrypted_text.decode()
+
+    
 def rsa_generate():
     key = RSA.generate(1024)
     f = open('keyfile_private.pem', 'wb')
@@ -312,10 +378,10 @@ 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(rsa_encrypt,[rsa_to_enc,out2],rsa_enc_mes)
-    rsa_dec_btn.click(rsa_decrypt,[rsa_enc_mes,out1],rsa_dec_mes)
+    rsa_enc_btn.click(encrypt_text,[rsa_to_enc,out2],rsa_enc_mes)
+    rsa_dec_btn.click(decrypt_text,[rsa_enc_mes,out1],rsa_dec_mes)
     
-    gen_wal_btn.click(rsa_generate,None,[out1,out2,out3,img3,seed])
+    gen_wal_btn.click(generate_keys,None,[out1,out2,out3,img3,seed])
     dec_qr_btn.click(decode_qr,[dec_qr_im,pass_out],[dec_qr_im_out,dec_qr_txt,d_json])
     en_btn.click(encrypt,[pass_in,send,rec,am],[enc_out,enc_qr_out,dec_qr_im])
     url_btn.click(fn=predict, inputs=[text_input,url_params],outputs=[text_output,url_params], _js=get_window_url_params)