Update app.py

app.py

@@ -47,7 +47,7 @@ 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+'
-
+    enc_qr = qr.make_qr(data=enc_mes)
     if doc != None:
         
         print(dir(doc))
@@ -70,8 +70,8 @@ def encrypt(passw=None,mes=None,img=None,doc=None):
         enc_file=f'{doc.name}.ocrpt'
         with open(enc_file, "wb") as file:
             file.write(enc_doc)
-        
-    return enc_mes,enc_file
+    
+    return enc_mes,enc_file,enc_qr
     
 def decrypt(passw=None,enc_in=None):
     key = create_key(passw)
@@ -144,13 +144,17 @@ def decode_doc(passw=None,doc=None):
 with gr.Blocks() as app:
     with gr.Tab("Encrypt"):
         pass_in=gr.Textbox(label="Set Password")
-        mes = gr.Textbox(label = "Message")
-        with gr.Row():
-            im = gr.Image(type="filepath")
-            doc=gr.File()
+        with gr.Tab("String"):
+            mes = gr.Textbox(label = "Message")
+            with gr.Row():
+                im = gr.Image(type="filepath")
+                doc=gr.File()
+        with gr.Tab("File"):
+            enc_doc_out=gr.File()
         en_btn = gr.Button("Encrypt")
         enc_out = gr.Textbox(label="Encrypted Bytes")
-        enc_doc_out=gr.File()
+        enc_qr_out = gr.Image("Encrypted QR")
+
     with gr.Tab("Decrypt"):
         pass_out = gr.Textbox(label="Enter Password")
         with gr.Tab("String"):
@@ -164,7 +168,7 @@ with gr.Blocks() as app:
             dec_doc_btn = gr.Button("Decrypt")
             dec_doc_out=gr.File()
     dec_doc_btn.click(decode_doc,[pass_out,dec_doc_in],dec_doc_out)
-    en_btn.click(encrypt,[pass_in,mes,im,doc],[enc_out,enc_doc_out])
+    en_btn.click(encrypt,[pass_in,mes,im,doc],[enc_out,enc_doc_out,enc_qr_out])
     d_btn.click(decrypt,[pass_out,enc_in],[d_im,d_txt])
 app.launch()