export type EngineType =
  | "cartesian_image"
  | "cartesian_video"
  | "spherical_image"
  | "spherical_video"
  | "spherical_stereogram"
  | "spherical_stereovideo"

export interface Engine {
  type: EngineType
  label: string
  modelName: string
  modelUrl: string
  visible: boolean
  enabled: boolean
}

export const engines: Record<string, Engine> = {
  cartesian_image: {
    type: "cartesian_image",
    label: "Cartesian image",
    visible: true,
    enabled: true,
    modelName: "SDXL",
    modelUrl: "https://huggingface.co/stabilityai/stable-diffusion-xl-base-1.0",
  },
  cartesian_video: {
    type: "cartesian_video",
    label: "Cartesian video",
    visible: true,
    enabled: true,
    modelName: "Zeroscope",
    modelUrl: "https://huggingface.co/cerspense/zeroscope_v2_576w",
  },
  spherical_image: {
    type: "spherical_image",
    label: "Spherical image",
    visible: true,
    enabled: true,
    modelName: "SDXL 360",
    modelUrl: "https://huggingface.co/stabilityai/stable-diffusion-xl-base-1.0",
  },

  spherical_video: {
    type: "spherical_video",
    label: "Spherical video",
    visible: false,
    enabled: false,
    modelName: "",
    modelUrl: "",
  },

  spherical_stereogram: {
    type: "spherical_stereogram",
    label: "Spherical stereogram",
    visible: false,
    enabled: false,
    modelName: "SDXL 360",
    modelUrl: "https://huggingface.co/stabilityai/stable-diffusion-xl-base-1.0",
  },

  // A.K.A. the Holy Graal
  spherical_stereovideo: {
      type: "spherical_stereovideo",
      label: "Spherical stereovideo",
      visible: false,
      enabled: false,
      modelName: "",
      modelUrl: "",
    }
}

export const defaultEngine: EngineType = "cartesian_image"

export const getEngine = (type?: EngineType): Engine => engines[type || defaultEngine] || engines[defaultEngine]