Update app.py

app.py

@@ -494,22 +494,20 @@ 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):
-    # Open the images
-    base = canvas.convert("RGBA")
-    watermark = watermark.convert("RGBA")
+    watermark = Image.open(watermark_path).convert("RGBA")
+    
+    # Get dimensions of both the canvas and watermark
+    canvas_width, canvas_height = canvas.size
+    watermark_width, watermark_height = watermark.size
 
     # Calculate the position to center the watermark
-    base_width, base_height = base.size
-    watermark_width, watermark_height = watermark.size
-    x = (base_width - watermark_width) // 2
-    y = (base_height - watermark_height) // 2
+    x = (canvas_width - watermark_width) // 2
+    y = (canvas_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)
+    # Paste the watermark on the canvas
+    canvas.paste(watermark, (x, y), watermark)
 
-    return combined.convert("RGB")  # Convert back to RGB if needed
+    return canvas
     
 def save_image(canvas, output_folder, filename, output_format):
     output_ext = 'jpg' if output_format == 'JPG' else 'png'