import cv2
import numpy as np

def detect_green(image):
    # Converter a imagem de BGR para HSV
    hsv = cv2.cvtColor(image, cv2.COLOR_BGR2HSV)
    
    # Definir os intervalos de cor verde (pode precisar ajustar os valores)
    lower_green = np.array([35, 50, 50])
    upper_green = np.array([90, 255, 255])

    # Criar uma máscara para a cor verde na faixa especificada
    mask = cv2.inRange(hsv, lower_green, upper_green)

    # Encontrar os contornos na máscara
    contours, _ = cv2.findContours(mask, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)

    # Criar uma imagem preta para desenhar as caixas delimitadoras
    black = np.zeros_like(image)

    # Desenhar caixas delimitadoras ao redor dos contornos detectados na imagem preta
    for contour in contours:
        x, y, w, h = cv2.boundingRect(contour)
        cv2.rectangle(black, (x, y), (x + w, y + h), (0, 255, 0), 2)
    
    # Mostrar apenas as áreas verdes na imagem original
    result = cv2.bitwise_and(image, image, mask=mask)
    result = cv2.add(result, black)  # Adicionar as caixas delimitadoras à imagem original
    
    return result

cap = cv2.VideoCapture(1)

while True:
    ret, frame = cap.read()
    if not ret:
        break

    # Detectar áreas verdes na imagem e mostrar apenas elas com as caixas delimitadoras
    green_boxes = detect_green(frame)

    cv2.imshow('Green Detection', green_boxes)
    if cv2.waitKey(1) & 0xFF == ord('q'):
        break

cap.release()
cv2.destroyAllWindows()