import streamlit as st
import qrcode
from io import BytesIO
from PIL import Image

def generate_qr_code(data: str):
    qr = qrcode.QRCode(
        version=1,
        error_correction=qrcode.constants.ERROR_CORRECT_L,
        box_size=10,
        border=4,
    )
    qr.add_data(data)
    qr.make(fit=True)
    img = qr.make_image(fill_color="black", back_color="white")
    return img

# Streamlit UI
st.title("QR Code Generator")

data = st.text_input("Enter the text or URL for the QR code:")

if data:
    img = generate_qr_code(data)
    buffer = BytesIO()
    img.save(buffer, format="PNG")
    st.image(buffer.getvalue(), caption="Generated QR Code")

    # Option to download the QR code
    btn = st.download_button(
        label="Download QR Code",
        data=buffer,
        file_name="qr_code.png",
        mime="image/png",
    )