from pathlib import Path

from PIL import Image, ImageDraw
from sentence_transformers import SentenceTransformer


IMAGE_DIR = Path("demo-images")
CATALOG = [
    ("red-square.png", "red square", (220, 30, 30), "square"),
    ("blue-circle.png", "blue circle", (30, 80, 220), "circle"),
    ("green-triangle.png", "green triangle", (40, 155, 75), "triangle"),
]