import importlib.metadata
import os
from pathlib import Path

devnull_fd = os.open(os.devnull, os.O_WRONLY)
os.dup2(devnull_fd, 2)
os.close(devnull_fd)

import numpy as np
from ai_edge_litert.interpreter import Interpreter

MODEL_PATH = Path("support_score.tflite")
SAMPLE_INPUT = np.array(
    [[0.12, 0.24, 0.18, 0.30]],
    dtype=np.float32,
)

interpreter = Interpreter(model_path=str(MODEL_PATH))
interpreter.allocate_tensors()

input_details = interpreter.get_input_details()[0]
output_details = interpreter.get_output_details()[0]

interpreter.set_tensor(
    input_details["index"],
    SAMPLE_INPUT,
)
interpreter.invoke()
output = interpreter.get_tensor(output_details["index"])
output_text = np.array2string(
    output,
    precision=6,
    suppress_small=False,
)

print(f"ai_edge_litert={importlib.metadata.version('ai-edge-litert')}")
print(f"input_shape={input_details['shape'].tolist()}")
print(f"input_dtype={np.dtype(input_details['dtype']).name}")
print(f"output_shape={output_details['shape'].tolist()}")
print(f"output_dtype={np.dtype(output_details['dtype']).name}")
print(f"sample_output={output_text}")