ask_my_thesis / make_QR.py
jordyvl's picture
QR code and changed reqs
27a13c3
raw
history blame contribute delete
No virus
1.79 kB
from PIL import Image
import qrcode
for image, data in zip(["DUDER"], ["https://huggingface.co/spaces/jordyvl/ask_my_thesis"]):
# ["dataset", "competition"],
# ["https://huggingface.co/datasets/jordyvl/DUDE_loader", "https://rrc.cvc.uab.es/?ch=23"]
# ["dataset1", "dataset2", "code"],
# [
# "https://huggingface.co/datasets/bdpc/rvl_cdip_mp",
# "https://huggingface.co/datasets/bdpc/rvl_cdip_n_mp",
# "https://huggingface.co/bdpc/src",
# ],
# ):
# Create a QR code object
qr = qrcode.QRCode(box_size=10, border=4, error_correction=qrcode.constants.ERROR_CORRECT_H) # version=1,
# Define the data to be encoded in the QR code
# Add the data to the QR code object
qr.add_data(data)
# Make the QR code
qr.make(fit=True)
# Create an image from the QR code
img = qr.make_image(back_color=(255, 254, 255), fill_color=(0, 0, 1))
# Open the logo or image file
# remove transparency
logo = Image.open(f"{image}.png").convert("RGBA")
new_image = Image.new("RGBA", logo.size, "WHITE") # Create a white rgba background
new_image.paste(logo, (0, 0), logo) # Paste the image on the background. Go to the links given below for details.
new_image.convert("RGB").save(f"{image}.jpg", "JPEG")
logo = Image.open(f"{image}.jpg").convert("RGB")
# Resize the logo or image if needed
logo = logo.resize((150, 150))
# Position the logo or image in the center of the QR code
img_w, img_h = img.size
logo_w, logo_h = logo.size
pos = ((img_w - logo_w) // 2, (img_h - logo_h) // 2)
# Paste the logo or image onto the QR code
img.paste(logo, pos)
# Save the QR code image with logo or image
img.save(f"qr_code_{image}.png")