Update app.py

app.py

@@ -495,18 +495,31 @@ def create_canvas(canvas_size, bg_choice, custom_color):
 
 # Function to add watermark
 def add_watermark(image, watermark_path):
-    watermark = Image.open(watermark_path).convert("RGBA")
-    image = image.convert("RGBA")
-
-    # Center the watermark
-    img_width, img_height = image.size
-    wm_width, wm_height = watermark.size
-    x = (img_width - wm_width) // 2
-    y = (img_height - wm_height) // 2
-
-    image.paste(watermark, (x, y), watermark)
+    if watermark_path:
+        watermark = Image.open(watermark_path).convert("RGBA")
+        width, height = image.size
+        watermark = watermark.resize((width // 4, height // 4), Image.ANTIALIAS)  # Resize watermark
+        w_width, w_height = watermark.size
+        
+        # Position the watermark in the center
+        position = ((width - w_width) // 2, (height - w_height) // 2)
+        image.paste(watermark, position, watermark)
     return image
-    
+
+def process_image(image_path, watermark_path):
+    try:
+        # Menghapus latar belakang
+        image_without_bg = remove_background_bria(image_path)
+        if image_without_bg is None:
+            raise Exception("Background removal failed.")
+
+        # Menambahkan watermark
+        final_image = add_watermark(image_without_bg, watermark_path)
+        return final_image
+    except Exception as e:
+        print(f"Error processing image: {e}")
+        return None
+        
 def save_image(canvas, output_folder, filename, output_format):
     output_ext = 'jpg' if output_format == 'JPG' else 'png'
     output_path = os.path.join(output_folder, f"{os.path.splitext(filename)[0]}.{output_ext}")