# NOVIC interface

# Imports
import sys
import random  # TODO: TEMP
import pathlib
import subprocess
import PIL.Image
# TODO: Then start coding infer.py (ANY additional import has to be clarified with novic_spaces/requirements.txt (pip install -r requirements.txt) and novic/requirements-infer.txt

# Ensure access to NOVIC code
REPO_ROOT = pathlib.Path(__file__).parent.resolve()
NOVIC_DIR = REPO_ROOT / 'novic'
NOVIC_TEST = NOVIC_DIR / '__init__.py'
if not NOVIC_TEST.exists():
	print("Initialising git submodules as NOVIC code was not found yet...")
	subprocess.run(['git', 'submodule', 'update', '--init', '--recursive'], cwd=REPO_ROOT, check=True)
	if not NOVIC_TEST.exists():
		raise RuntimeError("Failed to find NOVIC code")
if str(NOVIC_DIR) not in sys.path:
	sys.path.insert(0, str(NOVIC_DIR))

# NOVIC imports
import infer  # TODO: TEMP
print(infer)  # TODO: TEMP

# Classify an image
def classify_image(image: PIL.Image.Image, checkpoint: str) -> dict[str, float]:
	# TODO: TEMP
	print(image, checkpoint)
	labels = ('thing', 'item', 'stuff', 'object', 'entity')
	probs = [random.random() for _ in labels]
	prob_sum = sum(probs)
	return {label: prob / prob_sum for label, prob in zip(labels, probs)}
# EOF