Create app.py

app.py

@@ -0,0 +1,251 @@
+import gradio as gr
+from cryptography.fernet import Fernet
+from cryptography.hazmat.primitives import hashes
+from cryptography.hazmat.primitives.kdf.pbkdf2 import PBKDF2HMAC
+import os
+import base64
+import hashlib
+from pathlib import Path
+import qr
+import stegan
+import json
+
+def trans_block(sender,recipient,amount):
+    transaction = [{
+        'sender': sender,
+        'recipient': recipient,
+        'amount': amount
+    }]
+    json_object = json.dumps(transaction, indent=4)
+    print (json_object)
+    return json_object
+
+def create_key(passw):
+    key = passw.encode()
+    h = hashlib.new('sha256')
+    h.update(key)
+    key = h.hexdigest()
+    key = key.encode()
+    #salt = os.urandom(16)
+    salt = key
+    kdf = PBKDF2HMAC(
+        algorithm=hashes.SHA256(),
+        length=32,
+        salt=salt,
+        iterations=480000,
+    )
+    key = base64.urlsafe_b64encode(kdf.derive(key))
+    return key
+
+
+def encrypt(passw,mes=None,img=None,doc=None,trans_s=None,trans_r=None,trans_a=None):
+    #key = Fernet.generate_key()
+    key = create_key(passw)
+    fernet = Fernet(key)
+    enc_mes=""
+    enc_file=None
+    if mes != None and mes != "":
+        bytes_m = mes.encode()
+        enc_mes = fernet.encrypt(bytes_m)
+        enc_mes = f'{enc_mes}+aaa+'
+    if img != None:
+        with open(f'{img}', "rb") as image_file:
+            bytes_i = base64.b64encode(image_file.read()) 
+        if mes != None and mes != "":
+            im_bytes = fernet.encrypt(bytes_i)
+            enc_mes = f'{enc_mes}{im_bytes}+bbb+'
+        else:
+            im_bytes = fernet.encrypt(bytes_i)
+            enc_mes = f'{im_bytes}+bbb+'
+    if trans_s != None and trans_r != None and trans_a != None and trans_s != "" and trans_r != "" and trans_a != "":
+        trans = trans_block(trans_s,trans_r,trans_a)
+        bytes_j = trans.encode()
+        if mes != None or mes != "" or img != None or img !="":
+            json_bytes = fernet.encrypt(bytes_j)
+            enc_mes = f'{enc_mes}{json_bytes}+ccc+'
+        else:    
+            json_bytes = fernet.encrypt(bytes_j)
+            enc_mes = f'{json_bytes}+ccc+'
+
+    #enc_qr = qr.make_qr(txt=(enc_mes.strip('"')))
+    m = hashlib.sha256()
+    bytes_enc_mes = enc_mes.encode()
+    m.update(bytes_enc_mes)
+    out = m.hexdigest()
+    out = out.encode()
+    print(out)
+    qr_link = out
+    enc_qr = stegan.conv_im(qr_link=qr_link,data=enc_mes)
+    
+    if doc != None:
+        #print(dir(doc))
+        #print(doc.name)
+        doc_name = doc.name
+        doc_name = doc_name.split("/",4)[4]
+        with open(doc.name, "rb") as file:
+            # read all file data
+            file_data = file.read()
+        enc_doc = fernet.encrypt(file_data)
+        og_name = doc.name
+        og_end = og_name.split(".",1)[1]
+        og_front=og_name.split(".",1)[0]
+        
+        
+        enc_file=f'{doc.name}.ocrpt'
+        with open(enc_file, "wb") as file:
+            file.write(enc_doc)
+    #current_url.url_return("www.google.com")
+    return enc_mes,enc_file,enc_qr,enc_qr
+    
+def decrypt(passw,enc_in=None):
+    key = create_key(passw)
+    fernet = Fernet(key)
+    dec_im = None
+    mes_dec= None
+    enc_in=enc_in.strip('"')
+    print (f'enc_in :::: {enc_in}')
+    
+    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_im,mes_dec,dec_json)
+
+def decode_doc(passw,doc=None):
+    key = create_key(passw)
+    fernet = Fernet(key)
+    doc_name = doc.name
+    doc_name = doc_name.split("/",4)[4]
+    #bytes_d = doc.encode()
+    
+    #doc = Path(doc)
+    with open(doc.name, "rb") as file:
+        # read all file data
+        file_data = file.read()
+    dec_doc = fernet.decrypt(file_data)
+    og_name = doc.name
+    og_end = og_name.split(".",1)[1]
+    og_front=og_name.split(".",1)[0]
+    
+    dec_file = doc.name.strip(".ocrpt")
+    #enc_file=f'{doc.name}.ocrpt'
+    with open(dec_file, "wb") as file:
+        file.write(dec_doc)    
+    return dec_file
+
+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,mes_dec,dec_json = decrypt(passw, enc_in=decode_qr)
+    
+    return(dec_im,mes_dec,dec_json)
+    
+    
+with gr.Blocks() as app:
+    with gr.Tab("Encrypt"):
+        pass_in=gr.Textbox(label="Set Password")
+        with gr.Tab("String"):
+            with gr.Row():
+                with gr.Column():
+                    send=gr.Textbox(label="Sender")
+                    rec=gr.Textbox(label="Recipient")
+                    am=gr.Textbox(label="Amount")    
+                with gr.Column():
+                    mes = gr.Textbox(label = "Message")
+                    with gr.Row():
+                        im = gr.Image(type="filepath")
+        with gr.Tab("File"):
+            doc=gr.File()
+            enc_doc_out=gr.File()
+        en_btn = gr.Button("Encrypt")
+        enc_out = gr.Textbox(label="Encrypted Bytes")
+        enc_qr_out = gr.Image(label = "Encrypted QR")
+
+    with gr.Tab("Decrypt"):
+        pass_out = gr.Textbox(label="Enter Password")
+        with gr.Tab("String"):
+            enc_in = gr.Textbox(label="Encrypted Bytes")
+            d_btn = gr.Button("Decrypt")
+            d_txt = gr.Textbox(label="Decrypted Message")
+            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(type="filepath")
+            with gr.Row():
+                dec_qr_txt = gr.Textbox(label="Decrypted Message")
+                d_json = gr.Textbox(label="Decrypted JSON")
+
+            dec_qr_im_out =gr.Image(label="Decrytped Image")
+            dec_qr_btn = gr.Button("Decrypt")
+
+    dec_qr_btn.click(decode_qr,[dec_qr_im,pass_out],[dec_qr_im_out,dec_qr_txt,d_json])
+    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,dec_qr_im])
+    d_btn.click(decrypt,[pass_out,enc_in],[d_im,d_txt,d_json])
+app.launch()