AI3D 页面功能完成

This commit is contained in:
GeekMaster
2025-09-03 16:00:28 +08:00
parent ead30c8779
commit 54c8856adf
21 changed files with 1795 additions and 1319 deletions

View File

@@ -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

View File

@@ -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)
}
}

View File

@@ -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)
}

View File

@@ -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,

View File

@@ -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 物体。"},
}
}

View File

@@ -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"`
}

View File

@@ -41,7 +41,7 @@
"qrcode": "^1.5.3",
"qs": "^6.11.1",
"sortablejs": "^1.15.0",
"three": "^0.128.0",
"three": "^0.160.0",
"unplugin-auto-import": "^0.18.5",
"vant": "^4.5.0",
"vue": "^3.2.13",

10
web/pnpm-lock.yaml generated
View File

@@ -105,8 +105,8 @@ importers:
specifier: ^1.15.0
version: 1.15.6
three:
specifier: ^0.128.0
version: 0.128.0
specifier: ^0.160.0
version: 0.160.1
unplugin-auto-import:
specifier: ^0.18.5
version: 0.18.6(@vueuse/core@9.13.0(vue@3.5.18))(rollup@4.46.1)
@@ -2240,8 +2240,8 @@ packages:
peerDependencies:
tslib: ^2
three@0.128.0:
resolution: {integrity: sha512-i0ap/E+OaSfzw7bD1TtYnPo3VEplkl70WX5fZqZnfZsE3k3aSFudqrrC9ldFZfYFkn1zwDmBcdGfiIm/hnbyZA==}
three@0.160.1:
resolution: {integrity: sha512-Bgl2wPJypDOZ1stAxwfWAcJ0WQf7QzlptsxkjYiURPz+n5k4RBDLsq+6f9Y75TYxn6aHLcWz+JNmwTOXWrQTBQ==}
tiny-emitter@2.1.0:
resolution: {integrity: sha512-NB6Dk1A9xgQPMoGqC5CVXn123gWyte215ONT5Pp5a0yt4nlEoO1ZWeCwpncaekPHXO60i47ihFnZPiRPjRMq4Q==}
@@ -4549,7 +4549,7 @@ snapshots:
dependencies:
tslib: 2.8.1
three@0.128.0: {}
three@0.160.1: {}
tiny-emitter@2.1.0: {}

View File

