test.py

import os
import sys

from pathlib import Path
from PIL import Image

ROOT_DIR = Path(__file__).resolve().parent
sys.path.insert(0, str(ROOT_DIR))

from src.moderator import moderate

for img_path in sorted((ROOT_DIR / 'images').iterdir()):
    if img_path.suffix.lower() not in {'.jpg', '.jpeg', '.png'}:
        continue

    img = Image.open(img_path)
    fmt = 'PNG' if img_path.suffix.lower() == '.png' else 'JPEG'
    print(f"\n=== {img_path.name} ({img.size}) ===")
    result = moderate(img, fmt)
    print(f"action:     {result.action}")
    print(f"risk_level: {result.risk_level}")
    print(f"labels:     {result.labels}")
    print(f"score:      {round(result.score, 3)}")
    print(f"evidence:   {result.evidence}")