app.py

import gradio as gr
import qrcode
from PIL import Image, ImageDraw
import io


def generate_qr_code(url, logo, qr_color):
    # Create QR code instance
    qr = qrcode.QRCode(
        version=1,
        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)

    # Create the QR code image
    qr_image = qr.make_image(fill_color=qr_color, back_color="#ffffff")  # Set QR code color here

    # Add logo if provided
    if logo and len(logo) > 0:
        logo_image = Image.open(logo[0]).convert("RGBA")
        qr_image.paste(logo_image, (125, 125), logo_image)

    return qr_image


inputs = [
    gr.Textbox(label="URL to Site"),
    gr.inputs.Image(label="Logo (Optional)"),
    gr.ColorPicker(label="QR Code Color"),
]

outputs = gr.outputs.Image(type="pil")

gr.Interface(
    fn=generate_qr_code,
    inputs=inputs,
    outputs=outputs,
    title="QR Code Generator",
    description="Generate QR codes with customizable options",
    theme="huggingface",
).launch(share=True)