fix: 数据扩增修改

src/api/modules/task/task.ts

@@ -9,7 +9,9 @@ import { TaskVO, TaskForm, TaskQuery } from '@/api/interface/task/task'
 export const createTaskApi = (data: any) => {
   return http.post<any>('/identification/identificationTask/create', data, { loading: false })
 }
-
+export const createExTaskApi = (data: any) => {
+  return http.post<any>('/demo/data/amplify', data, { loading: false })
+}
 /**
  * @name 查询算法任务列表
  * @param query 参数

src/views/demo/data/index.vue

@@ -12,7 +12,7 @@
         <el-button type="primary" v-auth="['demo:data:add']" :icon="CirclePlus" @click="openDialog(1, '数据新增')"> 新增 </el-button>
         <el-button type="primary" v-auth="['demo:data:import']" :icon="Upload" plain @click="batchAdd"> 导入数据集 </el-button>
         <el-button type="primary" v-auth="['demo:data:export']" :icon="Download" plain @click="downloadFile(scope.selectedListIds)"> 导出 </el-button>
-        <el-button type="primary" v-auth="['system:user:add']" :icon="CirclePlus" @click="dataAmplify()"> 数据增广 </el-button>
+        <el-button type="primary" v-auth="['system:user:add']" :icon="CirclePlus" @click="expandDataDialogVisible = true"> 数据增广 </el-button>
 
         <el-button
           type="danger"
@@ -48,6 +48,23 @@
       :json-data="jsonData"
     >
     </ImgDetect>
+
+    <el-dialog v-model="expandDataDialogVisible" title="数据扩增超参配置">
+      <el-container style="margin-top: 5px">
+        <AmplifyForm
+          @update-data="updateBatchData"
+          @update-model="updateModelData"
+          @update-params="updateParamsList"
+          :show-select-batch-button="true"
+        />
+      </el-container>
+      <el-container style="height: 10px"></el-container>
+
+      <span class="dialog-footer">
+        <el-button type="primary" @click="submit">确 定</el-button>
+        <el-button @click="expandDataDialogVisible = false">取 消</el-button>
+      </span>
+    </el-dialog>
   </div>
 </template>
 
@@ -80,6 +97,67 @@ import { listDataApi as listDictDataApi } from '@/api/modules/system/dictData'
 import { uploadPure } from '@/api/modules/upload'
 import http from '@/api'
 import { useRouter } from 'vue-router'
+import AmplifyForm from '@/views/demo/data/AmplifyForm.vue'
+import { createExTaskApi, createTaskApi } from '@/api/modules/task/task'
+
+const expandDataDialogVisible = ref(false)
+let selectedBatchDataList1 = ref([] as any[])
+let model = ref({})
+let parameList = ref([] as any[])
+
+// 监听 AmplifyForm 组件传递的数据
+const updateBatchData = data => {
+  selectedBatchDataList1.value = data
+}
+
+const updateModelData = data => {
+  model.value = data
+}
+
+// 监听 parameList 更新
+const updateParamsList = data => {
+  parameList.value = data
+}
+
+// 数据扩增超参修改保存
+const submitExpandDataConfigDialog = () => {
+  expandDataDialogVisible.value = false
+}
+
+const getTrainAugmentationParams = () => {
+  let batchNum = selectedBatchDataList1.value.map(batch => batch.batchNum).join()
+
+  // 将 parameList 中的值也传递到接口
+  parameList.value.forEach(item => {
+    item.value = model.value[`${item.agName}`]
+  })
+
+  return {
+    batchNum,
+    taskName: model.value.taskName,
+    otherParams: parameList.value
+  }
+}
+
+const submit = () => {
+  const params = {
+    trainAugmentationParams: getTrainAugmentationParams()
+  }
+
+  createExTaskApi(getTrainAugmentationParams())
+    .then(res => {
+      // console.log(res)
+      if (res.code !== 200) {
+        ElMessage.error(res.msg)
+      } else {
+        expandDataDialogVisible.value = false
+        ElMessage.success('数据扩增完成')
+      }
+    })
+    .catch(err => {
+      console.log(err)
+    })
+}
 
 onMounted(() => {
   state.cacheData.url = 'http://localhost:9090/profile/upload/2024/08/08/144745610/test.png'