@@ -0,0 +1,632 @@
.page-threed {
display: flex;
height: 100vh;
background: #f5f5f5;
}
.params-panel {
width: 400px;
background: white;
border-right: 1px solid #e4e7ed;
padding: 20px;
overflow-y: auto;
}
.platform-tabs {
margin-bottom: 20px;
}
.platform-info {
display: flex;
align-items: center;
gap: 8px;
i {
font-size: 18px;
color: #409eff;
}
}
.params-container {
.param-line {
margin-bottom: 16px;
&.pt {
margin-top: 20px;
}
.label {
display: block;
font-weight: 600;
color: #414141;
}
}
.advanced-toggle-btn {
padding: 0;
font-size: 14px;
color: #409eff;
border: none;
background: none;
display: flex;
align-items: center;
gap: 4px;
&:hover {
color: #66b1ff;
background: #f0f9ff;
border-radius: 4px;
}
}
.advanced-params {
padding: 10px 16px;
border-radius: 8px;
border-left: 4px solid #2196f3;
}
}
.power-display {
display: flex;
align-items: center;
gap: 8px;
.power-value {
font-size: 24px;
font-weight: bold;
color: #409eff;
}
.power-unit {
color: #666;
}
}
.generate-section {
margin-top: 30px;
.generate-btn {
width: 100%;
height: 44px;
font-size: 16px;
}
}
.content-panel {
flex: 1;
padding: 20px;
overflow-y: auto;
}
.task-list {
.list-header {
display: flex;
justify-content: space-between;
align-items: center;
margin-bottom: 30px;
padding: 20px;
background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
border-radius: 12px;
color: white;
h3 {
margin: 0;
color: white;
font-size: 24px;
font-weight: 600;
display: flex;
align-items: center;
&::before {
content: '';
width: 4px;
height: 24px;
background: white;
margin-right: 12px;
border-radius: 2px;
}
}
.el-button {
background: rgba(255, 255, 255, 0.2);
border: 1px solid rgba(255, 255, 255, 0.3);
color: white;
&:hover {
background: rgba(255, 255, 255, 0.3);
border-color: rgba(255, 255, 255, 0.5);
}
}
}
}
.task-items {
.task-card {
background: white;
border-radius: 12px;
padding: 16px;
margin-bottom: 16px;
border: 1px solid #e4e7ed;
box-shadow: 0 2px 8px rgba(0, 0, 0, 0.1);
display: flex;
flex-direction: column;
transition: transform 0.2s ease-in-out, box-shadow 0.2s ease-in-out;
position: relative;
overflow: hidden;
&:hover {
transform: translateY(-5px);
box-shadow: 0 4px 12px rgba(0, 0, 0, 0.15);
}
&.task-card-completed {
border-left: 4px solid #67c23a;
background: #f0f9eb;
}
&.task-card-processing {
border-left: 4px solid #409eff;
background: #ecf5ff;
}
&.task-card-failed {
border-left: 4px solid #f56c6c;
background: #fef0f0;
}
&.task-card-default {
border-left: 4px solid #909399;
background: #f4f4f4;
}
}
.task-card-header {
display: flex;
justify-content: space-between;
align-items: center;
margin-bottom: 12px;
padding-bottom: 12px;
border-bottom: 1px dashed #eee;
.task-info {
display: flex;
align-items: center;
gap: 10px;
.task-id {
font-size: 18px;
font-weight: bold;
color: #333;
display: flex;
align-items: center;
i {
font-size: 20px;
color: #409eff;
}
}
.task-platform {
font-size: 14px;
color: #666;
display: flex;
align-items: center;
i {
font-size: 16px;
color: #409eff;
}
}
}
.task-status-wrapper {
display: flex;
align-items: center;
gap: 10px;
.task-status {
padding: 4px 10px;
border-radius: 6px;
font-size: 13px;
font-weight: bold;
display: flex;
align-items: center;
&.pending {
background: #fffbe6;
color: #e6a23c;
}
&.processing {
background: #e1f3d8;
color: #67c23a;
}
&.completed {
background: #e1f3d8;
color: #67c23a;
}
&.failed {
background: #fef0f0;
color: #f56c6c;
}
i {
font-size: 14px;
margin-right: 4px;
}
}
.task-power {
font-size: 14px;
color: #666;
display: flex;
align-items: center;
i {
font-size: 14px;
margin-right: 4px;
color: #409eff;
}
}
}
}
.task-card-content {
display: flex;
gap: 16px;
margin-bottom: 12px;
.task-preview {
flex: 1;
position: relative;
border-radius: 8px;
overflow: hidden;
background: #f0f0f0;
display: flex;
align-items: center;
justify-content: center;
color: #666;
min-height: 120px;
max-width: 200px;
.preview-image {
width: 100%;
height: 100%;
object-fit: cover;
position: relative;
img {
width: 100%;
height: 100%;
object-fit: cover;
}
.preview-overlay {
position: absolute;
top: 0;
left: 0;
width: 100%;
height: 100%;
background: rgba(0, 0, 0, 0.5);
display: flex;
align-items: center;
justify-content: center;
color: white;
font-size: 24px;
opacity: 0;
transition: opacity 0.3s ease;
&:hover {
opacity: 1;
}
}
}
.input-image {
width: 100%;
height: 100%;
position: relative;
img {
width: 100%;
height: 100%;
object-fit: cover;
}
.input-overlay {
position: absolute;
top: 0;
left: 0;
width: 100%;
height: 100%;
background: rgba(0, 0, 0, 0.5);
display: flex;
align-items: center;
justify-content: center;
color: white;
font-size: 24px;
opacity: 0;
transition: opacity 0.3s ease;
&:hover {
opacity: 1;
}
}
}
.prompt-placeholder {
width: 100%;
height: 100%;
display: flex;
flex-direction: column;
align-items: center;
justify-content: center;
color: #999;
i {
font-size: 48px;
margin-bottom: 12px;
}
span {
font-size: 14px;
}
}
}
.task-details {
flex: 1;
display: flex;
flex-direction: column;
justify-content: space-between;
.task-model {
font-size: 16px;
font-weight: bold;
color: #333;
display: flex;
align-items: center;
i {
font-size: 18px;
margin-right: 6px;
color: #409eff;
}
}
.task-prompt {
font-size: 14px;
color: #666;
display: flex;
align-items: center;
margin-top: 4px;
i {
font-size: 14px;
margin-right: 6px;
color: #909399;
}
}
.task-params {
font-size: 14px;
color: #666;
display: flex;
align-items: center;
margin-top: 4px;
i {
font-size: 14px;
margin-right: 6px;
color: #909399;
}
}
.task-time {
font-size: 12px;
color: #999;
display: flex;
align-items: center;
margin-top: 4px;
i {
font-size: 12px;
margin-right: 6px;
}
}
.task-error {
font-size: 12px;
color: #f56c6c;
display: flex;
align-items: center;
margin-top: 4px;
i {
font-size: 12px;
margin-right: 6px;
}
}
}
}
.task-card-footer {
display: flex;
justify-content: flex-end;
gap: 8px;
padding-top: 12px;
border-top: 1px dashed #eee;
.task-actions {
display: flex;
gap: 8px;
.action-btn {
padding: 6px 12px;
font-size: 13px;
border-radius: 6px;
display: flex;
align-items: center;
justify-content: center;
gap: 4px;
&.preview-btn {
background: #409eff;
color: white;
border: 1px solid #409eff;
&:hover {
background: #66b1ff;
border-color: #66b1ff;
}
}
&.download-btn {
background: #67c23a;
color: white;
border: 1px solid #67c23a;
&:hover {
background: #85ce61;
border-color: #85ce61;
}
}
&.delete-btn {
background: #f56c6c;
color: white;
border: 1px solid #f56c6c;
&:hover {
background: #f78989;
border-color: #f78989;
}
}
&.processing-btn {
background: #909399;
color: white;
border: 1px solid #909399;
cursor: not-allowed;
}
}
}
}
.empty-state {
width: 100%;
height: 200px;
background: #f0f0f0;
border-radius: 12px;
display: flex;
flex-direction: column;
align-items: center;
justify-content: center;
color: #999;
font-size: 14px;
i {
font-size: 48px;
margin-bottom: 12px;
}
}
}
.pagination {
display: flex;
justify-content: center;
margin-top: 20px;
}
.preview-container {
width: 100%;
height: 500px;
min-height: 500px;
background: #f0f0f0;
border-radius: 8px;
position: relative;
.three-container {
width: 100%;
height: 500px;
background: #f0f0f0;
border-radius: 8px;
display: flex;
align-items: center;
justify-content: center;
color: #666;
}
.preview-placeholder {
width: 100%;
height: 500px;
background: #f0f0f0;
border-radius: 8px;
display: flex;
flex-direction: column;
align-items: center;
justify-content: center;
color: #666;
i {
font-size: 48px;
margin-bottom: 12px;
color: #999;
}
p {
margin: 0;
font-size: 14px;
}
}
}
@media (max-width: 768px) {
.page-threed {
flex-direction: column;
}
.params-panel {
width: 100%;
border-right: none;
border-bottom: 1px solid #e4e7ed;
}
.task-card-content {
flex-direction: column;
gap: 12px;
}
.task-preview {
max-width: 100%;
min-height: 150px;
}
.task-card-header {
flex-direction: column;
align-items: flex-start;
gap: 12px;
}
.task-status-wrapper {
align-self: flex-end;
}
}
@media (max-width: 480px) {
.task-card {
padding: 12px;
}
.task-actions {
flex-wrap: wrap;
justify-content: center;
}
.action-btn {
flex: 1;
min-width: 80px;
}
}

View File

