import gradio as gr
import qrcode
from PIL import Image

def generate_qr(link):
    if link.strip() == "":
        return None

    # Create QR code object
    qr = qrcode.QRCode(
        version=1,
        box_size=10,
        border=5
    )
    qr.add_data(link)
    qr.make(fit=True)

    # Make the image (this is already a PIL image)
    img = qr.make_image(fill_color="black", back_color="white")

    # Ensure it's in RGB mode for Gradio
    img = img.convert("RGB")

    return img

with gr.Blocks(title="Simple QR Code Maker") as demo:
    gr.Markdown("### 🔳 Simple QR Code Maker\nEnter any link to generate a QR code instantly!")

    link_input = gr.Textbox(label="Enter URL", placeholder="https://example.com")
    output_image = gr.Image(label="Generated QR Code", type="pil")

    generate_btn = gr.Button("Generate QR Code")
    generate_btn.click(fn=generate_qr, inputs=link_input, outputs=output_image)

demo.launch(theme=gr.themes.Soft())