Update app.py

app.py

@@ -1,40 +1,33 @@
 
 import gradio as gr
 from cryptography.fernet import Fernet
-def encrypt():
-    # we will be encrypting the below string.
-    message = "hello geeks"
-     
-    # generate a key for encryption and decryption
-    # You can use fernet to generate
-    # the key or use random key generator
-    # here I'm using fernet to generate key
-     
+def encrypt(message=None,image=None,document=None):
+
+    if message != None and message != "":
+        bytes = message.encode()
+        print (bytes)
+        
     key = Fernet.generate_key()
-     
-    # Instance the Fernet class with the key
+    print (f"The Key is: {key}")     
      
     fernet = Fernet(key)
-     
+    
     # then use the Fernet class instance
     # to encrypt the string string must
     # be encoded to byte string before encryption
-    encMessage = fernet.encrypt(message.encode())
+    encMessage = fernet.encrypt(bytes)
      
     print("original string: ", message)
     print("encrypted string: ", encMessage)
      
-    # decrypt the encrypted string with the
-    # Fernet instance of the key,
-    # that was used for encrypting the string
-    # encoded byte string is returned by decrypt method,
-    # so decode it to string with decode methods
+
     decMessage = fernet.decrypt(encMessage).decode()
      
     print("decrypted string: ", decMessage)
 
 with gr.Blocks() as app:
+    mes = gr.Textbox()
     btn = gr.Button()
-    btn.click(encrypt,None,None)
+    btn.click(encrypt,mes,None)
 app.launch()