Update app.py

app.py

@@ -117,9 +117,8 @@ def decrypt(passw=None,enc_in=None):
                 fh.write(base64.decodebytes(bytes(mes2, 'utf-8')))
             fh.close     
             dec_im = "finished_im.png"
-
-
     return(dec_im,mes_dec)
+
 def decode_doc(passw=None,doc=None):
     key = create_key(passw)
     fernet = Fernet(key)
@@ -140,8 +139,13 @@ def decode_doc(passw=None,doc=None):
     #enc_file=f'{doc.name}.ocrpt'
     with open(dec_file, "wb") as file:
         file.write(dec_doc)    
-
     return dec_file
+
+def decode_qr(im):
+    decode_qr = stegan.decode(im)
+    print(f'decoded QR::{decode_qr}')
+    dec_im,mes_dec=decrypt(enc_in=decode_qr)
+    return(dec_im,mes_dec)    
     
 with gr.Blocks() as app:
     with gr.Tab("Encrypt"):
@@ -165,10 +169,16 @@ with gr.Blocks() as app:
             d_txt = gr.Textbox(label="Decrypted")
             d_im =gr.Image(label="Decrytped Image")
         with gr.Tab("File"):
-            
             dec_doc_in = gr.File()
             dec_doc_btn = gr.Button("Decrypt")
             dec_doc_out=gr.File()
+        with gr.Tab("QR"):
+            dec_qr_im = gr.Image()
+            dec_qr_txt = gr.Textbox(label="Decrypted")
+            dec_qr_im_out =gr.Image(label="Decrytped Image")
+            dec_qr_btn = gr.Button("Decrypt")
+
+    dec_qr_btn.click(decode_qr,dec_qr_im,[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],[enc_out,enc_doc_out,enc_qr_out])
     d_btn.click(decrypt,[pass_out,enc_in],[d_im,d_txt])