Update app.py

app.py

@@ -23,6 +23,7 @@ def create_key(passw):
         salt=salt,
         iterations=480000,
     )
+    key = base64.urlsafe_b64encode(kdf.derive(key))
 
     return kdf
 
@@ -38,7 +39,20 @@ def encrypt(passw=None,mes=None,img=None,document=None):
         print (bytes_e)
     #key = Fernet.generate_key()
         
-    key = create_key(passw)  
+    #key = create_key(passw)
+    key = passw.encode()
+    h = hashlib.new('sha256')
+    h.update(key)
+    key = h.hexdigest()
+    key = key.encode()
+    #salt = os.urandom(16)
+    salt = key
+    kdf = PBKDF2HMAC(
+        algorithm=hashes.SHA256(),
+        length=32,
+        salt=salt,
+        iterations=480000,
+    )
     key = base64.urlsafe_b64encode(kdf.derive(key))
     print (f'key == {key}')
     
@@ -57,9 +71,20 @@ def encrypt(passw=None,mes=None,img=None,document=None):
 def decrypt(passw=None,enc_in=None):
     enc_in=enc_in.strip('"')
     print (f'encoded string == {enc_in}')
-    key = create_key(passw)
+    key = passw.encode()
+    h = hashlib.new('sha256')
+    h.update(key)
+    key = h.hexdigest()
+    key = key.encode()
+    #salt = os.urandom(16)
+    salt = key
+    kdf = PBKDF2HMAC(
+        algorithm=hashes.SHA256(),
+        length=32,
+        salt=salt,
+        iterations=480000,
+    )
     key = base64.urlsafe_b64encode(kdf.derive(key))
-    print (f'key == {key}')
     
     fernet = Fernet(key)
     decMessage = fernet.decrypt(enc_in).decode()