import os
from PIL import Image, ImageDraw


IMAGE_PATH = r"d:\dsWork\aiData\Output\Screenshot_20260117_073155.jpg"
OUTPUT_PATH = r"d:\dsWork\aiData\Output\rabbit_blue_rect_20260117_073155.jpg"


def draw_rabbit_blue_rect(image_path: str, output_path: str):
    if not os.path.exists(image_path):
        raise FileNotFoundError(image_path)

    img = Image.open(image_path).convert("RGB")
    w, h = img.size

    # 水平方向收一点边：左右各预留 4% 宽度
    x1 = int(w * 0.04)
    x2 = int(w * 0.96)
    # 垂直方向：稍微向上抬一点上边界，下边界大幅向上
    y1 = int(h * 0.74)
    y2 = int(h * 0.86)

    draw = ImageDraw.Draw(img)
    draw.rectangle((x1, y1, x2, y2), outline="blue", width=4)

    os.makedirs(os.path.dirname(output_path), exist_ok=True)
    img.save(output_path)
    print(f"saved: {output_path}")


if __name__ == "__main__":
    draw_rabbit_blue_rect(IMAGE_PATH, OUTPUT_PATH)