Omnibus's picture
Update app.py
cd41808
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,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 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()
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)
return enc_mes,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.Row():
gr.Column()
with gr.Column():
with gr.Tab("Encrypt"):
pass_in=gr.Textbox(label="Set Password")
with gr.Row():
with gr.Column():
send=gr.Textbox(label="Sender")
rec=gr.Textbox(label="Recipient")
am=gr.Textbox(label="Amount")
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")
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],[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()