fix issue

templates/custom/select_model.tmpl

@@ -0,0 +1,37 @@
+<input type="hidden" id="ai_model_version" name="model_version" value="{{$.model_version}}">
+<div class="inline min_title  fields" style="{{if not .job_name}}width: 96.8%{{else}}width: 94.8%{{end}};">
+    <label class="{{if not .job_name}}label-fix-width{{end}}" style="font-weight: normal;">{{.i18n.Tr "repo.modelarts.infer_job.select_model"}}</label>&nbsp;
+    <div class="six wide field">
+        <div class="ui fluid search selection dropdown" id="select_model">
+            <input type="hidden" name="model_name" required value="{{$.model_name}}">
+            <div class="text"></div>
+            <i class="dropdown icon"></i>
+            <div class="menu" id="model_name">
+            </div>
+        </div>
+      </div>
+      <div class="three wide field">
+        <div class="ui fluid search selection dropdown" id="select_model_version">
+            <input type="hidden" name="pre_train_model_url" value="{{$.pre_train_model_url}}" required>
+            <div class="text"></div>
+            <i class="dropdown icon"></i>
+            <div class="menu" id="model_name_version">
+            </div>
+
+        </div>
+      </div>
+      <div class="five wide field">
+        <div class="ui fluid search selection dropdown" id="select_model_checkpoint">
+            <input type="hidden" name="ckpt_name" value="{{$.ckpt_name}}" required>
+            <div class="text"></div>
+            <i class="dropdown icon"></i>
+            <div class="menu" id="model_checkpoint">
+            </div>
+        </div>
+
+      </div>
+      <span >
+        <i class="question circle icon" data-content="{{.i18n.Tr "cloudbrain.model_file_postfix_rule"}}" data-position="top center" data-variation="inverted mini"></i>
+    </span>
+
+</div>

templates/repo/cloudbrain/trainjob/new.tmpl

@@ -70,7 +70,7 @@
 <div class="repository">
     {{template "repo/header" .}}
     <div class="ui container">
-        <div class="cloudbrain-type" style="display: none;" data-cloudbrain-type="{{.datasetType}}" data-repo-link="{{.RepoLink}}" data-queue="{{.QueuesDetail}}" data-queue-start="{{.i18n.Tr "repo.wait_count_start"}}" data-queue-end="{{.i18n.Tr "repo.wait_count_end"}}"></div>
+        <div class="cloudbrain-type" style="display: none;" data-cloudbrain-type="{{.datasetType}}" data-repo-link="{{.RepoLink}}" data-flag-model="true" data-queue="{{.QueuesDetail}}" data-queue-start="{{.i18n.Tr "repo.wait_count_start"}}" data-queue-end="{{.i18n.Tr "repo.wait_count_end"}}"></div>
         {{template "base/alert" .}}
         <h4 class="ui top attached header">
             {{.i18n.Tr "repo.modelarts.train_job.new"}}
@@ -168,7 +168,7 @@
                         {{end}}
                     </select>
                 </div>
-
+                {{template "custom/select_model" .}}
                 <div class="inline required field" style="display: none;">
                     <label>{{.i18n.Tr "cloudbrain.task_type"}}</label>
                     <select id="cloudbrain_job_type" class="ui search dropdown" placeholder="选择任务类型" style='width:385px'

templates/repo/grampus/trainjob/gpu/new.tmpl

@@ -62,6 +62,7 @@
 <div class="repository">
     {{template "repo/header" .}}
     <div class="ui container">
+        <div class="cloudbrain-type" style="display: none;" data-cloudbrain-type="{{.datasetType}}" data-repo-link="{{.RepoLink}}" data-flag-model="true"></div>
         {{template "base/alert" .}}
         <h4 class="ui top attached header">
             {{.i18n.Tr "repo.modelarts.train_job.new"}}
@@ -155,7 +156,7 @@
                             {{end}}
                         </select>
                 </div>
-
+                {{template "custom/select_model" .}}
                 <div id="images-new-grampus">
                 </div>
 

