JavaProjectRepo/business-css/frontend/src/views/business/algorithmManagement/modelTrainTask/index.vue

<script lang="ts">
export default {
  name: "模型训练",
};
</script>

<script setup lang="ts">
import { onMounted, ref, nextTick } from "vue";
import { ElForm, ElMessage, ElMessageBox } from "element-plus";
import { searchAlgorithmsPage,addAlgorithms,updateAlgorithms,deleteAlgorithms,deleteBatchAlgorithms} from "@/api/business/algorithm";
import { getDictItemById } from '@/api/dict';
import Page from '@/components/Pagination/page.vue'
const apiUrl = import.meta.env.VITE_APP_BASE_API; // 基础路径
// 搜索框
const queryParams = ref({
  current: 1,
  size: 10,
  querystr: ''
});
//分页 总条数
const total = ref()
//定义表格数据

const tableData: any = ref([]);
// 表格加载
const loading = ref(false)
function gettableData() {
  let params = {
    name: input.value,
    pageNum: queryParams.value.current,
    pageSize: queryParams.value.size,
  };
  loading.value = true;
  searchAlgorithmsPage(params).then((result:any) => {
    tableData.value = result.records;
    total.value = result.total;
    loading.value = false;
  }).catch((err) => {
    loading.value = false;
  });
}

const algorithmData: any = ref([]);
function getAlgorithmList() { // 获取算法列表
  let params = {
    name: "",
    pageNum: 1,
    pageSize: 999,
  };
  loading.value = true;
  searchAlgorithmsPage(params).then((result:any) => {
    algorithmData.value = result.records;
  }).catch((err) => {
  });
}

const menuData:any = ref([]);
// 查询字典项
function menuInit() { // 获取算法类型字典项
  let params = {
    dictId: 'fe2c3418b8998f4e64d56ab46bfe0fed',
    size:99,
    current:1
  }
  getDictItemById(params).then((result: any) => {
    menuData.value = result.data.records;
  }).catch((err: any) => {
  });
}

const infoForm = ref();
//搜索内容及点击搜索按钮
const input = ref("");
//新建训练任务
const title = ref("");
const info: any = ref({
  name: "",
  description: "",
  version: "",
  principle: "",
  status:1,
  algorithmType: "",
  inputParams: "",
  outputParams: "",
  inferBaseUrl: "",
  trainBaseUrl: "",
  supportedDeviceTypes: "",
  defaultHyperParams: "",
});

const dialogVisible = ref(false);
function addClick() {
  title.value = "新增训练任务";
  info.value = {
    name: "",
    description: "",
    version: "",
    principle: "",
    status:1,
    algorithmType: "",
    inputParams: "",
    outputParams: "",
    inferBaseUrl: "",
    trainBaseUrl: "",
    supportedDeviceTypes: "",
    defaultHyperParams: "",
  };
  dialogVisible.value = true;
}
//新建训练任务-确认按钮/修改按钮
function confirmClick(formEl: any) {
  formEl.validate((valid: any) => {
    if (valid) {
      const params = {
        algorithmType: info.value.algorithmType,
        name: info.value.name,
        description: info.value.description,
        version: info.value.version,
        principle: info.value.principle,
        inputParams: info.value.inputParams,
        outputParams: info.value.outputParams,
        inferBaseUrl: info.value.inferBaseUrl,
        trainBaseUrl: info.value.trainBaseUrl,
        supportedDeviceTypes: info.value.supportedDeviceTypes, 
        defaultHyperParams: info.value.defaultHyperParams,
        status: info.value.status,
      }
      addAlgorithms(params).then((res) => {
        gettableData();
        dialogVisible.value = false;
      });
    }
  });
}

//新建角色-取消按钮
function handleClose() {
  dialogVisible.value = false;
  dialogViewVisible.value = false;
  if (infoForm.value != null) infoForm.value.resetFields();
}
//新建训练任务
const rules = ref({
  name: [{ required: true, message: "请输入训练任务名称", trigger: "blur" }],
  algorithmType: [{ required: true, message: "请选择算法类型", trigger: "change" }]
});
//修改训练任务
function viewClick(row: any) {
  title.value = "查看详情";
  info.value = JSON.parse(JSON.stringify(row));
  dialogViewVisible.value = true;
}

//删除训练任务
function delAloneClick(row: any) {
  ElMessageBox.confirm("确定删除此训练任务吗？", "删除提示", {
    confirmButtonText: "确定",
    cancelButtonText: "取消",
    type: "warning",
  })
    .then(() => {
      let params = {
        id: row.algorithmId,
      };
      deleteAlgorithms(params).then(() => {
        gettableData();
        ElMessage({
          type: "success",
          message: "删除成功",
        });
      });
    })
}


