Kvikontent commited on
Commit
7324c34
·
1 Parent(s): cf5398e

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +65 -0
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()