export type EngineType = "image" | "video"

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

export const engines: Record<string, Engine> = {
  image: {
    type: "image",
    label: "Image",
    modelName: "SDXL 1.0",
    modelUrl: "https://huggingface.co/stabilityai/stable-diffusion-xl-base-1.0",
  },
  video: {
    type: "video",
    label: "Video",
    modelName: "Zeroscope V2 576w",
    modelUrl: "https://huggingface.co/cerspense/zeroscope_v2_576w",
  }
}

export const defaultEngine: EngineType = "image"

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