1

server/biz/controller/admin.go

@@ -2,6 +2,7 @@ package controller
 
 import (
 	"context"
+	"strings"
 
 	"github.com/gogf/gf/v2/net/ghttp"
 
@@ -86,11 +87,21 @@ func (c *cAdmin) UploadImages(ctx context.Context, req *dto.AdminDatasetUploadRe
 	return service.Dataset.AdminUploadImages(ctx, req)
 }
 
-// GenerateImages AI 生成图片
+// GenerateImages AI 生成图片（异步任务，返回 taskId 供轮询进度）
 func (c *cAdmin) GenerateImages(ctx context.Context, req *dto.AdminDatasetGenerateReq) (*dto.AdminDatasetGenerateRes, error) {
 	return service.Dataset.AdminGenerateImages(ctx, req)
 }
 
+// GenTaskQuery 生成任务进度查询
+func (c *cAdmin) GenTaskQuery(ctx context.Context, req *dto.AdminGenTaskQueryReq) (*dto.AdminGenTaskQueryRes, error) {
+	return service.Dataset.AdminGenTaskQuery(ctx, req)
+}
+
+// GenPools VLM 重新生成数据集生成参数池
+func (c *cAdmin) GenPools(ctx context.Context, req *dto.AdminGenPoolsReq) (*dto.AdminGenPoolsRes, error) {
+	return service.Dataset.AdminGenPools(ctx, req)
+}
+
 // ListImages 数据集图片列表
 func (c *cAdmin) ListImages(ctx context.Context, req *dto.AdminDatasetImagesReq) (*dto.AdminDatasetImagesRes, error) {
 	return service.Dataset.AdminListImages(ctx, req)
@@ -117,9 +128,10 @@ func (c *cAdmin) UploadCover(ctx context.Context, req *dto.AdminDatasetCoverUplo
 	return service.Dataset.AdminUploadCover(ctx, req)
 }
 
-// Cover 封面访问（直写响应体；URL 不随封面更换变化，必须 no-store 防浏览器缓存旧图）
+// Cover 封面访问（直写响应体；URL 不随封面更换变化，必须 no-store 防浏览器缓存旧图；
+// datasetId=0 时按 name+filename 直读——新建对话框预生成封面回显）
 func (c *cAdmin) Cover(ctx context.Context, req *dto.AdminDatasetCoverReq) (*dto.AdminDatasetCoverRes, error) {
-	path, err := service.Dataset.CoverFile(ctx, req.DatasetId)
+	path, err := service.Dataset.CoverFile(ctx, req.DatasetId, strings.TrimSpace(req.Name), strings.TrimSpace(req.Filename))
 	if err != nil {
 		return nil, err
 	}
@@ -134,6 +146,11 @@ func (c *cAdmin) DeleteCover(ctx context.Context, req *dto.AdminDatasetCoverDele
 	return service.Dataset.AdminDeleteCover(ctx, req)
 }
 
+// GenCover 生成数据集封面（z-image 文生图；编辑对话框「AI 生成封面」按钮入口，datasetId=0 新建预生成）
+func (c *cAdmin) GenCover(ctx context.Context, req *dto.AdminGenCoverReq) (*dto.AdminGenCoverRes, error) {
+	return service.Dataset.AdminGenCover(ctx, req)
+}
+
 // ---------- 训练编排 ----------
 
 // ListTrainings 训练任务列表
@@ -192,8 +209,3 @@ func (c *cAdmin) SaveLabel(ctx context.Context, req *dto.AdminLabelSaveReq) (*dt
 func (c *cAdmin) VlmReview(ctx context.Context, req *dto.AdminImageVlmReviewReq) (*dto.AdminImageVlmReviewRes, error) {
 	return service.LabelTask.AdminImageVlmReview(ctx, req)
 }
-
-// GenOptions 生成表单选项（物种池/场景池，config 维护）
-func (c *cAdmin) GenOptions(ctx context.Context, req *dto.AdminDatasetGenOptionsReq) (*dto.AdminDatasetGenOptionsRes, error) {
-	return service.Dataset.AdminGenOptions(ctx, req)
-}