Update app.py

app.py

@@ -74,7 +74,7 @@ def generate_keys1():
 def encrypt_text1(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")
+    #file_out = open("encrypted_data.bin", "wb")
     
     recipient_key = RSA.import_key(open("receiver.pem").read())
     session_key = get_random_bytes(16)
@@ -86,11 +86,11 @@ def encrypt_text1(data,in2,address):
     # Encrypt the data with the AES session key
     cipher_aes = AES.new(session_key, AES.MODE_EAX)
     ciphertext, tag = cipher_aes.encrypt_and_digest(data)
-    [ file_out.write(x) for x in (enc_session_key, cipher_aes.nonce, tag, ciphertext) ]
-    file_out.close()
+    #[ file_out.write(x) for x in (enc_session_key, cipher_aes.nonce, tag, ciphertext) ]
+    #file_out.close()
     
     
-    out_string = f'{enc_session_key}+++aaa+++{cipher_aes.nonce}+++bbb+++{tag}+++ccc+++{ciphertext}+++ddd+++'
+    out_string = f'{enc_session_key}+aaa+{cipher_aes.nonce}+bbb+{tag}+ccc+{ciphertext}+ddd+'
     
     #output=open("encrypted_data.bin", "rb")
     
@@ -102,14 +102,37 @@ def encrypt_text1(data,in2,address):
     #output.close()
     return str(out_string),enc_qr
 
-def decrypt_text1(in1,in2):
-    #in1= in1.encode("utf-8")
-    #file_out = open("encrypted_data2.bin","wb")
-    #[ file_out.write(x) for x in (in1.enc_session_key, in1.cipher_aes.nonce, in1.tag, in1.ciphertext) ]
+def decrypt_text1(im,in2):
 
-    #file_out.write(in1)
-    #[ file_out.write(x) for x in (enc_session_key, cipher_aes.nonce, tag, ciphertext) ]
-    #file_out.close()
+    enc_in = stegan.decode(im)
+    if "+aaa+" in enc_in:
+        mes1=enc_in.split("+aaa+",1)[0]
+        mes1=mes1.strip("b'").strip("'")
+        mes_bytes = bytes(mes1,'utf-8')
+        #mes_dec = fernet.decrypt(mes_bytes).decode()
+    if "+bbb+" in decode_qr:
+        mes2=enc_in.split("+aaa+",1)[1]
+        mes2=mes2.split("+bbb+",1)[0]
+        mes2=mes2.strip("b'").strip("'")
+        mes_bytes2 = bytes(mes2,'utf-8')
+    if "+ccc+" in decode_qr:
+        mes3=enc_in.split("+bbb+",1)[1]
+        mes3=mes3.split("+bbb+",1)[0]
+        mes3=mes3.strip("b'").strip("'")
+        mes_bytes3 = bytes(mes3,'utf-8')
+    if "+ddd+" in decode_qr:
+        mes4=enc_in.split("+ccc+",1)[1]
+        mes4=mes4.split("+ddd+",1)[0]
+        mes4=mes4.strip("b'").strip("'")
+        mes_bytes4 = bytes(mes4,'utf-8')        
+
+    file_out = open("encrypted_data.bin", "wb")
+    [ file_out.write(x) for x in (mes1, mes2, mes3, mes4) ]
+    file_out.close()
+
+
+    
+    
     file_in = open("encrypted_data.bin", "rb")
     
     private_key = RSA.import_key(open("private.pem").read())