Update app.py

app.py

@@ -2,29 +2,35 @@ import gradio as gr
 import requests
 import qrcode
 import io
-from PIL import Image, ImageDraw
+from PIL import Image
 
 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
+# Function to generate a QR code with transparent fill color
 def generate_qr_image(url, image_size):
+    # Determine box size based on the larger dimension of the image
+    max_dim = max(image_size[0], image_size[1])
+    box_size = max_dim // 25
+    border = 1
+    
     qr = qrcode.QRCode(
         version=1,
         error_correction=qrcode.constants.ERROR_CORRECT_L,
-        box_size=10,
-        border=4,
+        box_size=box_size,
+        border=border,
     )
     qr.add_data(url)
     qr.make(fit=True)
 
-    qr_img = qr.make_image(fill_color="black", back_color="white").convert("RGBA")
+    qr_img = qr.make_image(fill_color="black", back_color=None).convert("RGBA")
     qr_img = qr_img.resize(image_size)
 
     return qr_img
 
+
 # Function to query the image from the Hugging Face API
 def query_image(text):
     API_URL = "https://api-inference.huggingface.co/models/goofyai/3d_render_style_xl"
@@ -40,7 +46,7 @@ def generate_image(url, text):
     api_image = Image.open(io.BytesIO(image_bytes))
 
     # Generate QR code image
-    qr_image = generate_qr_image(url, (250, 250))
+    qr_image = generate_qr_image(url, api_image.size)
 
     # Create a blank image with transparency
     final_image = Image.new('RGBA', api_image.size, (255, 255, 255, 0))