import pyautogui
from PIL import Image, ImageDraw

def count_images_on_screen(image_path, confidence=0.992):
    # Görüntüleri bulmak için pyautogui kullanın
    found_images = list(pyautogui.locateAllOnScreen(image_path, confidence=confidence))
    return found_images

def save_image_with_boxes(image_path, found_images, output_path):
    # Ekran görüntüsünü alın
    screenshot = pyautogui.screenshot()

    # Ekran görüntüsünü Pillow Image formatına çevirin
    screenshot_image = Image.frombytes('RGB', screenshot.size, screenshot.tobytes())

    # Çizim yapabilmek için bir ImageDraw nesnesi oluşturun
    draw = ImageDraw.Draw(screenshot_image)

    for box in found_images:
        # Kare içine alma
        left, top, width, height = box
        right = left + width
        bottom = top + height
        draw.rectangle([left, top, right, bottom], outline="red", width=2)

    # Sonuç resmini kaydedin
    screenshot_image.save(output_path)

# Resmin dosya yolunu belirtin
image_path = "images/icon.png"

# Ekrandaki resimleri bulun
found_images = count_images_on_screen(image_path)

# Kaydedilecek resim dosyasının yolu
output_path = "output_image_with_boxes.png"

# Ekrandaki eşleşmeleri bir resim olarak kaydedin ve kare içine alın
save_image_with_boxes(image_path, found_images, output_path)

# Sonucu konsola yazdırın
print(f"Ekranda {image_path} resimlerinden {len(found_images)} adet var.")