|
import gradio as gr |
|
import qrcode as qr |
|
import base64 |
|
import cv2 |
|
import os |
|
from PIL import Image |
|
|
|
def make_qr(txt=None,data=None,im_size=None): |
|
if txt != None and txt != "" and data != None: |
|
f = Image.open(f'{data}') |
|
f.thumbnail((im_size,im_size)) |
|
f.save("tmp.jpg") |
|
imr = open(f'tmp.jpg','rb') |
|
out = f'{txt}+++{base64.b64encode(imr.read())}' |
|
print (f'txt+data {out}') |
|
img1 = qr.make(out,box_size=10,error_correction=qr.constants.ERROR_CORRECT_H) |
|
img1.save("im.png") |
|
return "im.png" |
|
if txt == None or txt == "" and data != None: |
|
f = Image.open(f'{data}') |
|
f.thumbnail((im_size,im_size)) |
|
f.save("tmp1.jpg") |
|
imr = open(f'tmp1.jpg','rb') |
|
out = f'+++{base64.b64encode(imr.read())}' |
|
print (f'data {out}') |
|
img1 = qr.make(out,box_size=10,error_correction=qr.constants.ERROR_CORRECT_H) |
|
img1.save("im1.png") |
|
return "im1.png" |
|
|
|
if txt != None and txt != "" and data == None: |
|
out = f'{txt}' |
|
print (f'txt {out}') |
|
img1 = qr.make(out,box_size=10,error_correction=qr.constants.ERROR_CORRECT_H) |
|
img1.save("im2.png") |
|
return "im2.png" |
|
|
|
def cnt_im_bytes(im,txt_cnt,im_size): |
|
f = Image.open(f'{im}') |
|
f.thumbnail((im_size,im_size)) |
|
f.save("tmp11.jpg") |
|
im_cnt=os.stat('tmp11.jpg').st_size |
|
print(im_cnt) |
|
tot_cnt=im_cnt+int(txt_cnt) |
|
return im_cnt,tot_cnt |
|
|
|
def cnt_bytes(txt,im_cnt): |
|
txt_cnt = (len(txt.encode('utf-8'))) |
|
tot_cnt = txt_cnt + int(im_cnt) |
|
return txt_cnt, tot_cnt |
|
|
|
|
|
def decode(im): |
|
|
|
image = cv2.imread(f'{im}') |
|
qrCodeDetector = cv2.QRCodeDetector() |
|
decodedText, points, _ = qrCodeDetector.detectAndDecode(image) |
|
if points is not None: |
|
text = decodedText |
|
else: |
|
text = "No QR Code Found" |
|
return text |
|
def make_im(tx_str): |
|
out = tx_str.split("+++b",1)[1] |
|
out.replace("'","") |
|
print(out) |
|
decoded_data=base64.b64decode((out)) |
|
|
|
|
|
img_file = open('image.jpeg', 'wb') |
|
img_file.write(decoded_data) |
|
img_file.close() |
|
return ('image.jpeg') |