from PIL import Image

def black_to_transparent_exact(input_path, output_path):
    """
    將純黑色背景替換為透明色（精確匹配RGB=0,0,0）
    
    參數(shù):
        input_path: 輸入圖片路徑（支持PNG/JPG等）
        output_path: 輸出透明背景圖片（必須用.png格式）
    """
    img = Image.open(input_path).convert("RGBA")  # 轉(zhuǎn)為RGBA模式
    datas = img.getdata()  # 獲取所有像素?cái)?shù)據(jù)
    
    new_data = []
    for item in datas:
        # 如果像素是純黑色（RGB=0,0,0），則設(shè)為透明
        if item[0] == 0 and item[1] == 0 and item[2] == 0:
            new_data.append((0, 0, 0, 0))  # RGBA中的A=0表示透明
        else:
            new_data.append(item)  # 保留原像素
    
    img.putdata(new_data)  # 更新像素?cái)?shù)據(jù)
    img.save(output_path, "PNG")  # 保存為PNG（支持透明）
    print(f"純黑背景已替換為透明，保存至：{output_path}")

# 示例調(diào)用
black_to_transparent_exact("input_black_bg.jpg", "output_transparent.png")

from PIL import Image

def black_to_transparent_fuzzy(input_path, output_path, threshold=30):
    """
    將接近黑色的背景替換為透明色（支持抗鋸齒邊緣）
    
    參數(shù):
        input_path: 輸入圖片路徑
        output_path: 輸出透明背景圖片（必須用.png格式）
        threshold: 顏色閾值（0-255），值越小越嚴(yán)格（默認(rèn)30）
    """
    img = Image.open(input_path).convert("RGBA")
    datas = img.getdata()
    
    new_data = []
    for item in datas:
        # 計(jì)算像素的亮度（灰度值）
        brightness = sum(item[:3]) / 3  # RGB平均值
        
        # 如果亮度低于閾值，則設(shè)為透明
        if brightness < threshold:
            new_data.append((0, 0, 0, 0))
        else:
            new_data.append(item)
    
    img.putdata(new_data)
    img.save(output_path, "PNG")
    print(f"模糊黑背景已替換為透明（閾值={threshold}），保存至：{output_path}")

# 示例調(diào)用
black_to_transparent_fuzzy("input_fuzzy_bg.png", "output_transparent_fuzzy.png", threshold=20)

from PIL import Image

def black_to_semi_transparent(input_path, output_path, threshold=30, alpha_scale=0.5):
    """
    將黑色背景轉(zhuǎn)為半透明（保留陰影效果）
    
    參數(shù):
        alpha_scale: 透明度縮放因子（0-1，越小越透明）
    """
    img = Image.open(input_path).convert("RGBA")
    datas = img.getdata()
    
    new_data = []
    for item in datas:
        brightness = sum(item[:3]) / 3
        if brightness < threshold:
            # 保留原RGB，但降低Alpha通道
            new_alpha = int(item[3] * alpha_scale)
            new_data.append((item[0], item[1], item[2], new_alpha))
        else:
            new_data.append(item)
    
    img.putdata(new_data)
    img.save(output_path, "PNG")
    print(f"黑色背景已轉(zhuǎn)為半透明，保存至：{output_path}")

# 示例調(diào)用
black_to_semi_transparent("input_with_shadow.png", "output_semi_transparent.png", threshold=40, alpha_scale=0.3)