import {
  loadTensorflowModel,
  type TensorflowModel,
} from '@get-id/react-native-fast-tflite';
import { postProcessOutput } from './documentDetection';

const MODEL_URL =
  'https://cdn.getid.cloud/doc-detection/2024-04-22/doc-detector.tflite';

export async function loadDocumentDetectionModel() {
  return await loadTensorflowModel({ url: MODEL_URL });
}

const QUALITY_CHECK_MODEL_URL =
  'https://cdn.getid.cloud/doc-qc/doc_qc_glare_v1_7.tflite';

export async function loadQualityCheckModel() {
  return await loadTensorflowModel({
    url: QUALITY_CHECK_MODEL_URL,
  });
}

export function detectDocument(
  model: TensorflowModel,
  frame: Float32Array,
  scaleSize: {
    width: number;
    height: number;
  } = { width: 1, height: 1 }
) {
  'worklet';

  const outputs = model.runSync([frame]);
  return postProcessOutput(outputs[0], scaleSize.width, scaleSize.height);
}