Update app.py

app.py

@@ -73,15 +73,74 @@ def decrypt(passw,enc_in=None):
     print (f'enc_in :::: {enc_in}')
     
 
-    if "+ccc+" in enc_in:
-        mes4=enc_in.split("+ccc+",1)[0]
-        mes14=mes4.strip("b'").strip("'")
-        json_bytes = bytes(mes14,'utf-8')
-        print(f'im_bytes::{json_bytes}')
-        dec_json = fernet.decrypt(json_bytes).decode()        
+    
+    if "+aaa+" in enc_in:
+        mes1=enc_in.split("+aaa+",1)[0]
+        mes1=mes1.strip("b'").strip("'")
+        mes_bytes = bytes(mes1,'utf-8')
+        mes_dec = fernet.decrypt(mes_bytes).decode()
+        if "+bbb+" in enc_in:
+            mes12=enc_in.split("+aaa+",1)[1]
+            mes2=mes12.split("+bbb+",1)[0]
+            mes2=mes2.strip("b'").strip("'")
+            im_bytes = bytes(mes2,'utf-8')
+            print(f'im_bytes::{im_bytes}')
+            
+            mes2 = fernet.decrypt(mes2).decode()
+            #base = bytes(decMessage, 'utf-8')
+            with open(f"finished_im.png", "wb") as fh:
+                #fh.write(base64.decodebytes(im_bytes))
+                fh.write(base64.decodebytes(bytes(mes2, 'utf-8')))
+            fh.close     
+            dec_im = "finished_im.png"
+            if "+ccc+" in enc_in:
+                mes13=enc_in.split("+bbb+",1)[1]
+                mes3=mes13.split("+ccc+",1)[0]
+                mes3=mes3.strip("b'").strip("'")
+                json_bytes = bytes(mes3,'utf-8')
+                print(f'im_bytes::{json_bytes}')
+                dec_json = fernet.decrypt(json_bytes).decode()
+        if not "+bbb+" in enc_in:
+            if "+ccc+" in enc_in:
+                mes14=enc_in.split("+aaa+",1)[1]
+                
+                mes4=mes14.split("+ccc+",1)[0]
+                mes4=mes4.strip("b'").strip("'")
+                json_bytes = bytes(mes4,'utf-8')
+                print(f'im_bytes::{json_bytes}')
+                dec_json = fernet.decrypt(json_bytes).decode()
+   
+    if not "+aaa+" in enc_in:
+        if "+bbb+" in enc_in:
+            mes2 = enc_in.split("+bbb+",1)[0]
+            mes2=mes2.strip("b'").strip("'")
+            im_bytes = bytes(mes2,'utf-8')
+            print(f'im_bytes2::{im_bytes}')
+            mes2 = fernet.decrypt(mes2).decode()
+
+            #base = bytes(decMessage, 'utf-8')
+            with open(f"finished_im.png", "wb") as fh:
+                #fh.write(base64.decodebytes(im_bytes))
+                fh.write(base64.decodebytes(bytes(mes2, 'utf-8')))
+            fh.close     
+            dec_im = "finished_im.png"
+            if "+ccc+" in enc_in:
+                mes13=enc_in.split("+bbb+",1)[1]
+                mes3=mes13.split("+ccc+",1)[0]
+                mes3=mes3.strip("b'").strip("'")
+                json_bytes = bytes(mes3,'utf-8')
+                print(f'im_bytes::{json_bytes}')
+                dec_json = fernet.decrypt(json_bytes).decode()
+        if not "+bbb+" in enc_in:
+            if "+ccc+" in enc_in:
+                mes4=enc_in.split("+ccc+",1)[0]
+                mes14=mes4.strip("b'").strip("'")
+                json_bytes = bytes(mes14,'utf-8')
+                print(f'im_bytes::{json_bytes}')
+                dec_json = fernet.decrypt(json_bytes).decode()       
 
     
-    return(dec_json)
+    return(dec_im, mes_dec, dec_json)
 
 def decode_doc(passw,doc=None):
     key = create_key(passw)
@@ -135,11 +194,11 @@ with gr.Blocks() as app:
                 pass_out = gr.Textbox(label="Enter Password")
         
                 dec_qr_im = gr.Image(type="filepath")
-
                 d_json = gr.JSON(label="Decrypted JSON")
-    
+                dec_qr_txt = gr.Textbox(label="Decrypted Message")
+                dec_qr_im_out =gr.Image(label="Decrytped Image")
                 dec_qr_btn = gr.Button("Decrypt")
         gr.Column()
-    dec_qr_btn.click(decode_qr,[dec_qr_im,pass_out],[d_json])
+    dec_qr_btn.click(decode_qr,[dec_qr_im,pass_out],[dec_qr_im_out,dec_qr_txt,d_json])
     en_btn.click(encrypt,[pass_in,send,rec,am],[enc_out,enc_qr_out,dec_qr_im])
 app.launch()