templates/repo/grampus/trainjob/npu/new.tmpl

@@ -57,6 +57,7 @@
 <div class="repository">
     {{template "repo/header" .}}
     <div class="ui container">
+        <div class="cloudbrain-type" style="display: none;" data-cloudbrain-type="{{.datasetType}}" data-repo-link="{{.RepoLink}}" data-flag-model="true"></div>
         {{template "base/alert" .}}
         <h4 class="ui top attached header">
             {{.i18n.Tr "repo.modelarts.train_job.new"}}
@@ -149,6 +150,7 @@
                             {{end}}
                         </select>
                 </div>
+                {{template "custom/select_model" .}}
                 <div class="required min_title inline field" id="engine_name">
                     <label class="label-fix-width" style="font-weight: normal;">{{.i18n.Tr "cloudbrain.mirror"}}</label>
                     <select class="ui dropdown width81" id="trainjob_images" name="image_id">

templates/repo/modelarts/trainjob/new.tmpl

@@ -62,7 +62,7 @@
 <div class="repository">
     {{template "repo/header" .}}
     <div class="ui container">
-        <div class="cloudbrain-type" style="display: none;" data-cloudbrain-type="{{.datasetType}}" data-repo-link="{{.RepoLink}}"></div>
+        <div class="cloudbrain-type" style="display: none;" data-cloudbrain-type="{{.datasetType}}" data-repo-link="{{.RepoLink}}" data-flag-model="true"></div>
         {{template "base/alert" .}}
         <h4 class="ui top attached header">
             {{.i18n.Tr "repo.modelarts.train_job.new"}}
@@ -157,8 +157,8 @@
                             {{end}}
                         </select>
                 </div>
-
-
+                
+                {{template "custom/select_model" .}}
 
                 <div class="required inline min_title fields" style="width: 95%;">
                     <label class="label-fix-width" style="font-weight: normal;">{{.i18n.Tr "repo.modelarts.train_job.AI_driver"}}</label>

templates/repo/modelarts/trainjob/version_new.tmpl

@@ -55,7 +55,7 @@
 <div class="repository">
     {{template "repo/header" .}}
     <div class="ui container">
-        <div class="cloudbrain-type" style="display: none;" data-cloudbrain-type="{{.datasetType}}" data-repo-link="{{.RepoLink}}" data-dataset-uuid="{{.uuid}}" data-dataset-name="{{.dataset_name}}"></div>
+        <div class="cloudbrain-type" style="display: none;" data-cloudbrain-type="{{.datasetType}}" data-repo-link="{{.RepoLink}}" data-flag-model="true" data-dataset-uuid="{{.uuid}}" data-dataset-name="{{.dataset_name}}"></div>
         {{template "base/alert" .}}
         <h4 class="ui top attached header">
             {{.i18n.Tr "repo.modelarts.train_job.new"}}
@@ -154,7 +154,7 @@
 
                 </div>
 
-
+                {{template "custom/select_model" .}}
 
                 <div class="required unite min_title inline fields" style="width: 90%;">
                     <label style="font-weight: normal;">{{.i18n.Tr "repo.modelarts.train_job.AI_driver"}}&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;</label>

web_src/js/features/cloudbrainShow.js

@@ -443,4 +443,134 @@ export default async function initCloudrainSow() {
     html += "</div>";
     $(`#dir_list${version_name}`).append(html);
   }
