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

def generate_qr_code(data):
    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

def main():
    st.title("QR Code Generator")

    data = st.text_input("Enter the data to encode:")
    
    if st.button("Generate QR Code"):
        if data:
            qr_img = generate_qr_code(data)
            # Convert PIL image to bytes
            img_bytes = io.BytesIO()
            qr_img.save(img_bytes, format="PNG")
            st.image(img_bytes, caption="Generated QR Code", use_column_width=True)
        else:
            st.warning("Please enter some data.")

if __name__ == "__main__":
    main()