Update app.py

app.py

@@ -1,72 +1,58 @@
-
 import gradio as gr
 import qrcode
 
-def generate_qrcode(url, color, gradient, gradient_colors, border_radius, logo):
+def generate_qr_code(url, gradient_color1, gradient_color2, logo, qr_color):
+    # Create QR code instance
     qr = qrcode.QRCode(
         version=1,
-        error_correction=qrcode.constants.ERROR_CORRECT_H,
+        error_correction=qrcode.constants.ERROR_CORRECT_L,
         box_size=10,
         border=4,
     )
-
+    
+    # Add URL to the QR code
     qr.add_data(url)
     qr.make(fit=True)
 
-    qr_image = qr.make_image(fill_color=color, back_color="white")
-
-    if gradient:
+    # Create the QR code image
+    qr_image = qr.make_image(fill_color=qr_color, back_color="#ffffff")  # Set QR code color here
+    
+    # Add gradient if colors are provided
+    if gradient_color1 and gradient_color2:
         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)
+        qr_image = qr_image.resize((500, 500))
+
+        gradient = Image.new("RGB", qr_image.size)
+        grad = ImageDraw.Gradient(grad.size, "LR", gradient_color1, gradient_color2)
+        qr_image = Image.blend(qr_image, gradient, 0.5)
+    
+    # Add logo if provided
+    if logo:
+        logo_image = Image.open(logo).convert("RGBA")
+        qr_image.paste(logo_image, (125, 125), logo_image)
 
-    if border_radius > 0:
-        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)
+    # Save the QR code to a byte stream
+    qr_byte_stream = io.BytesIO()
+    qr_image.save(qr_byte_stream, format="PNG")
+    qr_byte_stream.seek(0)
 
-    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)
+    return qr_byte_stream.read()
 
-    qr_image.save("qr_code.png")
+inputs = [
+    gr.inputs.Textbox(label="URL to Site"),
+    gr.inputs.ColorPicker(label="Gradient Color 1 (Optional)"),
+    gr.inputs.ColorPicker(label="Gradient Color 2 (Optional)"),
+    gr.inputs.Image(label="Logo (Optional)"),
+    gr.inputs.ColorPicker(label="QR Code Color"),
+]
 
-    return "qr_code.png"
+outputs = gr.outputs.Image(type="bytes")
 
-iface = gr.Interface(
-    fn=generate_qrcode,
-    inputs=[
-        gr.inputs.Textbox(label="URL", default="example.com"),
-        gr.ColorPicker(label="Color", default="#000000"),
-        gr.Checkbox(label="Gradient", default=False),
-        gr.ColorPicker(label="Gradient Colors", default=["#FFFFFF", "#000000"], nargs=2),
-        gr.Number(label="Border Radius", default=0),
-        gr.inputs.File(label="Logo")
-    ],
-    outputs="image",
+gr.Interface(
+    fn=generate_qr_code,
+    inputs=inputs,
+    outputs=outputs,
     title="QR Code Generator",
-    description="Enter a URL and customize the QR code appearance",
-    examples=[
-        ["instagram.com", "#FF0000", True, ["#FF00FF", "#00FFFF"], 10, None],
-        ["twitter.com", "#00FF00", False, ["#FFFFFF", "#000000"], 0, "logo.png"],
-        ["huggingface.co", "#0000FF", True, ["#FF00FF", "#00FFFF"], 15, "logo.png"]
-    ]
-)
-
-iface.launch()
+    description="Generate QR codes with customizable options",
+    theme="huggingface",
+).launch()