Update app.py

app.py

@@ -41,7 +41,7 @@ pipe = StableDiffusionControlNetImg2ImgPipeline.from_pretrained(
 # pipe.enable_xformers_memory_efficient_attention()
 
 
-def resize_for_condition_image(input_image: Image.Image, resolution: int):
+def resize_for_condition_image(input_image: Image.Image, resolution: int, canvas_width: int = None, canvas_height: int = None):
     input_image = input_image.convert("RGB")
     W, H = input_image.size
     k = float(resolution) / min(H, W)
@@ -49,8 +49,22 @@ def resize_for_condition_image(input_image: Image.Image, resolution: int):
     W *= k
     H = int(round(H / 64.0)) * 64
     W = int(round(W / 64.0)) * 64
-    img = input_image.resize((W, H), resample=Image.LANCZOS)
-    return img
+    input_image = input_image.resize((W, H), resample=Image.LANCZOS)
+    
+    # Create a blank canvas with the specified dimensions
+    if canvas_width is None:
+        canvas_width = W
+    if canvas_height is None:
+        canvas_height = H
+
+    canvas = Image.new("RGB", (canvas_width, canvas_height), "white")
+
+    # Paste the resized QR code on the right half of the canvas
+    qr_x = canvas_width // 2  # Start pasting in the middle of the canvas
+    qr_y = (canvas_height - H) // 2  # Center the QR code vertically
+    canvas.paste(input_image, (qr_x, qr_y))
+    return canvas
+
 
 
 SAMPLER_MAP = {
@@ -100,10 +114,10 @@ def inference(
         qr.make(fit=True)
 
         qrcode_image = qr.make_image(fill_color="black", back_color="white")
-        qrcode_image = resize_for_condition_image(qrcode_image, 768)
+        qrcode_image = resize_for_condition_image(qrcode_image, 768, width, height)
     else:
         print("Using QR Code Image")
-        qrcode_image = resize_for_condition_image(qrcode_image, 768)
+        qrcode_image = resize_for_condition_image(qrcode_image, 768, width, height)
 
     # hack due to gradio examples
     init_image = qrcode_image