// 多选删除？
function delClick(row:any) {
  ElMessageBox.confirm("确定删除已选择训练任务吗?", "删除提示", {
    confirmButtonText: "确定",
    cancelButtonText: "取消",
    type: "warning",
  }).then(() => {
    deleteBatchAlgorithms(row.id).then(() => {
      gettableData();
      ElMessage({
        message: "删除成功",
        type: "success",
      });
    });
  });
}
function dateFormat(row: any) {
  const daterc = row;
  if (daterc != null) {
    var date = new Date(daterc);
    var year = date.getFullYear();
    var month =
      date.getMonth() + 1 < 10 ? "0" + (date.getMonth() + 1) : date.getMonth() + 1;
    date.getMonth() + 1 < 10 ? "0" + (date.getMonth() + 1) : date.getMonth() + 1;
    var day = date.getDate() < 10 ? "0" + date.getDate() : date.getDate();
    var hours = date.getHours() < 10 ? "0" + date.getHours() : date.getHours();
    var minutes = date.getMinutes() < 10 ? "0" + date.getMinutes() : date.getMinutes();
    var seconds = date.getSeconds() < 10 ? "0" + date.getSeconds() : date.getSeconds();
    // 拼接
    return year + "-" + month + "-" + day + " " + hours + ":" + minutes + ":" + seconds;
  }
}
onMounted(() => {
  getAlgorithmList() // 获取算法列表
  menuInit() // 获取算法类型字典项
  gettableData();
});
function changeStatus(row: any) {
  // let params = {
  //   algorithmId: row.algorithmId,
  //   status: row.status,
  // };
  // updateAlgorithms(params).then((res) => {
  //   gettableData();
  //   ElMessage({
  //     type: "success",
  //     message: "更新成功",
  //   });
  // });
}

function releaseClick(row: any) {
  title.value = "发布模型";
  info.value = JSON.parse(JSON.stringify(row));
  // dialogVisible.value = true;
}

const isEchartsModel = ref(false) // 是否显示echarts模型

function changeShowResult(isShow:boolean){ // 切换显示结果模型
  isEchartsModel.value = isShow

}

const dialogViewVisible = ref(false) // 模型详情弹窗
</script>

<template>
  <div class="Algorithms-box">
    <div class="conductproject-bg-box">
      <div
        style="display: flex;display: -webkit-flex; justify-content: space-between; -webkit-justify-content: space-between;margin-bottom: 10px">
        <div style="display: flex;display: -webkit-flex;">
          <el-input v-model="input" placeholder="请输入训练任务名称" @keyup.enter="gettableData" style="width: 200px" clearable />
          <el-button type="primary" style="margin-left: 10px" @click="gettableData">搜索</el-button>
        </div>
        <div style="display: flex;display: -webkit-flex;">
          <el-button  type="primary" @click="addClick">
            新增</el-button>

        </div>
      </div>
      <el-table v-loading="loading" :data="tableData" style="width: 100%; height: calc(100vh - 260px);margin-bottom: 10px;" border
        :header-cell-style="{ background: 'rgb(250 250 250)', color: '#383838', height: '50px' }">
        <el-table-column type="selection" width="50" align="center"></el-table-column>
        <el-table-column type="index" label="序号" width="70" align="center"></el-table-column>
        <el-table-column prop="name" label="任务名称" width="180"></el-table-column>
        <el-table-column prop="version" label="算法类型" width="100"></el-table-column>
        <el-table-column prop="description" label="设备类型" min-width="100"></el-table-column>
        <el-table-column prop="description" label="任务状态" min-width="100"></el-table-column>

        <el-table-column prop="updatedAt" label="创建时间" width="200">
          <template #default="scope">
            {{ dateFormat(scope.row.updatedAt) }}
          </template>
        </el-table-column>
        <el-table-column fixed="right" label="操作" width="100" align="center">
          <template #default="scope">
             <span
              style="display: flex;display: -webkit-flex; justify-content: space-around;-webkit-justify-content: space-around; ">
               <img  src="@/assets/table/view.png" alt="" title="查看详情"
                  @click="viewClick(scope.row)" style="cursor: pointer; ">

               <img  src="@/assets/table/annotation.png" alt="" title="发布模型"
                  @click="releaseClick(scope.row)" style="cursor: pointer; ">
   

              <img  src="@/assets/table/del.png" alt="" title="删除"
                @click="delAloneClick(scope.row)" style="cursor: pointer; "> 
            </span>
          </template>
        </el-table-column>
      </el-table>
      <Page :total="total" v-model:size="queryParams.size" v-model:current="queryParams.current" @pagination="gettableData()" ></Page>

    </div>

    <el-dialog v-model="dialogVisible" :close-on-click-modal="false"
     :modal="false" draggable :before-close="handleClose" :title="title"
      append-to-body width="1145px" height="600px">
      <el-form ref="infoForm" :model="info" :rules="rules" label-width="100px" 
      style="margin-top: 20px;">
        <el-form-item label="训练名称名称" prop="name" style="width: 100%;" >
          <el-input v-model="info.task_name" style="width: 100%" placeholder="输入算法名称"></el-input>
        </el-form-item>
        
        <span class="dialog-footer"
          style="display: flex;display: -webkit-flex; justify-content: flex-end;-webkit-justify-content: flex-end;">
          <el-button @click="handleClose">取 消</el-button>
          <el-button type="primary" @click="confirmClick(infoForm)">确 定</el-button>
        </span>
      </el-form>
    </el-dialog>

    <el-dialog v-model="dialogViewVisible" :close-on-click-modal="false"
      :modal="false" draggable :before-close="handleClose" :title="'模型详情'"
      append-to-body width="1145px" height="600px">
      <div style=" width: calc(100%);    height: calc(100vh - 200px);">
        <div style="display: flex; margin-bottom: 20px; border-bottom: 1px solid #e5e5e5;padding-bottom: 5px;">
          <div @click="changeShowResult(false)" class="adddevice_navigation_left" :class="{'adddevice_navigation_activeleft':!isEchartsModel}">基本信息</div>
          <div @click="changeShowResult(true)" class="adddevice_navigation_right" :class="{'adddevice_navigation_activeright':isEchartsModel}">指标与图表</div>
        </div>
        <el-form ref="infoForm" :model="info" :rules="rules" label-width="100px" 
          style="margin-top: 20px;" v-if="!isEchartsModel">
          <el-form-item label="版本" style="width: 100%;" >
            <el-input v-model="info.versionTag" style="width: 100%" placeholder="输入算法名称" :disabled="true"></el-input>
          </el-form-item>
        </el-form>
        <div v-else style="width: 100%; height: calc(100% - 50px);">
          <img :src="apiUrl +'/models/'+ info.metricsImagePath" alt="" style="width: 100%; height: 100%;object-fit:contain;">
        </div>
      </div>
   
    </el-dialog>
  </div>
