MK-316's picture
Update app.py
65a07f6 verified
import gradio as gr
import pyqrcode
from PIL import Image, ImageDraw, ImageFont
import io
def generate_qr_code(url, title):
# Generate QR code
qr = pyqrcode.create(url)
buffer = io.BytesIO()
# Create the QR code as a PNG
qr.png(buffer, scale=10)
# Move to the beginning of the StringIO buffer
buffer.seek(0)
qr_image = Image.open(buffer)
# Prepare to add title to the image
image_width = qr_image.width
image_height = qr_image.height + 70 # Add 70 pixels space for title
# Create a new image with white background
result_image = Image.new('RGB', (image_width, image_height), 'white')
# Paste the QR code onto this new image
result_image.paste(qr_image, (0, 70))
# Draw the title on the image
draw = ImageDraw.Draw(result_image)
font_path = "dejavu-sans-bold.ttf" # Ensure the font path is correct
font = ImageFont.truetype(font_path, 24) # Use a specific font size
# Get the size of the text using textbbox
text_bbox = draw.textbbox((0, 0), title, font=font)
text_width = text_bbox[2] - text_bbox[0]
text_height = text_bbox[3] - text_bbox[1]
text_x = (image_width - text_width) / 2 # Center the text
draw.text((text_x, 20), title, fill="black", font=font)
# Return the final image
return result_image
# Create the Gradio interface
iface = gr.Interface(
fn=generate_qr_code,
inputs=[
gr.Textbox(label="Enter URL", placeholder="Type or paste URL here..."),
gr.Textbox(label="Enter Title for QR Code", placeholder="Type the title here...")
],
outputs=gr.Image(label="QR Code Image", type="pil", format="png"),
title="QR Code Generator",
description="Enter a URL and a title to generate a QR Code. The title and the QR Code will be displayed in the same image."
)
iface.launch(share=True)