Update app.py

app.py

@@ -494,13 +494,22 @@ def create_canvas(canvas_size, bg_choice, custom_color):
         return Image.new("RGBA", canvas_size, (0, 0, 0, 0))
 
 def apply_watermark(canvas, watermark_path):
-    watermark = Image.open(watermark_path).convert("RGBA")
-    
+    # Open the images
+    base = canvas.convert("RGBA")
+    watermark = watermark.convert("RGBA")
+
     # Calculate the position to center the watermark
-    x = (canvas_width - watermark_width) // 2
-    y = (canvas_height - watermark_height) // 2
+    base_width, base_height = base.size
+    watermark_width, watermark_height = watermark.size
+    x = (base_width - watermark_width) // 2
+    y = (base_height - watermark_height) // 2
+
+    # Create a new image with an alpha layer for transparency
+    combined = Image.new("RGBA", base.size)
+    combined.paste(base, (0, 0))
+    combined.paste(watermark, (x, y), watermark)
 
-    canvas.paste(watermark, (x, y), watermark)
+    return combined.convert("RGB")  # Convert back to RGB if needed
     
 def save_image(canvas, output_folder, filename, output_format):
     output_ext = 'jpg' if output_format == 'JPG' else 'png'