Update app_og.py

app_og.py

@@ -56,10 +56,27 @@ def generate_keys():
     
     return public_key,private_key,address_im,address,priv_key,pub_key
 
-def encrypt_text(data,in2,address):
-    pub_key = stegan2.decode(in2)
+def sign(data,in2):
+    priv_key = stegan2.decode(in2)
+    print(f'priv_key:: {priv_key}')
+    private_key = RSA.import_key(priv_key)    
+
+    hash_object = SHA256.new(data)
+    signature = pkcs1_15.new(private_key).sign(hash_object)
+    signature = binascii.hexlify(signature).decode("utf-8")
+    data_json = {
+        "data": data,
+        "signature": signature
+    }
+    return data_json
+    
+def encrypt_text(data,pub_im,priv_im,address):
+    pub_key = stegan2.decode(pub_im)
     
     data = data.encode("utf-8")
+    data = sign(data,priv_im)
+    
+    
     recipient_key = RSA.import_key(pub_key)
 
     session_key = get_random_bytes(16)