Update app.py

app.py

@@ -85,7 +85,7 @@ def encrypt(passw,mes=None,img=None,doc=None,trans_s=None,trans_r=None,trans_a=N
         with open(enc_file, "wb") as file:
             file.write(enc_doc)
     
-    return enc_mes,enc_file,enc_qr
+    return enc_mes,enc_file,enc_qr,enc_qr
     
 def decrypt(passw,enc_in=None):
     key = create_key(passw)
@@ -189,7 +189,7 @@ def decode_qr(im,passw):
     with open(f'{im}', "rb") as image_file:
         bytes_i = base64.b64encode(image_file.read())     
     decode_qr = stegan.decode(im)
-    dec_im = decrypt(passw, enc_in=decode_qr)
+    dec_im,mes_dec,dec_json = decrypt(passw, enc_in=decode_qr)
     '''
     print(f'decoded QR::{decode_qr}')
     key = create_key(passw)
@@ -234,7 +234,7 @@ def decode_qr(im,passw):
             fh.close     
             dec_im = "finished_im.png"
     '''
-    return(dec_im,mes_dec)
+    return(dec_im,mes_dec,dec_json)
     
     
 with gr.Blocks() as app:
@@ -279,7 +279,7 @@ with gr.Blocks() as app:
 
     dec_qr_btn.click(decode_qr,[dec_qr_im,pass_out],[dec_qr_im_out,dec_qr_txt])
     dec_doc_btn.click(decode_doc,[pass_out,dec_doc_in],dec_doc_out)
-    en_btn.click(encrypt,[pass_in,mes,im,doc,send,rec,am],[enc_out,enc_doc_out,enc_qr_out])
+    en_btn.click(encrypt,[pass_in,mes,im,doc,send,rec,am],[enc_out,enc_doc_out,enc_qr_out,dec_qr_im])
     d_btn.click(decrypt,[pass_out,enc_in],[d_im,d_txt,d_json])
 app.launch()