LaiTool/src/main/Service/MJ/mjDefine.ts

import { apiUrl } from "@/define/api/apiUrlDefine"
import { MJRobotType, MJSpeed } from "@/define/enum/mjEnum"

/**
 * 获取MJ的请求模式列表
 * @returns 
 */
function GetMJRequestModelOptions() {
  let mjRequstModel = [
    {
      label: "API模式",
      value: "api_mj",
      disable: false
    }, {
      label: "本地代理模式",
      value: "local_mj",
      disable: false
    },
    {
      label: "代理MJ（token）",
      value: "remote_mj",
      disable: false
    },
    {
      label: "浏览器模式",
      value: "browser_mj",
      disable: true
    }]
  return mjRequstModel.filter(item => !item.disable)
}

/**
 * 获取MJ的机器人列表
 * @returns 
 */
function GetMJRobotOptions() {
  return [{
    label: 'MJ',
    value: MJRobotType.MJ
  },
  {
    label: 'NIJI',
    value: MJRobotType.NIJI
  }]
}

/**
 * 获取机器人对应的出图模型
 */
function GetMJRobotModelOptions(mjRobot?: MJRobotType) {
  let allRobotModel = [
    {
      label: "MJ V6.0",
      text: "v 6",
      type: MJRobotType.MJ,
      value: "3e6473ab-9a64-4574-9a38-f5c75af552b6"
    },
    {
      label: "MJ V5.2",
      text: "v 5.2",
      type: MJRobotType.MJ,
      value: "27a0d30e-f46c-4684-96c8-d91334deb94f"
    },
    {
      label: "MJ V5.1",
      text: "v 5.1",
      type: MJRobotType.MJ,
      value: "e1226715-e969-44c4-b18b-f2ad5dae5d2f"
    }, {
      label: "MJ V5.0",
      text: "v 5",
      type: MJRobotType.MJ,
      value: "afb7bea1-4eda-46ea-8165-34701b4566bf"
    }, {
      label: "MJ V4.0",
      text: "v 4",
      type: MJRobotType.MJ,
      value: "d05b8497-7f4a-4890-8fac-89f1803984d2"
    }, {
      label: "NIJI V6",
      text: "niji 6",
      type: MJRobotType.NIJI,
      value: "99377cad-c103-4cee-a958-86a104879328"
    }, {
      label: "NIJI V5",
      text: "niji 5",
      type: MJRobotType.NIJI,
      value: "53cec077-9885-4635-ab18-e021066b2c4c"
    }, {
      label: "NIJI V4",
      text: "niji 4",
      type: MJRobotType.NIJI,
      value: "6a7199fe-6e0d-40a9-9772-b5eb3d2e2e66"
    },
  ]

  switch (mjRobot) {
    case MJRobotType.MJ:
      return allRobotModel.filter(item => item.type == MJRobotType.MJ)
    case MJRobotType.NIJI:
      return allRobotModel.filter(item => item.type == MJRobotType.NIJI)
    default:
      return allRobotModel
  }
}

/**
 * 获取MJ出图的比例Options
 * @returns 
 */
function GetMJImageScaleOptions() {
  return [{
    label: "1:1",
    text: "1:1",
    value: "3e2772f2-041c-49c6-ba13-d0ed120310b8"
  }, {
    label: "4:3",
    text: "4:3",
    value: "fcef555c-1958-4082-88fe-434782aa8151"
  }, {
    label: "3:4",
    text: "3:4",
    value: "13f71d53-73a3-4c9b-9c1e-6e7e939aee73"
  }, {
    label: "16:9",
    text: "16:9",
    value: "bf33ce1a-15cd-4901-b38e-89543cf14a1f"
  }, {
    label: "9:16",
    text: "9:16",
    value: "fd4641e2-97f4-4a86-8616-4965e05f3348"
  }]
}

/**
 * 获取MJ API 可用的URL Options
 * @returns 
 */
function GetMJAPIUrlOptions() {
  return apiUrl.filter((item) => item.mj_url)
}

/**
 * 获取MJ的速度Options
 * @returns 
 */
function GetMJSpeedOptions() {
  return [{
    label: "FAST",
    value: MJSpeed.FAST
  }, {
    label: "RELAXED",
    value: MJSpeed.RELAX
  }]
}

/**
 * MJ的一些数据的定义
 */
let MJDefine = {
  GetMJRequestModelOptions,
  GetMJRobotOptions,
  GetMJRobotModelOptions,
  GetMJImageScaleOptions,
  GetMJAPIUrlOptions,
  GetMJSpeedOptions
}

export default MJDefine