"""
Constants for the demo.
"""

import os
from pathlib import Path

DEMO_DIRECTORY = Path(__file__).parent

ONNX_MODEL_DIRECTORY = DEMO_DIRECTORY / "onnx-models"
LEELA_MODEL_DIRECTORY = DEMO_DIRECTORY / "leela-models"
FIGURE_DIRECTORY = DEMO_DIRECTORY / "figures"

ONNX_MODEL_NAMES = [
    f for f in os.listdir(ONNX_MODEL_DIRECTORY)
    if f.endswith(".onnx")
]
LEELA_MODEL_NAMES = [
    f for f in os.listdir(LEELA_MODEL_DIRECTORY)
    if f.endswith(".pb.gz")
]