</template>

<style scoped>
.Algorithms-box {
  padding-right: 10px;
}

:deep(.el-tree-node__content) {
  font-size: 15px;
  font-weight: 500;
  width: 100%;
  height: 40px;
}

.dialog-footer {
  display: block;
  margin-top: 20px;
}

.conductproject-bg-box {
  padding: 20px;
  width: 100%;
  height: calc(100vh - 130px);
  overflow: auto;
  background-color: rgba(255, 255, 255, 1);
  border: none;
  border-radius: 3px;
  box-sizing: border-box;
}
.Algorithms_dialog_tabbox{
  display: flex;
  border-bottom: 1px solid #e4e7ed;
  padding-bottom: 10px;
}
.Algorithms_dialog_tabbox_item{
  position: relative;
  display: flex;
  justify-content: center;
  align-items: center;
  cursor: pointer;
}

.Algorithms_dialog_tabbox_item span{
  position: absolute;
  top: 0;
  left: 0;
  width: 100%;
  height: 100%;
  display: flex;
  justify-content: center;
  align-items: center;
  font-family: 'Arial Normal', 'Arial';
  font-weight: 400;
  font-style: normal;
  font-size: 14px;
  color: #363636;
}
.Algorithms_dialog_tabbox_item span:hover{
  color: #266fff;
}
.Algorithms_dialog_tabbox_active span{
  color: #fff !important;
}
</style>
<style>
.el-dialog {
  padding: 0 !important;
  border-radius: 10px !important;
  border: 1px solid #363636 !important;
}
.el-dialog  .el-dialog__header{
  display: flex;
  display: -webkit-flex;
  justify-content: flex-start;-webkit-justify-content: flex-start;
  align-items: center;-webkit-align-items: center;
  padding: 10px 20px;
  background-color: #f1f3f8 !important;
  font-family: 'Arial Negreta', 'Arial Normal', 'Arial', sans-serif;
  font-weight: 700;
  font-style: normal;
  font-size: 16px;
  color: #1B1B1B;
  text-align: left;
  border-radius: 10px 10px 0 0;
  height: 50px;
}
.el-dialog .el-dialog__close{
   font-size: 22px;
   color: rgb(80, 80, 80);
}
.el-dialog .el-dialog__headerbtn{
  display: flex;
  align-items: center;
}
.el-dialog .el-dialog__body{
  padding: 20px 40px !important;
}
.el-dialog .el-input{
 --el-input-inner-height: 38px
}
.Algorithms-box .el-button{
  height: 36px;
}
.el-dialog .el-button{
  height: 40px;
}


</style>