app.py

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


# Function to generate QR code
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


# Streamlit app
st.title('QR Code Generator')
st.write('Enter text or a URL to generate a QR code.')

# User input
user_input = st.text_input("Enter text or URL:")

if user_input:
    # Generate QR code
    img = generate_qr_code(user_input)

    # Display QR code
    st.image(img, caption='Generated QR Code')

    # Option to download the QR code
    buffer = BytesIO()
    img.save(buffer, format="PNG")
    buffer.seek(0)
    st.download_button(
        label="Download QR Code",
        data=buffer,
        file_name="qr_code.png",
        mime="image/png"
    )