AI3D 页面功能完成

2026-04-24 20:14:26 +08:00 · 2025-09-03 16:00:28 +08:00
parent ead30c8779
commit 54c8856adf
21 changed files with 1795 additions and 1319 deletions
--- a/api/core/types/ai3d.go
+++ b/api/core/types/ai3d.go
@@ -22,6 +22,13 @@ type Gitee3DConfig struct {
 	Models  []AI3DModel `json:"models,omitempty"`
 }

+type AI3DTaskType string
+
+const (
+	AI3DTaskTypeTencent AI3DTaskType = "tencent"
+	AI3DTaskTypeGitee   AI3DTaskType = "gitee"
+)
+
 // AI3DJobResult 3D任务结果
 type AI3DJobResult struct {
 	JobId      string `json:"job_id"`      // 任务ID
--- a/api/handler/admin/upload_handler.go
+++ b/api/handler/admin/upload_handler.go
@@ -31,12 +31,12 @@ func NewUploadHandler(app *core.AppServer, db *gorm.DB, manager *oss.UploaderMan

 // RegisterRoutes 注册路由
 func (h *UploadHandler) RegisterRoutes() {
-	group := h.App.Engine.Group("/api/admin/upload/")
+	group := h.App.Engine.Group("/api/admin/upload")

 	// 需要管理员授权的接口
 	group.Use(middleware.AdminAuthMiddleware(h.App.Config.AdminSession.SecretKey, h.App.Redis))
 	{
-		group.POST("upload", h.Upload)
+		group.POST("", h.Upload)
 	}
 }

--- a/api/handler/ai3d_handler.go
+++ b/api/handler/ai3d_handler.go
@@ -46,42 +46,61 @@ func (h *AI3DHandler) RegisterRoutes() {
 	{
 		group.POST("generate", h.Generate)
 		group.GET("jobs", h.JobList)
+		group.GET("jobs/mock", h.ListMock) // 演示数据接口
 		group.GET("job/:id", h.JobDetail)
-		group.DELETE("job/:id", h.DeleteJob)
+		group.GET("job/delete", h.DeleteJob)
 		group.GET("download/:id", h.Download)
 	}
 }

 // Generate 创建3D生成任务
 func (h *AI3DHandler) Generate(c *gin.Context) {
-	var request vo.AI3DJobCreate
+	var request struct {
+		// 通用参数
+		Type       types.AI3DTaskType `json:"type" binding:"required"`  // API类型 (tencent/gitee)
+		Model      string             `json:"model" binding:"required"` // 3D模型类型
+		Prompt     string             `json:"prompt"`                   // 文本提示词
+		ImageURL   string             `json:"image_url"`                // 输入图片URL
+		FileFormat string             `json:"file_format"`              // 输出文件格式
+		// 腾讯3d专有参数
+		EnablePBR bool `json:"enable_pbr"` // 是否开启PBR材质
+		// Gitee3d专有参数
+		Texture           bool    `json:"texture"`             // 是否开启纹理
+		Seed              int     `json:"seed"`                // 随机种子
+		NumInferenceSteps int     `json:"num_inference_steps"` //迭代次数
+		GuidanceScale     float64 `json:"guidance_scale"`      //引导系数
+		OctreeResolution  int     `json:"octree_resolution"`   // 3D 渲染精度，越高3D 细节越丰富
+	}
 	if err := c.ShouldBindJSON(&request); err != nil {
 		resp.ERROR(c, "参数错误")
 		return
 	}

-	// 验证必填参数
-	if request.Type == "" || request.Model == "" || request.Power <= 0 {
-		resp.ERROR(c, "缺少必要参数")
+	// 提示词和图片不能同时为空
+	if request.Prompt == "" && request.ImageURL == "" {
+		resp.ERROR(c, "提示词和图片不能同时为空")
 		return
 	}

-	// 获取用户ID
-	userId := h.GetLoginUserId(c)
-	if userId == 0 {
-		resp.ERROR(c, "用户未登录")
+	// Gitee 只支持图片
+	if request.Type == types.AI3DTaskTypeGitee && request.ImageURL == "" {
+		resp.ERROR(c, "Gitee 只支持图生3D")
 		return
 	}

-	// 创建任务
-	job, err := h.service.CreateJob(uint(userId), request)
-	if err != nil {
-		resp.ERROR(c, fmt.Sprintf("创建任务失败: %v", err))
-		return
-	}
+	logger.Infof("request: %+v", request)
+
+	// // 获取用户ID
+	// userId := h.GetLoginUserId(c)
+	// // 创建任务
+	// job, err := h.service.CreateJob(uint(userId), request)
+	// if err != nil {
+	// 	resp.ERROR(c, fmt.Sprintf("创建任务失败: %v", err))
+	// 	return
+	// }

 	resp.SUCCESS(c, gin.H{
-		"job_id":  job.Id,
+		"job_id":  0,
 		"message": "任务创建成功",
 	})
 }
@@ -147,7 +166,7 @@ func (h *AI3DHandler) JobDetail(c *gin.Context) {
 		Type:       job.Type,
 		Power:      job.Power,
 		TaskId:     job.TaskId,
-		ImgURL:     job.FileURL,
+		FileURL:    job.FileURL,
 		PreviewURL: job.PreviewURL,
 		Model:      job.Model,
 		Status:     job.Status,
@@ -163,24 +182,38 @@ func (h *AI3DHandler) JobDetail(c *gin.Context) {
 // DeleteJob 删除任务
 func (h *AI3DHandler) DeleteJob(c *gin.Context) {
 	userId := h.GetLoginUserId(c)
-	if userId == 0 {
-		resp.ERROR(c, "用户未登录")
+	id := c.Query("id")
+	if id == "" {
+		resp.ERROR(c, "任务ID不能为空")
 		return
 	}

-	idStr := c.Param("id")
-	id, err := strconv.ParseUint(idStr, 10, 32)
+	var job model.AI3DJob
+	err := h.DB.Where("id = ?", id).Where("user_id = ?", userId).First(&job).Error
 	if err != nil {
-		resp.ERROR(c, "任务ID格式错误")
+		resp.ERROR(c, err.Error())
 		return
 	}

-	err = h.service.DeleteJob(uint(id), uint(userId))
+	err = h.DB.Delete(&job).Error
 	if err != nil {
-		resp.ERROR(c, fmt.Sprintf("删除任务失败: %v", err))
+		resp.ERROR(c, err.Error())
 		return
 	}

+	// 失败的任务要退回算力
+	if job.Status == types.AI3DJobStatusFailed {
+		err = h.userService.IncreasePower(userId, job.Power, model.PowerLog{
+			Type:   types.PowerRefund,
+			Model:  job.Model,
+			Remark: fmt.Sprintf("删除任务，退回%d算力", job.Power),
+		})
+		if err != nil {
+			resp.ERROR(c, err.Error())
+			return
+		}
+	}
+
 	resp.SUCCESS(c, gin.H{"message": "删除成功"})
 }

@@ -252,3 +285,110 @@ func (h *AI3DHandler) GetConfigs(c *gin.Context) {

 	resp.SUCCESS(c, config3d)
 }
+
+// ListMock 返回演示数据
+func (h *AI3DHandler) ListMock(c *gin.Context) {
+	// 创建各种状态的演示数据
+	mockJobs := []vo.AI3DJob{
+		{
+			Id:         1,
+			UserId:     1,
+			Type:       "gitee",
+			Power:      10,
+			TaskId:     "mock_task_1",
+			FileURL:    "https://img.r9it.com/R03TQZ7PZ386RGL7PTMNGFOHAJW15WYF.glb",
+			PreviewURL: "/static/upload/2025/9/1756873317505073.png",
+			Model:      "gitee-3d-v1",
+			Status:     types.AI3DJobStatusCompleted,
+			ErrMsg:     "",
+			Params:     `{"prompt":"一只可爱的小猫","image_url":"","texture":true,"seed":42}`,
+			CreatedAt:  1704067200, // 2024-01-01 00:00:00
+			UpdatedAt:  1704067800, // 2024-01-01 00:10:00
+		},
+		{
+			Id:         2,
+			UserId:     1,
+			Type:       "tencent",
+			Power:      15,
+			TaskId:     "mock_task_2",
+			FileURL:    "",
+			PreviewURL: "/static/upload/2025/9/1756873317505073.png",
+			Model:      "tencent-3d-v2",
+			Status:     types.AI3DJobStatusProcessing,
+			ErrMsg:     "",
+			Params:     `{"prompt":"一个现代建筑模型","image_url":"","enable_pbr":true}`,
+			CreatedAt:  1704070800, // 2024-01-01 01:00:00
+			UpdatedAt:  1704070800, // 2024-01-01 01:00:00
+		},
+		{
+			Id:         3,
+			UserId:     1,
+			Type:       "gitee",
+			Power:      8,
+			TaskId:     "mock_task_3",
+			FileURL:    "",
+			PreviewURL: "",
+			Model:      "gitee-3d-v1",
+			Status:     types.AI3DJobStatusPending,
+			ErrMsg:     "",
+			Params:     `{"prompt":"一辆跑车模型","image_url":"https://example.com/car.jpg","texture":false}`,
+			CreatedAt:  1704074400, // 2024-01-01 02:00:00
+			UpdatedAt:  1704074400, // 2024-01-01 02:00:00
+		},
+		{
+			Id:         4,
+			UserId:     1,
+			Type:       "tencent",
+			Power:      12,
+			TaskId:     "mock_task_4",
+			FileURL:    "",
+			PreviewURL: "",
+			Model:      "tencent-3d-v1",
+			Status:     types.AI3DJobStatusFailed,
+			ErrMsg:     "模型生成失败：输入图片质量不符合要求",
+			Params:     `{"prompt":"一个机器人模型","image_url":"https://example.com/robot.jpg","enable_pbr":false}`,
+			CreatedAt:  1704078000, // 2024-01-01 03:00:00
+			UpdatedAt:  1704078600, // 2024-01-01 03:10:00
+		},
+		{
+			Id:         5,
+			UserId:     1,
+			Type:       "gitee",
+			Power:      20,
+			TaskId:     "mock_task_5",
+			FileURL:    "https://ai.gitee.com/a8c1af8e-26e9-4ca6-aa5c-6d4ba86bfdac",
+			PreviewURL: "https://ai.gitee.com/a8c1af8e-26e9-4ca6-aa5c-6d4ba86bfdac",
+			Model:      "gitee-3d-v2",
+			Status:     types.AI3DJobStatusCompleted,
+			ErrMsg:     "",
+			Params:     `{"prompt":"一个复杂的机械装置","image_url":"","texture":true,"octree_resolution":512}`,
+			CreatedAt:  1704081600, // 2024-01-01 04:00:00
+			UpdatedAt:  1704082200, // 2024-01-01 04:10:00
+		},
+		{
+			Id:         6,
+			UserId:     1,
+			Type:       "tencent",
+			Power:      18,
+			TaskId:     "mock_task_6",
+			FileURL:    "",
+			PreviewURL: "",
+			Model:      "tencent-3d-v2",
+			Status:     types.AI3DJobStatusProcessing,
+			ErrMsg:     "",
+			Params:     `{"prompt":"一个科幻飞船","image_url":"","enable_pbr":true}`,
+			CreatedAt:  1704085200, // 2024-01-01 05:00:00
+			UpdatedAt:  1704085200, // 2024-01-01 05:00:00
+		},
+	}
+
+	// 创建分页响应
+	mockResponse := vo.ThreeDJobList{
+		Page:     1,
+		PageSize: 10,
+		Total:    len(mockJobs),
+		Items:    mockJobs,
+	}
+
+	resp.SUCCESS(c, mockResponse)
+}
--- a/api/service/ai3d/service.go
+++ b/api/service/ai3d/service.go
@@ -249,7 +249,7 @@ func (s *Service) GetJobList(userId uint, page, pageSize int) (*vo.Page, error)
 			Type:       job.Type,
 			Power:      job.Power,
 			TaskId:     job.TaskId,
-			ImgURL:     job.FileURL,
+			FileURL:    job.FileURL,
 			PreviewURL: job.PreviewURL,
 			Model:      job.Model,
 			Status:     job.Status,
--- a/api/service/ai3d/tencent_client.go
+++ b/api/service/ai3d/tencent_client.go
@@ -136,7 +136,7 @@ func (c *Tencent3DClient) QueryJob(jobId string) (*types.AI3DJobResult, error) {
 				if file.PreviewImageUrl != nil {
 					result.PreviewURL = *file.PreviewImageUrl
 				}
-				break // 取第一个文件
+				// TODO 取第一个文件
 			}
 		}
 	case "FAIL":
@@ -153,6 +153,6 @@ func (c *Tencent3DClient) QueryJob(jobId string) (*types.AI3DJobResult, error) {
 // GetSupportedModels 获取支持的模型列表
 func (c *Tencent3DClient) GetSupportedModels() []types.AI3DModel {
 	return []types.AI3DModel{
-		{Name: "Hunyuan3D-3", Power: 500, Formats: []string{"OBJ", "GLB", "STL", "USDZ", "FBX", "MP4"}, Desc: "Hunyuan3D 是腾讯混元团队推出的高质量 3D 生成模型，具备高保真度、细节丰富和高效生成的特点，可快速将文本或图像转换为逼真的 3D 物体。"},
+		{Name: "Hunyuan3D-3", Power: 500, Formats: []string{"GLB", "OBJ", "STL", "USDZ", "FBX", "MP4"}, Desc: "Hunyuan3D 是腾讯混元团队推出的高质量 3D 生成模型，具备高保真度、细节丰富和高效生成的特点，可快速将文本或图像转换为逼真的 3D 物体。"},
 	}
 }
--- a/api/store/vo/ai3d_job.go
+++ b/api/store/vo/ai3d_job.go
@@ -6,7 +6,7 @@ type AI3DJob struct {
 	Type       string `json:"type"`
 	Power      int    `json:"power"`
 	TaskId     string `json:"task_id"`
-	ImgURL     string `json:"img_url"`
+	FileURL    string `json:"file_url"`
 	PreviewURL string `json:"preview_url"`
 	Model      string `json:"model"`
 	Status     string `json:"status"`
@@ -29,4 +29,5 @@ type ThreeDJobList struct {
 	PageSize int       `json:"page_size"`
 	Total    int       `json:"total"`
 	List     []AI3DJob `json:"list"`
+	Items    []AI3DJob `json:"items"`
 }