import { MLKitCameraView, MLKitVisionOptions, MLKitVisionResult } from "../index";

export interface MLKitCustomModelResultValue {
  text: string;
  confidence: number;
}

export interface MLKitCustomModelResult extends MLKitVisionResult {
  result: Array<MLKitCustomModelResultValue>;
}

export type MLKitCustomModelType = "FLOAT32" | "QUANT";

export interface TNSCustomModelInput {
  shape: Array<number>,
  type: MLKitCustomModelType
}

// see https://firebase.google.com/docs/ml-kit/ios/use-custom-models
export interface MLKitCustomModelOptions extends MLKitVisionOptions {
  localModelFile?: string;
  labelsFile: string;
  /**
   * Default 5
   */
  maxResults?: number;
  modelInput: Array<TNSCustomModelInput>
  /**
   * Ignoring this for now as we deduct it from the model spec.
   */
  // modelOutput?: Array<{
  //   shape: Array<number>,
  //   type: MLKitCustomModelType
  // }>
  /**
   * Never got this working, so not supporting it for now.
   */
  // cloudModelName?: string;
  /**
   * Default false
   */
  // requireWifiForCloudModelDownload?: boolean;
}

export declare function useCustomModel(options: MLKitCustomModelOptions): Promise<MLKitCustomModelResult>;

export declare class MLKitCustomModel extends MLKitCameraView {
}