Update app.py

app.py

@@ -243,14 +243,21 @@ def outpaint(image, width, height, overlap_percentage, num_inference_steps, resi
 
     # Extract original alpha from the input image
     original_alpha = background.split()[3] if background.mode == "RGBA" else Image.new("L", background.size, 255)
-    original_alpha_mask = original_alpha.point(lambda p: 0 if p < 255 else 255)  # 0 where transparent
-
+    combined_mask = ImageChops.logical_or(original_alpha, mask)  # 0 where to paint
     # Combine original and generated mask
-    combined_mask = ImageChops.logical_or(original_alpha_mask, mask)  # 0 where to fill
-
-    # Use the combined_mask in the pipeline
+    # combined_mask = ImageChops.logical_or(original_alpha_mask, mask)  # 0 where to fill
     cnet_image = background.copy()
-    cnet_image.paste(0, (0, 0), combined_mask)  # Overlay black on combined_mask area
+    if cnet_image.mode == "RGBA":
+        # Extract the alpha channel and apply the combined mask
+        alpha = combined_mask.convert("L")
+        cnet_image = cnet_image.convert("RGBA")
+        cnet_image.putalpha(alpha)  # Replace the alpha channel
+    else:
+        # If no alpha, just use the background
+        cnet_image = background
+    # Use the combined_mask in the pipeline
+    #cnet_image = background.copy()
+    # cnet_image.paste(0, (0, 0), combined_mask)  # Overlay black on combined_mask area
 
     final_prompt = f"score_9, score_8_up, score_7_up, {prompt_input} , high quality, 4k"
     print(f"Outpainting using SDXL model: {pipe.config.model_name}")
@@ -271,9 +278,10 @@ def outpaint(image, width, height, overlap_percentage, num_inference_steps, resi
         yield cnet_image, image
     # Invert the combined_mask and paste the generated image back
     filled_mask = combined_mask.point(lambda p: 255 - p)
+    # image = image.convert("RGBA")
+    # cnet_image.paste(image, (0, 0), filled_mask)
     image = image.convert("RGBA")
-    cnet_image.paste(image, (0, 0), filled_mask)
-
+    cnet_image.paste(image, (0, 0), combined_mask)
     yield background, cnet_image
     
 @spaces.GPU(duration=7)