Update app.py

app.py

@@ -89,16 +89,30 @@ def encrypt_text1(data,in2,address):
     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()
-    out_file=[]
-    #[ out_file.append(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+'
-    print(f'out_string::{out_string}')
-    #output=open("encrypted_data.bin", "rb")
+################################
+    #doc_name = doc.name
+    #doc_name = doc_name.split("/",4)[4]
+    doc_name = "encrypted_data.bin"
+    with open(doc.name, "rb") as file:
+        # read all file data
+        file_data = file.read()
+    qr_link="test"
+    enc_qr = stegan2.conv_im(qr_link=qr_link,data=enc_mes)
+    
+    
+    #enc_doc = fernet.encrypt(file_data)
+    #og_name = doc.name
+    #og_end = og_name.split(".",1)[1]
+    #og_front=og_name.split(".",1)[0]
     
     
+    #enc_file=f'{doc.name}.ocrpt'
+    #with open(enc_file, "wb") as file:
+    #    file.write(enc_doc)
+########################################    
+    
     #output1 = pickle.dumps(output)
-    enc_qr = stegan2.conv_im(qr_link=address,data=out_file)
+    #enc_qr = stegan2.conv_im(qr_link=address,data=out_file)
     
     #out = output
     #output.close()
@@ -149,10 +163,12 @@ def decrypt_text1(im,in2):
     #[ file_out.write(x) for x in (mes_bytes1, mes_bytes2, mes_bytes3, mes_bytes4) ]
     #file_out.close()
 
-
-    
+    enc_in = stegan2.decode(im)
+    with open("encrypted_data1.bin", "wb") as file:
+        file.write(enc_in)   
+    file.close()
     
-    file_in = open("encrypted_data.bin", "rb")
+    file_in = open("encrypted_data1.bin", "rb")
     
     private_key = RSA.import_key(open("private.pem").read())
     enc_in=enc_in.strip('"')