@@ -5,26 +5,28 @@
<!-- 控制面板 -->
<div class="control-panel">
<div class="control-group">
<label>旋转速度</label>
<el-slider
v-model="rotationSpeed"
:min="0"
:max="0.1"
:step="0.01"
@change="updateRotationSpeed"
/>
<label>缩放</label>
<div class="scale-controls">
<el-button size="small" @click="zoomOut" :disabled="scale <= 0.1">
<el-icon><Minus /></el-icon>
</el-button>
<span class="scale-value">{{ scale.toFixed(1) }}x</span>
<el-button size="small" @click="zoomIn" :disabled="scale >= 3">
<el-icon><Plus /></el-icon>
</el-button>
</div>
</div>
<div class="control-group">
<label>缩放</label>
<el-slider v-model="scale" :min="0.1" :max="3" :step="0.1" @change="updateScale" />
</div>
<div class="control-buttons">
<el-button size="small" @click="resetCamera">重置视角</el-button>
<el-button size="small" @click="toggleAutoRotate">
{{ autoRotate ? '停止旋转' : '自动旋转' }}
</el-button>
<label>模型颜色</label>
<div class="color-picker">
<el-color-picker
v-model="modelColor"
@change="updateModelColor"
:predefine="predefineColors"
size="small"
/>
</div>
</div>
</div>
@@ -33,6 +35,12 @@
<div class="loading-content">
<el-icon class="is-loading"><Loading /></el-icon>
<p>加载3D模型中...</p>
<div v-if="loadingProgress > 0" class="loading-progress">
<div class="progress-bar">
<div class="progress-fill" :style="{ width: loadingProgress + '%' }"></div>
</div>
<span class="progress-text">{{ loadingProgress.toFixed(1) }}%</span>
</div>
</div>
</div>
@@ -48,20 +56,20 @@
</template>
<script setup>
import { Loading, Warning } from '@element-plus/icons-vue'
import { ElButton, ElIcon, ElSlider } from 'element-plus'
import { Loading, Minus, Plus, Warning } from '@element-plus/icons-vue'
import { ElButton, ElIcon } from 'element-plus'
import * as THREE from 'three'
import { OrbitControls } from 'three/examples/jsm/controls/OrbitControls.js'
import { GLTFLoader } from 'three/examples/jsm/loaders/GLTFLoader.js'
import { OBJLoader } from 'three/examples/jsm/loaders/OBJLoader.js'
import { STLLoader } from 'three/examples/jsm/loaders/STLLoader.js'
import { onMounted, onUnmounted, ref, watch } from 'vue'
import { OrbitControls } from 'three/addons/controls/OrbitControls.js'
import { GLTFLoader } from 'three/addons/loaders/GLTFLoader.js'
import { OBJLoader } from 'three/addons/loaders/OBJLoader.js'
import { STLLoader } from 'three/addons/loaders/STLLoader.js'
import { nextTick, onMounted, onUnmounted, ref, watch } from 'vue'
// Props
const props = defineProps({
modelUrl: {
type: String,
required: true,
required: false,
},
modelType: {
type: String,
@@ -73,33 +81,58 @@ const props = defineProps({
const container = ref(null)
const loading = ref(true)
const error = ref('')
const rotationSpeed = ref(0.02)
const loadingProgress = ref(0)
const scale = ref(1)
const autoRotate = ref(true)
const modelColor = ref('#00ff88')
const predefineColors = ref([
'#00ff88', // 亮绿色
'#ff6b6b', // 亮红色
'#4ecdc4', // 亮青色
'#45b7d1', // 亮蓝色
'#f9ca24', // 亮黄色
'#f0932b', // 亮橙色
'#eb4d4b', // 亮粉红
'#6c5ce7', // 亮紫色
'#a29bfe', // 亮靛蓝
'#fd79a8', // 亮玫瑰
])
// Three.js 相关变量
let scene, camera, renderer, controls, model, mixer, clock
let animationId
let baseScale = 1 // 存储基础缩放值
// 初始化Three.js场景
const initThreeJS = () => {
if (!container.value) return
if (!container.value) {
console.error('ThreeDPreview: 容器元素不存在')
return
}
// 创建场景
scene = new THREE.Scene()
scene.background = new THREE.Color(0xf0f0f0)
scene.background = new THREE.Color(0x2a2a2a) // 深灰色背景,类似截图
// 创建相机
// 获取容器尺寸,确保有最小尺寸
const containerRect = container.value.getBoundingClientRect()
camera = new THREE.PerspectiveCamera(75, containerRect.width / containerRect.height, 0.1, 1000)
camera.position.set(0, 0, 5)
const width = Math.max(containerRect.width || 400, 400)
const height = Math.max(containerRect.height || 300, 300)
// 创建相机 - 参考截图的视角(稍微俯视,从左上角观察)
camera = new THREE.PerspectiveCamera(75, width / height, 0.1, 1000)
camera.position.set(3, 3, 3) // 从左上角俯视角度
// 创建渲染器
renderer = new THREE.WebGLRenderer({ antialias: true })
renderer.setSize(containerRect.width, containerRect.height)
renderer.setPixelRatio(window.devicePixelRatio)
renderer = new THREE.WebGLRenderer({
antialias: true,
alpha: true,
preserveDrawingBuffer: true,
})
renderer.setSize(width, height)
renderer.setPixelRatio(Math.min(window.devicePixelRatio, 2))
renderer.shadowMap.enabled = true
renderer.shadowMap.type = THREE.PCFSoftShadowMap
renderer.outputColorSpace = THREE.SRGBColorSpace
// 添加到容器
container.value.appendChild(renderer.domElement)
@@ -108,8 +141,6 @@ const initThreeJS = () => {
controls = new OrbitControls(camera, renderer.domElement)
controls.enableDamping = true
controls.dampingFactor = 0.05
controls.autoRotate = autoRotate.value
controls.autoRotateSpeed = rotationSpeed.value
// 添加光源
addLights()
@@ -117,6 +148,9 @@ const initThreeJS = () => {
// 添加地面
addGround()
// 添加坐标轴辅助线
addAxesHelper()
// 创建时钟
clock = new THREE.Clock()
@@ -127,31 +161,50 @@ const initThreeJS = () => {
window.addEventListener('resize', onWindowResize)
}
// 添加光源
//
// 添加光源 - 参考截图的柔和光照效果
const addLights = () => {
// 环境光
// 环境光 - 提供基础照明,参考截图的柔和效果
const ambientLight = new THREE.AmbientLight(0x404040, 0.6)
scene.add(ambientLight)
// 方向光
// 方向光 - 从左上角照射,模拟截图中的光照方向
const directionalLight = new THREE.DirectionalLight(0xffffff, 0.8)
directionalLight.position.set(10, 10, 5)
directionalLight.position.set(5, 5, 3)
directionalLight.castShadow = true
directionalLight.shadow.mapSize.width = 2048
directionalLight.shadow.mapSize.height = 2048
directionalLight.shadow.camera.near = 0.5
directionalLight.shadow.camera.far = 50
directionalLight.shadow.camera.left = -10
directionalLight.shadow.camera.right = 10
directionalLight.shadow.camera.top = 10
directionalLight.shadow.camera.bottom = -10
scene.add(directionalLight)
// 点光源
const pointLight = new THREE.PointLight(0xffffff, 0.5)
pointLight.position.set(-10, 10, -5)
scene.add(pointLight)
// 补充光源 - 从右侧照射,提供更均匀的光照
const fillLight = new THREE.DirectionalLight(0xffffff, 0.4)
fillLight.position.set(-3, 3, 3)
scene.add(fillLight)
// 背光 - 增加轮廓,但强度较低
const backLight = new THREE.DirectionalLight(0xffffff, 0.15)
backLight.position.set(0, 2, -5)
scene.add(backLight)
}
// 添加地面
// 添加地面网格 - 参考截图的深色背景和浅色网格线
const addGround = () => {
// 创建网格辅助线 - 使用深色背景配浅色网格线,增加网格密度
const gridHelper = new THREE.GridHelper(20, 40, 0x666666, 0x666666)
gridHelper.position.y = -0.01 // 稍微向下一点避免z-fighting
scene.add(gridHelper)
// 添加半透明地面 - 使用更深的颜色
const groundGeometry = new THREE.PlaneGeometry(20, 20)
const groundMaterial = new THREE.MeshLambertMaterial({
color: 0xcccccc,
color: 0x1a1a1a, // 更深的背景色
transparent: true,
opacity: 0.3,
})
@@ -161,12 +214,26 @@ const addGround = () => {
scene.add(ground)
}
// 添加坐标轴辅助线 - 参考截图的样式
const addAxesHelper = () => {
const axesHelper = new THREE.AxesHelper(3) // 稍微小一点的坐标轴
scene.add(axesHelper)
}
//
//
// 加载3D模型
const loadModel = async () => {
if (!props.modelUrl) return
if (!props.modelUrl) {
console.warn('ThreeDPreview: 没有提供模型URL')
return
}
try {
loading.value = true
loadingProgress.value = 0
error.value = ''
// 清除现有模型
@@ -196,24 +263,69 @@ const loadModel = async () => {
model = loadedModel
scene.add(model)
// 调整模型位置和大小
centerModel()
fitCameraToModel()
// 计算模型边界并调整相机位置
const box = new THREE.Box3().setFromObject(model)
const size = box.getSize(new THREE.Vector3())
const center = box.getCenter(new THREE.Vector3())
// 设置阴影
// 调整模型位置到原点
model.position.sub(center)
// 计算并保存基础缩放值
const maxDim = Math.max(size.x, size.y, size.z)
baseScale = maxDim > 0 ? 2 / maxDim : 1
// 应用初始缩放
model.scale.setScalar(baseScale * scale.value)
// 根据模型大小调整相机距离 - 保持截图中的俯视角度
const cameraDistance = maxDim > 0 ? maxDim * 2 : 5
// 设置相机位置为左上角俯视角度
camera.position.set(cameraDistance * 0.6, cameraDistance * 0.6, cameraDistance * 0.6)
camera.lookAt(0, 0, 0)
if (controls) {
controls.target.set(0, 0, 0)
controls.update()
}
// 设置阴影和材质
model.traverse((child) => {
if (child.isMesh) {
child.castShadow = true
child.receiveShadow = true
// 将模型材质改为亮色
if (child.material) {
const colorHex = modelColor.value.replace('#', '0x')
// 如果是数组材质
if (Array.isArray(child.material)) {
child.material.forEach((mat) => {
if (mat.color) {
mat.color.setHex(colorHex)
}
})
} else {
// 单个材质
if (child.material.color) {
child.material.color.setHex(colorHex)
}
}
}
}
})
} else {
console.warn('ThreeDPreview: 模型加载返回空值')
}
loading.value = false
loadingProgress.value = 100
} catch (err) {
console.error('加载3D模型失败:', err)
console.error('ThreeDPreview: 加载3D模型失败:', err)
error.value = `加载模型失败: ${err.message}`
loading.value = false
loadingProgress.value = 0
}
}
@@ -235,8 +347,16 @@ const loadGLTF = (url) => {
resolve(model)
},
undefined,
reject
(xhr) => {
if (xhr.total > 0) {
const percent = (xhr.loaded / xhr.total) * 100
loadingProgress.value = percent
}
},
(error) => {
console.error('ThreeDPreview: GLTF模型加载失败', error)
reject(error)
}
)
})
}
@@ -283,74 +403,52 @@ const loadSTL = (url) => {
})
}
// 居中模型
const centerModel = () => {
if (!model) return
const box = new THREE.Box3().setFromObject(model)
const center = box.getCenter(new THREE.Vector3())
const size = box.getSize(new THREE.Vector3())
// 居中
model.position.sub(center)
// 调整缩放
const maxDim = Math.max(size.x, size.y, size.z)
const scale = 2 / maxDim
model.scale.setScalar(scale * props.scale)
// 放大
const zoomIn = () => {
if (scale.value < 3) {
scale.value = Math.min(scale.value + 0.1, 3)
updateScale(scale.value)
}
}
// 调整相机以适应模型
const fitCameraToModel = () => {
if (!model) return
const box = new THREE.Box3().setFromObject(model)
const size = box.getSize(new THREE.Vector3())
const center = box.getCenter(new THREE.Vector3())
const maxDim = Math.max(size.x, size.y, size.z)
const fov = camera.fov * (Math.PI / 180)
let cameraZ = Math.abs(maxDim / 2 / Math.tan(fov / 2))
camera.position.set(center.x, center.y, center.z + cameraZ)
camera.lookAt(center)
controls.target.copy(center)
controls.update()
}
// 更新旋转速度
const updateRotationSpeed = (value) => {
if (controls) {
controls.autoRotateSpeed = value
// 缩小
const zoomOut = () => {
if (scale.value > 0.1) {
scale.value = Math.max(scale.value - 0.1, 0.1)
updateScale(scale.value)
}
}
// 更新缩放
const updateScale = (value) => {
if (model) {
const box = new THREE.Box3().setFromObject(model)
const size = box.getSize(new THREE.Vector3())
const maxDim = Math.max(size.x, size.y, size.z)
const baseScale = 2 / maxDim
model.scale.setScalar(baseScale * value)
console.log('ThreeDPreview: 更新缩放', { value, baseScale, finalScale: baseScale * value })
}
}
// 重置相机
const resetCamera = () => {
if (camera && model) {
fitCameraToModel()
// 更新模型颜色
const updateModelColor = (color) => {
if (model && color) {
model.traverse((child) => {
if (child.isMesh && child.material) {
if (Array.isArray(child.material)) {
child.material.forEach((mat) => {
if (mat.color) {
mat.color.setHex(color.replace('#', '0x'))
}
})
} else {
if (child.material.color) {
child.material.color.setHex(color.replace('#', '0x'))
}
}
}
})
}
}
// 切换自动旋转
const toggleAutoRotate = () => {
autoRotate.value = !autoRotate.value
if (controls) {
controls.autoRotate = autoRotate.value
}
}
//
// 重试加载
const retryLoad = () => {
@@ -362,9 +460,12 @@ const onWindowResize = () => {
if (!container.value || !camera || !renderer) return
const containerRect = container.value.getBoundingClientRect()
camera.aspect = containerRect.width / containerRect.height
const width = Math.max(containerRect.width || 400, 400)
const height = Math.max(containerRect.height || 300, 300)
camera.aspect = width / height
camera.updateProjectionMatrix()
renderer.setSize(containerRect.width, containerRect.height)
renderer.setSize(width, height)
}
// 渲染循环
@@ -429,10 +530,16 @@ watch(
// 生命周期
onMounted(() => {
initThreeJS()
if (props.modelUrl) {
loadModel()
}
// 使用nextTick确保DOM完全渲染
nextTick(() => {
// 延迟初始化,确保容器有正确的尺寸
setTimeout(() => {
initThreeJS()
if (props.modelUrl) {
loadModel()
}
}, 100)
})
})
onUnmounted(() => {
@@ -440,7 +547,7 @@ onUnmounted(() => {
})
</script>
<style lang="scss" scoped>
<style lang="scss">
.three-d-preview {
position: relative;
width: 100%;
@@ -450,7 +557,11 @@ onUnmounted(() => {
.preview-container {
width: 100%;
height: 100%;
min-height: 400px;
position: relative;
background: #f0f0f0;
border-radius: 8px;
overflow: hidden;
}
.control-panel {
@@ -475,12 +586,27 @@ onUnmounted(() => {
}
}
.control-buttons {
.color-picker {
display: flex;
justify-content: center;
.el-color-picker--small {
width: 100% !important;
}
}
.scale-controls {
display: flex;
align-items: center;
justify-content: center;
gap: 8px;
.el-button {
flex: 1;
.scale-value {
min-width: 40px;
text-align: center;
font-size: 14px;
color: #333;
font-weight: 500;
}
}
}
@@ -519,6 +645,32 @@ onUnmounted(() => {
}
}
.loading-progress {
width: 200px;
margin-top: 16px;
.progress-bar {
width: 100%;
height: 4px;
background: rgba(0, 0, 0, 0.1);
border-radius: 2px;
overflow: hidden;
margin-bottom: 8px;
.progress-fill {
height: 100%;
background: #409eff;
border-radius: 2px;
transition: width 0.3s ease;
}
}
.progress-text {
font-size: 12px;
color: #666;
}
}
.error-content {
.el-icon {
color: #f56c6c;

View File

@@ -0,0 +1,237 @@
<template>
<div class="custom-switch" :class="{ 'is-active': modelValue }" @click="toggleSwitch">
<div class="switch-track" :style="trackStyle" ref="trackRef">
<div class="switch-thumb" :style="thumbStyle"></div>
<div class="switch-text inactive-text">
<slot name="inactive-text"></slot>
</div>
<div class="switch-text active-text">
<slot name="active-text"></slot>
</div>
</div>
</div>
</template>
<script setup>
import { computed, defineEmits, defineProps, nextTick, onMounted, ref } from 'vue'
const props = defineProps({
modelValue: {
type: Boolean,
default: false,
},
disabled: {
type: Boolean,
default: false,
},
activeColor: {
type: String,
default: '#67c23a', // 默认绿色
},
inactiveColor: {
type: String,
default: '#f56c6c', // 默认红色
},
width: {
type: [String, Number],
default: null, // 默认不设置固定宽度使用min-width
},
size: {
type: String,
default: 'default', // small, default, large
validator: (value) => ['small', 'default', 'large'].includes(value),
},
})
const emit = defineEmits(['update:modelValue', 'change'])
// 轨道宽度引用
const trackRef = ref(null)
const trackWidth = ref(120) // 默认宽度
// 获取轨道实际宽度
const updateTrackWidth = () => {
if (trackRef.value) {
trackWidth.value = trackRef.value.offsetWidth
}
}
// 计算尺寸相关样式
const sizeConfig = computed(() => {
const configs = {
small: {
height: 24,
thumbSize: 20,
thumbMargin: 2,
fontSize: 12,
padding: 12,
},
default: {
height: 28,
thumbSize: 24,
thumbMargin: 2,
fontSize: 13,
padding: 14,
},
large: {
height: 32,
thumbSize: 28,
thumbMargin: 2,
fontSize: 14,
padding: 16,
},
}
return configs[props.size]
})
// 计算轨道样式
const trackStyle = computed(() => {
const backgroundColor = props.modelValue ? props.activeColor : props.inactiveColor
const config = sizeConfig.value
const style = {
backgroundColor: backgroundColor,
height: `${config.height}px`,
padding: `0 ${config.padding}px`,
}
// 如果传入了width属性则设置固定宽度
if (props.width !== null) {
const widthValue = typeof props.width === 'number' ? `${props.width}px` : props.width
style.width = widthValue
style.minWidth = widthValue
}
return style
})
// 计算滑块样式 - 使用像素值而不是calc()
const thumbStyle = computed(() => {
const config = sizeConfig.value
const thumbWidth = config.thumbSize
const thumbMargin = config.thumbMargin
const maxTranslateX = trackWidth.value - thumbWidth - thumbMargin * 2
const transform = props.modelValue ? `translateX(${maxTranslateX}px)` : 'translateX(0px)'
console.log('Track width:', trackWidth.value, 'Transform:', transform)
return {
transform: transform,
width: `${thumbWidth}px`,
height: `${thumbWidth}px`,
top: `${thumbMargin}px`,
left: `${thumbMargin}px`,
}
})
const toggleSwitch = () => {
if (props.disabled) return
const newValue = !props.modelValue
emit('update:modelValue', newValue)
emit('change', newValue)
// 切换后更新宽度
nextTick(() => {
updateTrackWidth()
})
}
// 组件挂载后获取轨道宽度
onMounted(() => {
nextTick(() => {
updateTrackWidth()
})
})
</script>
<style scoped lang="scss">
.custom-switch {
display: inline-block;
cursor: pointer;
user-select: none;
&.is-active {
.switch-track {
.inactive-text {
opacity: 0;
transform: translateX(-20px);
}
.active-text {
opacity: 1;
transform: translateX(0);
}
}
}
&:not(.is-active) {
.switch-track {
.inactive-text {
opacity: 1;
transform: translateX(0);
}
.active-text {
opacity: 0;
transform: translateX(20px);
}
}
}
&:disabled {
cursor: not-allowed;
opacity: 0.6;
}
}
.switch-track {
position: relative;
min-width: 120px; // 最小宽度
border-radius: 16px;
transition: background-color 0.3s ease;
overflow: hidden;
display: flex;
align-items: center;
justify-content: space-between;
}
.switch-thumb {
position: absolute;
background-color: white;
border-radius: 50%;
transition: transform 0.3s ease;
z-index: 2;
box-shadow: 0 2px 4px rgba(0, 0, 0, 0.2);
}
.switch-text {
position: relative; // 改为相对定位
color: white;
font-weight: 500;
white-space: nowrap;
transition: all 0.3s ease;
z-index: 1;
flex: 1; // 让文字占据剩余空间
text-align: center; // 文字居中
}
.inactive-text {
opacity: 1;
transform: translateX(0);
}
.active-text {
opacity: 0;
transform: translateX(20px);
position: absolute; // 绝对定位,避免影响布局
left: 0;
right: 0;
}
// 响应式设计
@media (max-width: 768px) {
.switch-track {
min-width: 100px;
}
}
</style>

View File

@@ -98,6 +98,12 @@ const routes = [
meta: { title: 'AI3D模型生成' },
component: () => import('@/views/AIThreeDCreate.vue'),
},
{
name: 'test3d',
path: '/test3d',
meta: { title: '3D预览测试' },
component: () => import('@/views/test/Test3D.vue'),
},
{
name: 'ExternalLink',
path: '/external',
@@ -476,7 +482,7 @@ const routes = [
path: '/mobile/3d',
name: 'mobile-3d',
meta: { title: '3D模型生成' },
component: () => import('@/views/mobile/ThreeDCreate.vue'),
component: () => import('@/views/AIThreeDCreate.vue'),
},
],
},
@@ -485,7 +491,7 @@ const routes = [
name: 'test',
path: '/test',
meta: { title: '测试页面' },
component: () => import('@/views/Test.vue'),
component: () => import('@/views/test/Test.vue'),
},
{

View File

@@ -12,7 +12,7 @@ import { httpDownload, httpGet, httpPost } from '@/utils/http'
import { replaceImg, substr } from '@/utils/libs'
import { ElMessageBox } from 'element-plus'
import { defineStore } from 'pinia'
import { computed, nextTick, reactive, ref } from 'vue'
import { computed, reactive, ref } from 'vue'
export const useJimengStore = defineStore('jimeng', () => {
// 当前激活的功能分类和具体功能
@@ -431,7 +431,7 @@ export const useJimengStore = defineStore('jimeng', () => {
const downloadFile = async (item) => {
const url = replaceImg(item.video_url || item.img_url)
const downloadURL = `${import.meta.env.VITE_API_HOST}/api/download?url=${url}`
const downloadURL = `/api/download?url=${url}`
const urlObj = new URL(url)
const fileName = urlObj.pathname.split('/').pop()

View File

@@ -352,7 +352,7 @@ export const useJimengStore = defineStore('mobile-jimeng', () => {
const downloadFile = async (item) => {
const url = replaceImg(item.video_url || item.img_url)
const downloadURL = `${import.meta.env.VITE_API_HOST}/api/download?url=${url}`
const downloadURL = `/api/download?url=${url}`
const urlObj = new URL(url)
const fileName = urlObj.pathname.split('/').pop()

View File

@@ -262,7 +262,7 @@ export const useSunoStore = defineStore('suno', () => {
}
const download = (item) => {
const url = replaceImg(item.audio_url)
const downloadURL = `${import.meta.env.VITE_API_HOST}/api/download?url=${url}`
const downloadURL = `/api/download?url=${url}`
const urlObj = new URL(url)
const fileName = urlObj.pathname.split('/').pop()
item.downloading = true

View File

@@ -179,7 +179,7 @@ export const useSunoStore = defineStore('suno', () => {
const download = async (item) => {
const url = replaceImg(item.audio_url)
const downloadURL = `${import.meta.env.VITE_API_HOST}/api/download?url=${url}`
const downloadURL = `/api/download?url=${url}`
const urlObj = new URL(url)
const fileName = urlObj.pathname.split('/').pop()

View File

@@ -464,7 +464,7 @@ export const useVideoStore = defineStore('video', () => {
// 视频下载
const downloadVideo = async (item) => {
const url = replaceImg(item.video_url)
const downloadURL = `${import.meta.env.VITE_API_HOST}/api/download?url=${url}`
const downloadURL = `/api/download?url=${url}`
const urlObj = new URL(url)
const fileName = urlObj.pathname.split('/').pop()

File diff suppressed because it is too large Load Diff

View File

@@ -1,765 +0,0 @@
<template>
<div class="mobile-threed-create">
<!-- 顶部导航 -->
<div class="top-nav">
<div class="nav-left" @click="$router.go(-1)">
<i class="iconfont icon-arrow-left"></i>
</div>
<div class="nav-title">3D模型生成</div>
<div class="nav-right"></div>
</div>
<!-- 平台选择 -->
<div class="platform-selector">
<div class="selector-tabs">
<div
v-for="platform in platforms"
:key="platform.key"
:class="['selector-tab', { active: activePlatform === platform.key }]"
@click="activePlatform = platform.key"
>
<div class="tab-icon">
<i :class="platform.icon"></i>
</div>
<div class="tab-name">{{ platform.name }}</div>
</div>
</div>
</div>
<!-- 参数设置 -->
<div class="params-section">
<!-- 图片上传 -->
<div class="param-group">
<div class="param-label">上传图片</div>
<div class="image-upload-area">
<ImageUpload
v-model="currentImage"
:max-count="1"
:multiple="false"
@change="handleImageChange"
/>
</div>
</div>
<!-- 提示词输入 -->
<div class="param-group">
<div class="param-label">提示词描述</div>
<div class="prompt-input">
<el-input
v-model="currentPrompt"
type="textarea"
:rows="4"
placeholder="请输入3D模型描述越详细越好"
maxlength="2000"
show-word-limit
/>
</div>
</div>
<!-- 模型选择 -->
<div class="param-group">
<div class="param-label">输出格式</div>
<div class="model-selector">
<div
v-for="(model, key) in availableModels"
:key="key"
:class="['model-option', { active: selectedModel === key }]"
@click="selectedModel = key"
>
<div class="model-name">{{ model.name }}</div>
<div class="model-power">{{ model.power }}</div>
</div>
</div>
</div>
<!-- 算力消耗 -->
<div class="power-info">
<div class="power-label">算力消耗</div>
<div class="power-value">{{ currentPower }} </div>
</div>
</div>
<!-- 生成按钮 -->
<div class="generate-section">
<el-button
type="primary"
size="large"
:loading="generating"
:disabled="!canGenerate"
@click="generate3D"
class="generate-btn"
>
{{ generating ? '生成中...' : '开始生成' }}
</el-button>
</div>
<!-- 任务列表 -->
<div class="task-section">
<div class="section-header">
<h3>生成任务</h3>
<el-button size="small" @click="refreshTasks">刷新</el-button>
</div>
<div class="task-list">
<div
v-for="task in taskList"
:key="task.id"
class="task-item"
:class="{ completed: task.status === 'completed' }"
>
<div class="task-main">
<div class="task-info">
<div class="task-id">#{{ task.id }}</div>
<div class="task-status" :class="task.status">
{{ getStatusText(task.status) }}
</div>
</div>
<div class="task-prompt">
{{ task.params ? getPromptFromParams(task.params) : '' }}
</div>
<div class="task-progress" v-if="task.status === 'processing'">
<el-progress :percentage="task.progress" :stroke-width="6" />
</div>
</div>
<div class="task-actions">
<template v-if="task.status === 'completed'">
<el-button size="small" @click="preview3D(task)">预览</el-button>
<el-button size="small" type="primary" @click="download3D(task)">下载</el-button>
</template>
<template v-else>
<el-button size="small" @click="deleteTask(task.id)">删除</el-button>
</template>
</div>
</div>
</div>
<!-- 加载更多 -->
<div class="load-more" v-if="hasMore">
<el-button size="small" @click="loadMoreTasks">加载更多</el-button>
</div>
</div>
<!-- 3D预览弹窗 -->
<el-dialog
v-model="previewVisible"
title="3D模型预览"
width="90%"
:before-close="closePreview"
class="mobile-dialog"
>
<div class="preview-container">
<div id="three-container" class="three-container">
<div class="preview-placeholder">
<i class="iconfont icon-3d"></i>
<p>3D模型预览</p>
</div>
</div>
</div>
<template #footer>
<div class="dialog-footer">
<el-button @click="closePreview">关闭</el-button>
<el-button type="primary" @click="downloadCurrentModel">下载模型</el-button>
</div>
</template>
</el-dialog>
</div>
</template>
<script setup>
import ImageUpload from '@/components/ImageUpload.vue'
import { httpGet, httpPost } from '@/utils/http'
import { ElMessage, ElMessageBox } from 'element-plus'
import { computed, nextTick, onMounted, ref } from 'vue'
// 响应式数据
const activePlatform = ref('gitee')
const currentImage = ref([])
const currentPrompt = ref('')
const selectedModel = ref('obj')
const generating = ref(false)
const previewVisible = ref(false)
const currentPage = ref(1)
const pageSize = ref(10)
const total = ref(0)
const taskList = ref([])
const currentPreviewTask = ref(null)
const hasMore = ref(true)
// 平台配置
const platforms = [
{
key: 'gitee',
name: '魔力方舟',
icon: 'icon-gitee',
},
{
key: 'tencent',
name: '腾讯混元',
icon: 'icon-tencent',
},
]
const platformConfig = {
gitee: {
name: '魔力方舟',
models: {
obj: { name: 'OBJ格式', power: 45 },
glb: { name: 'GLB格式', power: 55 },
stl: { name: 'STL格式', power: 35 },
usdz: { name: 'USDZ格式', power: 65 },
fbx: { name: 'FBX格式', power: 75 },
mp4: { name: 'MP4格式', power: 85 },
},
},
tencent: {
name: '腾讯混元',
models: {
obj: { name: 'OBJ格式', power: 50 },
glb: { name: 'GLB格式', power: 60 },
stl: { name: 'STL格式', power: 40 },
usdz: { name: 'USDZ格式', power: 70 },
fbx: { name: 'FBX格式', power: 80 },
mp4: { name: 'MP4格式', power: 90 },
},
},
}
// 计算属性
const availableModels = computed(() => {
return platformConfig[activePlatform.value]?.models || {}
})
const currentPower = computed(() => {
return availableModels.value[selectedModel.value]?.power || 0
})
const canGenerate = computed(() => {
return currentPrompt.value.trim() && currentImage.value.length > 0 && selectedModel.value
})
// 方法
const handleImageChange = (files) => {
currentImage.value = files
}
const generate3D = async () => {
if (!canGenerate.value) {
ElMessage.warning('请完善生成参数')
return
}
try {
generating.value = true
const requestData = {
type: activePlatform.value,
model: selectedModel.value,
prompt: currentPrompt.value,
image_url: currentImage.value[0]?.url || '',
power: currentPower.value,
}
const response = await httpPost('/api/3d/generate', requestData)
if (response.code === 0) {
ElMessage.success('任务创建成功')
// 清空表单
currentImage.value = []
currentPrompt.value = ''
// 刷新任务列表
loadTasks(true)
} else {
ElMessage.error(response.message || '创建任务失败')
}
} catch (error) {
ElMessage.error('创建任务失败:' + error.message)
} finally {
generating.value = false
}
}
const loadTasks = async (reset = false) => {
try {
if (reset) {
currentPage.value = 1
taskList.value = []
}
const response = await httpGet('/api/3d/jobs', {
page: currentPage.value,
page_size: pageSize.value,
})
if (response.code === 0) {
if (reset) {
taskList.value = response.data.list
} else {
taskList.value.push(...response.data.list)
}
total.value = response.data.total
hasMore.value = taskList.value.length < total.value
}
} catch (error) {
ElMessage.error('加载任务列表失败:' + error.message)
}
}
const refreshTasks = () => {
loadTasks(true)
}
const loadMoreTasks = () => {
if (hasMore.value) {
currentPage.value++
loadTasks()
}
}
const deleteTask = async (taskId) => {
try {
await ElMessageBox.confirm('确定要删除这个任务吗?', '提示', {
confirmButtonText: '确定',
cancelButtonText: '取消',
type: 'warning',
})
const response = await httpGet(`/api/3d/job/${taskId}/delete`)
if (response.code === 0) {
ElMessage.success('删除成功')
loadTasks(true)
} else {
ElMessage.error(response.message || '删除失败')
}
} catch (error) {
if (error !== 'cancel') {
ElMessage.error('删除失败:' + error.message)
}
}
}
const preview3D = (task) => {
currentPreviewTask.value = task
previewVisible.value = true
nextTick(() => {
initThreeJS(task)
})
}
const closePreview = () => {
previewVisible.value = false
currentPreviewTask.value = null
}
const download3D = async (task) => {
if (!task.img_url) {
ElMessage.warning('模型文件不存在')
return
}
try {
// 创建一个隐藏的a标签来下载文件
const link = document.createElement('a')
link.href = task.img_url
link.download = `3d_model_${task.id}.${task.model}`
link.style.display = 'none'
document.body.appendChild(link)
link.click()
document.body.removeChild(link)
ElMessage.success('开始下载3D模型')
} catch (error) {
console.error('下载失败:', error)
ElMessage.error('下载失败,请重试')
}
}
const downloadCurrentModel = () => {
if (currentPreviewTask.value) {
download3D(currentPreviewTask.value)
}
}
const getStatusText = (status) => {
const statusMap = {
pending: '等待中',
processing: '处理中',
completed: '已完成',
failed: '失败',
}
return statusMap[status] || status
}
const getPromptFromParams = (paramsStr) => {
try {
const params = JSON.parse(paramsStr)
return params.prompt || ''
} catch {
return ''
}
}
// Three.js 初始化
const initThreeJS = (task) => {
// TODO: 实现Three.js 3D模型预览
console.log('初始化Three.js预览:', task)
}
// 生命周期
onMounted(() => {
loadTasks(true)
})
</script>
<style lang="scss" scoped>
.mobile-threed-create {
min-height: 100vh;
background: #f5f5f5;
padding-bottom: 20px;
}
.top-nav {
display: flex;
align-items: center;
justify-content: space-between;
padding: 16px 20px;
background: white;
border-bottom: 1px solid #e4e7ed;
position: sticky;
top: 0;
z-index: 100;
.nav-left {
width: 24px;
height: 24px;
display: flex;
align-items: center;
justify-content: center;
cursor: pointer;
i {
font-size: 20px;
color: #333;
}
}
.nav-title {
font-size: 18px;
font-weight: 500;
color: #333;
}
.nav-right {
width: 24px;
}
}
.platform-selector {
background: white;
margin: 16px 20px;
border-radius: 12px;
padding: 20px;
.selector-tabs {
display: flex;
gap: 12px;
.selector-tab {
flex: 1;
display: flex;
flex-direction: column;
align-items: center;
gap: 8px;
padding: 16px 12px;
border-radius: 8px;
border: 2px solid #e4e7ed;
cursor: pointer;
transition: all 0.3s;
&.active {
border-color: #409eff;
background: #ecf5ff;
}
.tab-icon {
width: 32px;
height: 32px;
display: flex;
align-items: center;
justify-content: center;
i {
font-size: 24px;
color: #409eff;
}
}
.tab-name {
font-size: 14px;
color: #333;
text-align: center;
}
}
}
}
.params-section {
background: white;
margin: 16px 20px;
border-radius: 12px;
padding: 20px;
.param-group {
margin-bottom: 24px;
.param-label {
font-size: 16px;
font-weight: 500;
color: #333;
margin-bottom: 12px;
}
.image-upload-area {
border: 2px dashed #d9d9d9;
border-radius: 8px;
padding: 20px;
text-align: center;
transition: border-color 0.3s;
&:hover {
border-color: #409eff;
}
}
.prompt-input {
.el-textarea {
.el-textarea__inner {
border-radius: 8px;
border: 1px solid #d9d9d9;
}
}
}
.model-selector {
display: grid;
grid-template-columns: repeat(2, 1fr);
gap: 12px;
.model-option {
padding: 16px 12px;
border: 2px solid #e4e7ed;
border-radius: 8px;
text-align: center;
cursor: pointer;
transition: all 0.3s;
&.active {
border-color: #409eff;
background: #ecf5ff;
}
.model-name {
font-size: 14px;
color: #333;
margin-bottom: 4px;
}
.model-power {
font-size: 12px;
color: #409eff;
font-weight: 500;
}
}
}
}
}
.power-info {
display: flex;
justify-content: space-between;
align-items: center;
padding: 16px;
background: #f0f9ff;
border-radius: 8px;
border: 1px solid #b3d8ff;
.power-label {
font-size: 16px;
color: #333;
}
.power-value {
font-size: 20px;
font-weight: bold;
color: #409eff;
}
}
.generate-section {
margin: 16px 20px;
.generate-btn {
width: 100%;
height: 48px;
font-size: 16px;
border-radius: 12px;
}
}
.task-section {
background: white;
margin: 16px 20px;
border-radius: 12px;
padding: 20px;
.section-header {
display: flex;
justify-content: space-between;
align-items: center;
margin-bottom: 20px;
h3 {
margin: 0;
font-size: 18px;
color: #333;
}
}
}
.task-list {
.task-item {
border: 1px solid #e4e7ed;
border-radius: 8px;
padding: 16px;
margin-bottom: 16px;
&.completed {
border-color: #67c23a;
background: #f0f9ff;
}
.task-main {
margin-bottom: 16px;
.task-info {
display: flex;
justify-content: space-between;
align-items: center;
margin-bottom: 12px;
.task-id {
font-weight: 500;
color: #666;
}
.task-status {
padding: 4px 8px;
border-radius: 4px;
font-size: 12px;
&.pending {
background: #fdf6ec;
color: #e6a23c;
}
&.processing {
background: #ecf5ff;
color: #409eff;
}
&.completed {
background: #f0f9ff;
color: #67c23a;
}
&.failed {
background: #fef0f0;
color: #f56c6c;
}
}
}
.task-prompt {
color: #666;
margin-bottom: 12px;
line-height: 1.4;
font-size: 14px;
}
}
.task-actions {
display: flex;
gap: 8px;
.el-button {
flex: 1;
}
}
}
}
.load-more {
text-align: center;
margin-top: 20px;
}
.preview-container {
.three-container {
width: 100%;
height: 300px;
background: #f0f0f0;
border-radius: 8px;
display: flex;
align-items: center;
justify-content: center;
.preview-placeholder {
text-align: center;
color: #666;
i {
font-size: 48px;
margin-bottom: 12px;
display: block;
}
p {
margin: 0;
font-size: 14px;
}
}
}
}
// 移动端弹窗样式
.mobile-dialog {
:deep(.el-dialog) {
margin: 20px;
border-radius: 12px;
}
:deep(.el-dialog__header) {
padding: 20px 20px 0;
}
:deep(.el-dialog__body) {
padding: 20px;
}
:deep(.el-dialog__footer) {
padding: 0 20px 20px;
.dialog-footer {
display: flex;
gap: 12px;
.el-button {
flex: 1;
}
}
}
}
</style>

View File

@@ -0,0 +1,66 @@
<template>
<div class="test-3d">
<h1>3D预览功能测试</h1>
<div class="test-container">
<h2>测试1: 默认立方体无模型URL</h2>
<div class="preview-wrapper">
<ThreeDPreview />
</div>
</div>
<div class="test-container">
<h2>测试2: 带模型URL如果有的话</h2>
<div class="preview-wrapper">
<ThreeDPreview v-if="testModelUrl" :model-url="testModelUrl" :model-type="testModelType" />
<div v-else class="no-model">
<p>没有测试模型URL</p>
</div>
</div>
</div>
</div>
</template>
<script setup>
import ThreeDPreview from '@/components/ThreeDPreview.vue'
import { ref } from 'vue'
// 测试用的模型URL可以替换为实际的模型文件
const testModelUrl = ref('https://img.r9it.com/R03TQZ7PZ386RGL7PTMNGFOHAJW15WYF.glb')
const testModelType = ref('glb')
</script>
<style lang="scss" scoped>
.test-3d {
padding: 20px;
max-width: 1200px;
margin: 0 auto;
}
.test-container {
margin-bottom: 40px;
h2 {
margin-bottom: 16px;
color: #333;
}
}
.preview-wrapper {
width: 100%;
height: 500px;
border: 2px solid #ddd;
border-radius: 8px;
overflow: hidden;
}
.no-model {
width: 100%;
height: 100%;
display: flex;
align-items: center;
justify-content: center;
background: #f5f5f5;
color: #666;
}
</style>