Spaces:
Runtime error
Runtime error
Commit
·
7324c34
1
Parent(s):
cf5398e
Create app.py
Browse files
app.py
ADDED
@@ -0,0 +1,65 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import qrcode
|
2 |
+
import gradio as gr
|
3 |
+
|
4 |
+
def generate_qr_code(url, color, use_gradient, gradient_color1, gradient_color2, border_radius, logo):
|
5 |
+
qr = qrcode.QRCode(
|
6 |
+
version=1,
|
7 |
+
error_correction=qrcode.constants.ERROR_CORRECT_L,
|
8 |
+
box_size=10,
|
9 |
+
border=4
|
10 |
+
)
|
11 |
+
|
12 |
+
qr.add_data(url)
|
13 |
+
qr.make(fit=True)
|
14 |
+
|
15 |
+
qr_img = qr.make_image(fill_color=color)
|
16 |
+
|
17 |
+
if use_gradient:
|
18 |
+
qr_img = qr_img.convert("RGB")
|
19 |
+
qr_pixels = qr_img.load()
|
20 |
+
width, height = qr_img.size
|
21 |
+
|
22 |
+
for y in range(height):
|
23 |
+
for x in range(width):
|
24 |
+
pixel = qr_pixels[x, y]
|
25 |
+
gradient_ratio = x / width
|
26 |
+
new_r = int((gradient_color2[0] - gradient_color1[0]) * gradient_ratio + gradient_color1[0])
|
27 |
+
new_g = int((gradient_color2[1] - gradient_color1[1]) * gradient_ratio + gradient_color1[1])
|
28 |
+
new_b = int((gradient_color2[2] - gradient_color1[2]) * gradient_ratio + gradient_color1[2])
|
29 |
+
qr_pixels[x, y] = (new_r, new_g, new_b)
|
30 |
+
|
31 |
+
if border_radius > 0:
|
32 |
+
qr_img = qr_img.resize((qr_img.size[0] + (border_radius * 2), qr_img.size[1] + (border_radius * 2)))
|
33 |
+
qr_img = qr_img.convert("RGBA")
|
34 |
+
border_img = qrcode.image.svg.SvgPathImage(qr_img.size)
|
35 |
+
border_img.path = border_img.rounded_rect((0, 0, *qr_img.size), border_radius)
|
36 |
+
border_img.fill = qr_img
|
37 |
+
qr_img = border_img
|
38 |
+
|
39 |
+
if logo:
|
40 |
+
qr_with_logo = qr_img.copy()
|
41 |
+
logo_img = qrcode.image.open(logo)
|
42 |
+
logo_width, logo_height = logo_img.size
|
43 |
+
pos = ((qr_img.size[0] - logo_width) // 2, (qr_img.size[1] - logo_height) // 2)
|
44 |
+
qr_with_logo.paste(logo_img, pos)
|
45 |
+
qr_img = qr_with_logo
|
46 |
+
|
47 |
+
return qr_img
|
48 |
+
|
49 |
+
inputs = [
|
50 |
+
gr.inputs.Textbox(label="URL"),
|
51 |
+
gr.inputs.Color(label="Color"),
|
52 |
+
gr.inputs.Checkbox(label="Use Gradient"),
|
53 |
+
gr.inputs.Color(label="Gradient Color 1"),
|
54 |
+
gr.inputs.Color(label="Gradient Color 2"),
|
55 |
+
gr.inputs.Number(label="Border Radius", min_value=0, max_value=50),
|
56 |
+
gr.inputs.Image(label="Logo (optional)"),
|
57 |
+
]
|
58 |
+
|
59 |
+
output = gr.outputs.Image(label="QR Code")
|
60 |
+
|
61 |
+
def generate_qr_code_app(url, color, use_gradient, gradient_color1, gradient_color2, border_radius, logo):
|
62 |
+
qr_code = generate_qr_code(url, color, use_gradient, gradient_color1, gradient_color2, border_radius, logo)
|
63 |
+
return qr_code
|
64 |
+
|
65 |
+
gr.Interface(fn=generate_qr_code_app, inputs=inputs, outputs=output).launch()
|