Update app.py

app.py

@@ -1,77 +1,69 @@
-import qrcode
 import gradio as gr
+import qrcode
 
-
-def generate_qr_code(url, color, use_gradient, gradient_color1, gradient_color2, border_radius, logo):
+def generate_qrcode(url, color, gradient, gradient_colors, border_radius, logo):
+    # Create QR code instance
     qr = qrcode.QRCode(
         version=1,
-        error_correction=qrcode.constants.ERROR_CORRECT_L,
+        error_correction=qrcode.constants.ERROR_CORRECT_H,
         box_size=10,
-        border=4
+        border=4,
     )
 
+    # Add data to QR code
     qr.add_data(url)
     qr.make(fit=True)
 
-    qr_img = qr.make_image(fill_color=color)
-
-    if use_gradient:
-        qr_img = qr_img.convert("RGB")
-        qr_pixels = qr_img.load()
-        width, height = qr_img.size
-
-        for y in range(height):
-            for x in range(width):
-                pixel = qr_pixels[x, y]
-                gradient_ratio = x / width
-                new_r = int((gradient_color2[0] - gradient_color1[0]) * gradient_ratio + gradient_color1[0])
-                new_g = int((gradient_color2[1] - gradient_color1[1]) * gradient_ratio + gradient_color1[1])
-                new_b = int((gradient_color2[2] - gradient_color1[2]) * gradient_ratio + gradient_color1[2])
-                qr_pixels[x, y] = (new_r, new_g, new_b)
-
+    # Create image from QR code
+    qr_image = qr.make_image(fill_color=color, back_color="white")
+
+    # Apply gradient to QR code if enabled
+    if gradient:
+        qr_image = qr_image.convert("RGB")
+        width, height = qr_image.size
+        gradient_color1, gradient_color2 = gradient_colors
+        for i in range(height):
+            shift = int((i / height) * 255)
+            fill_color = tuple(int(gradient_color1[c] + shift * (gradient_color2[c] - gradient_color1[c]) / 255) for c in range(3))
+            for j in range(width):
+                pixel = qr_image.getpixel((j, i))
+                if pixel == (0, 0, 0):
+                    qr_image.putpixel((j, i), fill_color)
+
+    # Apply border radius to QR code if specified
     if border_radius > 0:
-        qr_img = qr_img.resize((qr_img.size[0] + (border_radius * 2), qr_img.size[1] + (border_radius * 2)))
-        qr_img = qr_img.convert("RGBA")
-        border_img = qrcode.image.svg.SvgPathImage(qr_img.size)
-        border_img.path = border_img.rounded_rect((0, 0, *qr_img.size), border_radius)
-        border_img.fill = qr_img
-        qr_img = border_img
-
-    if logo:
-        qr_with_logo = qr_img.copy()
-        logo_img = qrcode.image.open(logo)
-        logo_width, logo_height = logo_img.size
-        pos = ((qr_img.size[0] - logo_width) // 2, (qr_img.size[1] - logo_height) // 2)
-        qr_with_logo.paste(logo_img, pos)
-        qr_img = qr_with_logo
-
-    return qr_img
-
-
-def generate_qr_code_app(url: str, color: str, use_gradient: bool, gradient_color1: str, gradient_color2: str,
-                         border_radius: int, logo: gr.inputs.Image):
-    qr_code = generate_qr_code(url, color, use_gradient, gradient_color1, gradient_color2, border_radius,
-                               logo.item.filename if logo is not None else None)
-    return qr_code
-
-
-inputs = [
-    gr.inputs.Textbox(label="URL"),
-    gr.ColorPicker(label="Color"),
-    gr.inputs.Checkbox(label="Use Gradient"),
-    gr.ColorPicker(label="Gradient Color 1"),
-    gr.ColorPicker(label="Gradient Color 2"),
-    gr.inputs.Number(label="Border Radius", default=0),
-    gr.inputs.Image(label="Logo (optional)")
-]
-
-output = gr.outputs.Image(label="QR Code")
-
-examples = [
-    ["https://instagram.com", "#E1306C", False, "#E1306C", "#4458A7", 0, None],
-    ["https://twitter.com", "#1DA1F2", False, "#1DA1F2", "#124964", 5, None],
-    ["https://huggingface.co", "#EE5A55", False, "#EE5A55", "#FBB040", 10, None]
-]
-
-iface = gr.Interface(fn=generate_qr_code_app, inputs=inputs, outputs=output, examples=examples)
-iface.launch()
+        qr_image = qr_image.resize((qr_image.size[0] + border_radius*2, qr_image.size[1] + border_radius*2))
+        qr_image = qr_image.convert("RGBA")
+        width, height = qr_image.size
+        border_radius = max(1, min(border_radius, int(width/2), int(height/2)))
+        mask = Image.new('L', (width, height), 0)
+        draw = ImageDraw.Draw(mask)
+        draw.ellipse((0, 0, 2*border_radius, 2*border_radius), fill=255)
+        draw.ellipse((width-2*border_radius, 0, width, 2*border_radius), fill=255)
+        draw.ellipse((0, height-2*border_radius, 2*border_radius, height), fill=255)
+        draw.ellipse((width-2*border_radius, height-2*border_radius, width, height), fill=255)
+        alpha = Image.new('L', (width, height), int(255 * 0.5))
+        qr_image.paste(alpha, (0, 0), mask)
+
+    # Add logo to center of QR code if specified
+    if logo is not None:
+        logo_image = Image.open(logo).convert("RGBA")
+        qr_image.paste(logo_image, (int((qr_image.size[0] - logo_image.size[0]) / 2), int((qr_image.size[1] - logo_image.size[1]) / 2)), logo_image)
+
+    # Save QR code as PNG
+    qr_image.save("qr_code.png")
+
+    return "qr_code.png"
+
+iface = gr.Interface(
+    fn=generate_qrcode,
+    inputs=["text", "text", "checkbox", "text[]", "number", gr.inputs.File()],
+    outputs="image",
+    title="QR Code Generator",
+    description="Enter a URL and customize the QR code appearance",
+    examples=[["instagram.com", "#FF0000", True, ["#FF0000", "#0000FF"], 10, None],
+              ["twitter.com", "#00FF00", False, [], 0, "logo.png"],
+              ["huggingface.co", "#0000FF", True, ["#FF0000", "#00FF00"], 15, "logo.png"]]
+)
+
+iface.launch()