Update app.py

app.py

@@ -2,14 +2,15 @@ import gradio as gr
 import requests
 import qrcode
 import io
-from PIL import Image
+from PIL import Image, ImageDraw
+
 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 from the input URL
-def generate_qr_code(url):
+# Function to generate a QR code with transparency
+def generate_qr_image(url):
     qr = qrcode.QRCode(
         version=1,
         error_correction=qrcode.constants.ERROR_CORRECT_L,
@@ -20,7 +21,20 @@ def generate_qr_code(url):
     qr.make(fit=True)
 
     qr_img = qr.make_image(fill_color="black", back_color="white")
-    return qr_img
+    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_rgba.putdata(new_qr_data)
+    
+    return qr_img_rgba
 
 # Function to query the image from the Hugging Face API
 def query_image(text):
@@ -32,17 +46,21 @@ def query_image(text):
 
 # Gradio app function
 def generate_image(url, text):
-    # Generate QR code
-    qr_img = generate_qr_code(url)
-    
     # Generate image from Hugging Face API
     image_bytes = query_image(text)
     api_image = Image.open(io.BytesIO(image_bytes))
     
-    # Create a blank image to combine the API image and QR code
-    final_image = Image.new('RGB', (600, 600))
-    final_image.paste(api_image, (0, 0))
-    final_image.paste(qr_img, (400, 400))
+    # 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)))
+    
+    # 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)
     
     return final_image