+
+  let nameMap, nameList;
+  let RepoLink = $(".cloudbrain-type").data("repo-link");
+  let type = $(".cloudbrain-type").data("cloudbrain-type");
+  let flagModel = $(".cloudbrain-type").data("flag-model");
+  // 获取模型列表和模型名称对应的模型版本
+  $(document).ready(function () {
+    if (!flagModel) return;
+    else {
+      $.get(
+        `${RepoLink}/modelmanage/query_model_for_predict?type=${type}`,
+        (data) => {
+          nameMap = data.nameMap;
+          nameList = data.nameList;
+          let html = "";
+          nameList.forEach((element) => {
+            html += `<div class="item" data-value=${element}>${element}</div>`;
+          });
+          if (nameList.length !== 0) {
+            $("#model_name").append(html);
+          }
+          let faildModelName = $('input[name="model_name"]').val();
+          let faildModelVersion = $('input[name="model_version"]').val();
+          let faildTrainUrl = $('input[name="pre_train_model_url"]').val();
+          let faildCkptName = $('input[name="ckpt_name"]').val();
+          // 新建错误的表单返回初始化
+          if (faildModelName) {
+            $("#select_model").dropdown("set text", faildModelName);
+            $("#select_model").dropdown("set value", faildModelName);
+            $("#select_model_version").dropdown("set text", faildModelVersion);
+            $("#select_model_version").dropdown("set value", faildTrainUrl);
+            $("#select_model_checkpoint").dropdown("set text", faildCkptName);
+            $("#select_model_checkpoint").dropdown("set value", faildCkptName);
+          }
+        }
+      );
+    }
+    $("#select_model").dropdown({
+      onChange: function (value, text, $selectedItem) {
+        $("#model_name_version").empty();
+        let html = "";
+        nameMap[value].forEach((element) => {
+          let { TrainTaskInfo } = element;
+          TrainTaskInfo = JSON.parse(TrainTaskInfo);
+          html += `<div class="item" data-label="${element.Label}" data-id="${element.ID}" data-value="${element.Path}">${element.Version}</div>`;
+        });
+        $("#model_name_version").append(html);
+        const initVersionText = $(
+          "#model_name_version div.item:first-child"
+        ).text();
+        const initVersionValue = $(
+          "#model_name_version div.item:first-child"
+        ).data("value");
+
+        $("#select_model_version").dropdown("set text", initVersionText);
+        $("#select_model_version").dropdown(
+          "set value",
+          initVersionValue,
+          initVersionText,
+          $("#model_name_version div.item:first-child")
+        );
+      },
+    });
+
+    $("#select_model_version").dropdown({
+      onChange: function (value, text, $selectedItem) {
+        const dataID =
+          $selectedItem && $selectedItem[0].getAttribute("data-id");
+        $("input#ai_model_version").val(text);
+        $("#select_model_checkpoint").addClass("loading");
+        $("#model_checkpoint").empty();
+        let html = "";
+        loadCheckpointList(dataID).then((res) => {
+          res.forEach((element) => {
+            const ckptSuffix = element.FileName.split(".");
+            const loadCheckpointFile = [
+              "ckpt",
+              "pb",
+              "h5",
+              "json",
+              "pkl",
+              "pth",
+              "t7",
+              "pdparams",
+              "onnx",
+              "pbtxt",
+              "keras",
+              "mlmodel",
+              "cfg",
+              "pt",
+            ];
+            if (
+              !element.IsDir &&
+              loadCheckpointFile.includes(ckptSuffix[ckptSuffix.length - 1])
+            ) {
+              html += `<div class="item" data-value=${element.FileName}>${element.FileName}</div>`;
+            }
+          });
+          $("#model_checkpoint").append(html);
+          $("#select_model_checkpoint").removeClass("loading");
+          const initVersionText = $(
+            "#model_checkpoint div.item:first-child"
+          ).text();
+          const initVersionValue = $(
+            "#model_checkpoint div.item:first-child"
+          ).data("value");
+
+          $("#select_model_checkpoint").dropdown("set text", initVersionText);
+          $("#select_model_checkpoint").dropdown(
+            "set value",
+            initVersionValue,
+            initVersionText,
+            $("#model_name_version div.item:first-child")
+          );
+        });
+      },
+    });
+  });
+
+  function loadCheckpointList(value) {
+    return new Promise((resolve, reject) => {
+      $.get(
+        `${RepoLink}/modelmanage/query_modelfile_for_predict`,
+        { ID: value },
+        (data) => {
+          resolve(data);
+        }
+      );
+    });
+  }
 }