Spaces:
Sleeping
Sleeping
import gradio as gr | |
import zxingcpp | |
import numpy as np | |
from PIL import Image | |
import cv2 | |
def decode(im,col,tol): | |
img = cv2.imread(f'{im}') | |
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY) | |
blur = cv2.medianBlur(gray, 5) | |
sharpen_kernel = np.array([[-1,-1,-1], [-1,9,-1], [-1,-1,-1]]) | |
sharpen = cv2.filter2D(blur, -1, sharpen_kernel) | |
try: | |
results = zxingcpp.read_barcodes(sharpen) | |
return results[0].text,sharpen | |
except Exception: | |
#qr_img=Image.open(im).convert("RGBA") | |
qr_img=Image.open(im) | |
datas = qr_img.getdata() | |
newData = [] | |
h1 = col.strip("#") | |
rgb_tup = tuple(int(h1[i:i+2], 16) for i in (0, 2, 4)) | |
tol=100 | |
for item in datas: | |
if item[0] in range(rgb_tup[0]-tol,rgb_tup[0]+tol) and item[1] in range(rgb_tup[1]-tol,rgb_tup[1]+tol) and item[2] in range(rgb_tup[2]-tol,rgb_tup[2]+tol): | |
newData.append((0,0,0)) | |
else: | |
newData.append((255,255,255)) | |
qr_img.putdata(newData) | |
qr_img.save('conv_im.png') | |
img = cv2.imread('conv_im.png') | |
try: | |
results = zxingcpp.read_barcodes(img) | |
print (results[0].text) | |
return results[0].text,'conv_im.png' | |
except Exception: | |
return "QR Code not Detected \nTry choosing the QR code color using the eyedropper in the Color Picker",'conv_im.png' | |
with gr.Blocks() as app: | |
with gr.Row(): | |
with gr.Column(): | |
in_im = gr.Image(label="QR Image to Decode",type='filepath') | |
out_im = gr.Image(type='filepath') | |
with gr.Column(): | |
with gr.Row(): | |
choose_color = gr.ColorPicker(label="Choose QR color (eyedropper)") | |
color_tol = gr.Slider(1,255, step=1,value=50,label="Color Detect Tolerance") | |
dec_btn = gr.Button("Decode QR") | |
text_out = gr.Textbox(label="Decoded Text") | |
dec_btn.click(decode,[in_im,choose_color,color_tol],[text_out,out_im]) | |
app.queue(concurrency_count=10).launch() |