| 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') |
