Update app.py

app.py

@@ -1,6 +1,9 @@
 
 import gradio as gr
 from cryptography.fernet import Fernet
+from cryptography.hazmat.primitives import hashes
+from cryptography.hazmat.primitives.kdf.pbkdf2 import PBKDF2HMAC
+import os
 import base64
 from hashlib import blake2b
 
@@ -16,29 +19,16 @@ def encrypt(mes=None,img=None,document=None):
     key = Fernet.generate_key()
     key = "potato"
     key = key.encode()
-    key_len = len(key)
-
-    print (f"length0:{key_len}")
-    
-    print (f"The Key is: {key}")   
-    h = blake2b(digest_size=32)
-    h.update(key)
-    key = h.hexdigest()
-    key = key.encode()
-    key_len = len(key)
-    print (f"length1:{key_len}")
+    password=key
+    salt = os.urandom(16)
+    kdf = PBKDF2HMAC(
+        algorithm=hashes.SHA256(),
+        length=32,
+        salt=salt,
+        iterations=480000,
+    )
+    key = base64.urlsafe_b64encode(kdf.derive(password))
     
-    print (f"The Key is: {key}")     
-
-    
-    #key = Key.fromUtf8(base64Url.encode(key.bytes))
-    #base64.urlsafe_b64encode(os.urandom(32))
-    key = base64.urlsafe_b64encode(key)
-    
-    key_len = len(key)
-    print (f"length2:{key_len}")
-    print (f"The Key is: {key}")     
-     
     fernet = Fernet(key)
     
     # then use the Fernet class instance