Update app.py

app.py

@@ -38,63 +38,16 @@ def encrypt(passw=None,mes=None,img=None,document=None):
             bytes_e = base64.b64encode(image_file.read()) 
         print (bytes_e)
     #key = Fernet.generate_key()
-        
     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)
-    
-    # then use the Fernet class instance
-    # to encrypt the string string must
-    # be encoded to byte string before encryption
-    #global encMessage
     encMessage = fernet.encrypt(bytes_e)
-    print(f'Length in Bytes: {len(encMessage)}')
-    #print("original string: ", mes)
-    print("encrypted string: ", encMessage)
-    print (f'encoded string == {encMessage}')
-    
+  
     return encMessage
+
 def decrypt(passw=None,enc_in=None):
     enc_in=enc_in.strip('"').strip("b'").strip("'")
-    print (f'enc_in stripped == {enc_in}')
     enc_in = bytes(enc_in,'utf-8')
-    print (f'enc_in encoded == {enc_in}')
-    
-    #enc_in=encMessage
-    #enc_in=bytes(enc_in,'utf-8')
-    #print (f'encoded string == {enc_in}')
-    '''
-    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))
-    '''
+
     key = create_key(passw)
     
     fernet = Fernet(key)
@@ -109,6 +62,9 @@ def decrypt(passw=None,enc_in=None):
     except Exception:
         dec_im = None
     return(dec_im,decMessage)
+
+
+
 with gr.Blocks() as app:
     with gr.Tab("Encrypt"):
         pass_in=gr.Textbox(label="Set Password")