"""

pip install opencv-python

pip install opencv-python

pip install --upgrade cv2

pip uninstall pydub

"""

# Importação das bibliotecas necessárias

import cv2

import numpy as np

# Carrega a imagem a partir do arquivo 'peru.jpg'

Imagem = 'imagem.jpg'

img = cv2.imread(Imagem)

# Converte a imagem para escala de cinza

gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)

# Aplica um threshold na imagem para criar uma máscara e remover o fundo

mask = cv2.threshold(gray, 250, 255, cv2.THRESH_BINARY)[1]

# Inverte a máscara

mask = 255 - mask

# Aplica um algoritmo de morfologia para remover ruídos

kernel = np.ones((3,3), np.uint8)

mask = cv2.morphologyEx(mask, cv2.MORPH_OPEN, kernel)

mask = cv2.morphologyEx(mask, cv2.MORPH_CLOSE, kernel)

# Aplica um efeito de anti-aliasing na máscara

mask = cv2.GaussianBlur(mask, (0,0), sigmaX=2, sigmaY=2, borderType = cv2.BORDER_DEFAULT)

mask = (2*(mask.astype(np.float32))-255.0).clip(0,255).astype(np.uint8)

# Coloca a máscara no canal alfa da imagem original

result = img.copy()

result = cv2.cvtColor(result, cv2.COLOR_BGR2BGRA)

result[:, :, 3] = mask

# Salva a imagem resultante com fundo transparente em um arquivo PNG

cv2.imwrite('Imagem sem fundo.png', result)

cv2.imwrite('Imagem sem INPUT.png', img)

cv2.imwrite('Imagem sem GRAY.png', gray)