import gradio as gr
import qrcode
from PIL import Image
import numpy as np

def generate_qr_code(url):
    # Check if the URL is not empty
    if not url:
        return "Please enter a URL."

    # Create QR code instance
    qr = qrcode.QRCode(
        version=1,
        error_correction=qrcode.constants.ERROR_CORRECT_L,
        box_size=10,
        border=4,
    )

    # Add data to the QR code
    qr.add_data(url)
    qr.make(fit=True)

    # Create an image from the QR code data
    img = qr.make_image(fill_color="black", back_color="white")

    # Convert the image to a numpy array
    img_array = np.array(img)

    # Convert the numpy array to a PIL Image
    img_pil = Image.fromarray(img_array)

    return img_pil

iface = gr.Interface(fn=generate_qr_code,
                     inputs="text",
                     outputs="image")

iface.launch(debug = True)