Update app.py

app.py

@@ -9,8 +9,8 @@ import os
 # Get the Hugging Face API token from the environment variable
 hf_token = os.environ.get("hf_token")
 
-# Function to generate a QR code with transparency
-def generate_qr_image(url):
+# Function to generate a QR code
+def generate_qr_image(url, image_size):
     qr = qrcode.QRCode(
         version=1,
         error_correction=qrcode.constants.ERROR_CORRECT_L,
@@ -20,21 +20,12 @@ def generate_qr_image(url):
     qr.add_data(url)
     qr.make(fit=True)
 
-    qr_img = qr.make_image(fill_color="black", back_color="white")
-    qr_img_rgba = qr_img.convert("RGBA")
-
-    # Add transparency to the QR code
-    qr_data = qr_img_rgba.getdata()
-    new_qr_data = []
-    for item in qr_data:
-        if item[0] == 0 and item[1] == 0 and item[2] == 0:
-            new_qr_data.append((0, 0, 0, 0))
-        else:
-            new_qr_data.append(item)
+    qr_img = qr.make_image(fill_color="black", back_color="white").convert("RGBA")
     
-    qr_img_rgba.putdata(new_qr_data)
+    # Resize the QR code to cover the entire image area
+    qr_img = qr_img.resize((image_size[0], image_size[1]))
     
-    return qr_img_rgba
+    return qr_img
 
 # Function to query the image from the Hugging Face API
 def query_image(text):
@@ -50,17 +41,11 @@ def generate_image(url, text):
     image_bytes = query_image(text)
     api_image = Image.open(io.BytesIO(image_bytes))
     
-    # Generate QR code image with transparency
-    qr_image = generate_qr_image(url)
-    
-    # Resize the QR code to fit into the generated image
-    qr_width, qr_height = qr_image.size
-    api_image_width, api_image_height = api_image.size
-    qr_image = qr_image.resize((int(qr_width * 0.5), int(qr_height * 0.5)))
+    # Generate QR code image
+    qr_image = generate_qr_image(url, api_image.size)
     
-    # Blend the two images together
-    final_image = api_image.copy()
-    final_image.paste(qr_image, (api_image_width - qr_image.width, api_image_height - qr_image.height), qr_image)
+    # Blend the API image and QR code
+    final_image = Image.alpha_composite(api_image, qr_image)
     
     return final_image