Update app.py

app.py

@@ -31,10 +31,23 @@ def remove_background_rembg(input_path):
     return img
 
 def remove_background_bria(input_path):
-    print(f"Removing background using bria for image: {input_path}")
+    image = Image.open(input_path)
     pipe = pipeline("image-segmentation", model="briaai/RMBG-1.4", trust_remote_code=True, device=0)
-    pillow_image = pipe(input_path)
-    return pillow_image
+    result = pipe(image)
+    output_image = result[0]['segmentation']
+    output_image = Image.fromarray(output_image)
+    return output_image
+
+# Fungsi untuk menambahkan watermark
+def add_watermark(original_image, watermark_image, position=(0, 0), transparency=128):
+    watermark = watermark_image.convert("RGBA")
+    combined = Image.new("RGBA", original_image.size)
+    
+    mask = watermark.split()[3].point(lambda i: i * transparency / 255.0)
+    combined.paste(original_image, (0, 0))
+    combined.paste(watermark, position, mask)
+    
+    return combined
 
 # Fungsi untuk memproses gambar menggunakan prompt
 def text_to_image(prompt):
@@ -403,6 +416,13 @@ def process_single_image(image_path, output_folder, bg_method, canvas_size_name,
             image_with_no_bg = remove_background_rembg(image_path)  # Placeholder for existing function
         elif bg_method == 'bria':
             image_with_no_bg = remove_background_bria(image_path)  # Placeholder for existing function
+        # Memuat gambar watermark
+        watermark = Image.open(watermark_path).convert("RGBA")
+        
+        # Menambahkan watermark ke gambar asli
+        final_image = add_watermark(original_image, watermark, position=(50, 50), transparency=128)
+        
+        return final_image    
         
         temp_image_path = os.path.join(output_folder, f"temp_{filename}")
         image_with_no_bg.save(temp_image_path, format='PNG')