Update app.py

app.py

@@ -45,10 +45,18 @@ def encrypt(passw=None,mes=None,img=None,doc=None):
             enc_mes = f'{enc_mes}{im_bytes}+bbb+'
         else:
             enc_mes = f'{im_bytes}+bbb+'
-    print (enc_mes)
-    
-    return enc_mes
 
+    if doc != None:
+        with open(doc, "rb") as file:
+            # read all file data
+            file_data = file.read()
+        enc_doc = fernet.encrypt(file_data)
+        enc_file=f'{doc}.ocrpt'
+        with open(enc_file, "wb") as file:
+            file.write(enc_doc)
+        
+    return enc_mes,enc_file
+    
 def decrypt(passw=None,enc_in=None):
     key = create_key(passw)
     fernet = Fernet(key)
@@ -100,7 +108,9 @@ with gr.Blocks() as app:
     with gr.Tab("Encrypt"):
         pass_in=gr.Textbox(label="Set Password")
         mes = gr.Textbox(label = "Message")
-        im = gr.Image(type="filepath")
+        with gr.Row():
+            im = gr.Image(type="filepath")
+            doc=gr.Files()
         en_btn = gr.Button("Encrypt")
         enc_out = gr.Textbox(label="Encrypted Bytes")
     with gr.Tab("Decrypt"):
@@ -110,7 +120,7 @@ with gr.Blocks() as app:
         d_txt = gr.Textbox(label="Decrypted")
         d_im =gr.Image(label="Decrytped Image")
         d_txt=gr.Textbox()
-    en_btn.click(encrypt,[pass_in,mes,im],enc_out)
+    en_btn.click(encrypt,[pass_in,mes,im,doc],enc_out)
     d_btn.click(decrypt,[pass_out,enc_in],[d_im,d_txt])
 app.launch()