Merge tag 'v4.2.7'

.gitattributes

@@ -0,0 +1,5 @@
+web/src/assets/fonts filter=lfs diff=lfs merge=lfs -text
+web/src/assets/iconfont filter=lfs diff=lfs merge=lfs -text
+web/src/assets/img filter=lfs diff=lfs merge=lfs -text
+web/public filter=lfs diff=lfs merge=lfs -text
+api/res filter=lfs diff=lfs merge=lfs -text

CHANGELOG.md

@@ -1,5 +1,13 @@
 # 更新日志
 
+## v4.2.7
+
+- Bug 修复：修复超级管理员无法修改密码的 Bug
+- Bug 修复：微信登录配置更新后，没有同步更新到系统配置
+- 功能优化： 给 AI 对话 API 加上线程锁，确保同一个用户同时只有一个对话请求
+- 功能新增：支持即梦 AI 4.0 图片编辑，即梦 AI 数字人，动作迁移功能。🔥🔥🔥
+- 功能新增：新增 AI 对话编辑功能，并优化了重新生成逻辑
+
 ## v4.2.6
 
 - 功能重构：优化系统配置管理功能，把 OSS，支付，短信，邮件等配置全部迁移到管理后台，无需通过修改配置文档的方式修改 🎉🎉🎉

api/core/types/config.go

@@ -89,6 +89,7 @@ type BaseConfig struct {
 	MjMode      string `json:"mj_mode"`       // midjourney 默认的API模式，relax, fast, turbo
 
 	IndexNavs []int  `json:"index_navs"` // 首页显示的导航菜单
+	IndexPage string `json:"index_page"` // 首页显示的页面
 	Copyright string `json:"copyright"`  // 版权信息
 	ICP       string `json:"icp"`        // ICP 备案号
 	GaBeian   string `json:"ga_beian"`   // 公安备案号

api/core/types/geekai.go

@@ -20,14 +20,14 @@ func init() {
 
 // CaptchaConfig 行为验证码配置
 type CaptchaConfig struct {
-	ApiKey  string `json:"api_key"`
+	ApiKey  string `json:"api_key,omitempty"`
-	Type    string `json:"type"` // 验证码类型, 可选值: "dot" 或 "slide"
+	Type    string `json:"type,omitempty"` // 验证码类型, 可选值: "dot" 或 "slide"
-	Enabled bool   `json:"enabled"`
+	Enabled bool   `json:"enabled,omitempty"`
 }
 
 // WxLoginConfig 微信登录配置
 type WxLoginConfig struct {
-	ApiKey    string `json:"api_key"`
+	ApiKey    string `json:"api_key,omitempty"`
-	NotifyURL string `json:"notify_url"` // 登录成功回调 URL
+	NotifyURL string `json:"notify_url,omitempty"` // 登录成功回调 URL
-	Enabled   bool   `json:"enabled"`    // 是否启用微信登录
+	Enabled   bool   `json:"enabled,omitempty"`    // 是否启用微信登录
 }

api/core/types/jimeng.go

@@ -2,17 +2,64 @@ package types
 
 // JimengConfig 即梦AI配置
 type JimengConfig struct {
-	AccessKey string      `json:"access_key"`
+	// 即梦AI的AccessKey和SecretKey
-	SecretKey string      `json:"secret_key"`
+	AccessKey string `json:"access_key,omitempty"`
-	Power     JimengPower `json:"power"`
+	SecretKey string `json:"secret_key,omitempty"`
+	// 火山引擎大模型专用的验证方式
+	ApiKey string `json:"api_key,omitempty"`
+	// 算力配置
+	Powers map[string]int `json:"powers,omitempty"`
 }
 
-// JimengPower 即梦AI算力配置
+// JMTaskStatus 任务状态
-type JimengPower struct {
+type JMTaskStatus string
-	TextToImage  int `json:"text_to_image"`
+
-	ImageToImage int `json:"image_to_image"`
+const (
-	ImageEdit    int `json:"image_edit"`
+	JMTaskStatusInQueue    = JMTaskStatus("in_queue")   // 任务已提交
-	ImageEffects int `json:"image_effects"`
+	JMTaskStatusGenerating = JMTaskStatus("generating") // 任务处理中
-	TextToVideo  int `json:"text_to_video"`
+	JMTaskStatusDone       = JMTaskStatus("done")       // 处理完成
-	ImageToVideo int `json:"image_to_video"`
+	JMTaskStatusNotFound   = JMTaskStatus("not_found")  // 任务未找到
+	JMTaskStatusSuccess    = JMTaskStatus("success")    // 任务成功
+	JMTaskStatusFailed     = JMTaskStatus("failed")     // 任务失败
+	JMTaskStatusExpired    = JMTaskStatus("expired")    // 任务过期
+)
+
+// JMTaskType 任务类型
+type JMTaskType string
+
+const (
+	JMTaskTypeImage          = JMTaskType("image")           // 文生图
+	JMTaskTypeVideo          = JMTaskType("video")           // 图生图
+	JMTaskTypeVirtualHuman   = JMTaskType("virtual_human")   // 图像编辑
+	JMTaskTypeActionTransfer = JMTaskType("action_transfer") // 图像特效
+)
+
+// JimengTaskRequest 即梦AI任务请求
+type JimengTaskRequest struct {
+	TaskType JMTaskType `json:"type"`    // 任务类型
+	ReqKey   string     `json:"req_key"` // 请求Key
+	Action   string     `json:"action"`  // 请求Action
+	Power    int        `json:"power"`   // 消耗算力
+	// 公共参数
+	Prompt    string   `json:"prompt,omitempty"`
+	ImageUrls []string `json:"image_urls,omitempty"`
+
+	// 图片生成参数
+	Size        string  `json:"size,omitempty"`
+	UsePreLLM   bool    `json:"use_pre_llm,omitempty"`
+	Scale       float64 `json:"scale,omitempty"`
+	ForceSingle bool    `json:"force_single,omitempty"`
+
+	// 视频生成参数
+	Duration       int    `json:"duration,omitempty"`    // 视频时长，单位：秒
+	TemplateId     string `json:"template_id,omitempty"` // 运镜模板ID
+	AspectRatio    string `json:"aspect_ratio,omitempty"`
+	CameraStrength string `json:"camera_strength,omitempty"` // 运镜强度
+
+	// 数字人视频生成参数
+	AudioURL     string `json:"audio_url,omitempty"`     // 音频URL
+	RecognizeKey string `json:"recognize_key,omitempty"` // 识别主体请求Key
+
+	// 视频动作迁移参数
+	VideoURL string `json:"video_url,omitempty"` // 动作视频URL
 }

api/core/types/moderation.go

@@ -9,13 +9,13 @@ package types
 
 // 文本审查
 type ModerationConfig struct {
-	Enable      bool                    `json:"enable"` // 是否启用文本审查
+	Enable      bool                    `json:"enable,omitempty"` // 是否启用文本审查
-	Active      string                  `json:"active"`
+	Active      string                  `json:"active,omitempty"`
-	EnableGuide bool                    `json:"enable_guide"` // 是否启用模型引导提示词
+	EnableGuide bool                    `json:"enable_guide,omitempty"` // 是否启用模型引导提示词
-	GuidePrompt string                  `json:"guide_prompt"` // 模型引导提示词
+	GuidePrompt string                  `json:"guide_prompt,omitempty"` // 模型引导提示词
-	Gitee       ModerationGiteeConfig   `json:"gitee"`
+	Gitee       ModerationGiteeConfig   `json:"gitee,omitempty"`
-	Baidu       ModerationBaiduConfig   `json:"baidu"`
+	Baidu       ModerationBaiduConfig   `json:"baidu,omitempty"`
-	Tencent     ModerationTencentConfig `json:"tencent"`
+	Tencent     ModerationTencentConfig `json:"tencent,omitempty"`
 }
 
 const (
@@ -26,26 +26,26 @@ const (
 
 // GiteeAI 文本审查配置
 type ModerationGiteeConfig struct {
-	ApiKey string `json:"api_key"`
+	ApiKey string `json:"api_key,omitempty"`
-	Model  string `json:"model"` // 文本审核模型
+	Model  string `json:"model,omitempty"` // 文本审核模型
 }
 
 // 百度文本审查配置
 type ModerationBaiduConfig struct {
-	AccessKey string `json:"access_key"`
+	AccessKey string `json:"access_key,omitempty"`
-	SecretKey string `json:"secret_key"`
+	SecretKey string `json:"secret_key,omitempty"`
 }
 
 // 腾讯云文本审查配置
 type ModerationTencentConfig struct {
-	AccessKey string `json:"access_key"`
+	AccessKey string `json:"access_key,omitempty"`
-	SecretKey string `json:"secret_key"`
+	SecretKey string `json:"secret_key,omitempty"`
 }
 
 type ModerationResult struct {
-	Flagged        bool               `json:"flagged"`
+	Flagged        bool               `json:"flagged,omitempty"`
-	Categories     map[string]bool    `json:"categories"`
+	Categories     map[string]bool    `json:"categories,omitempty"`
-	CategoryScores map[string]float64 `json:"category_scores"`
+	CategoryScores map[string]float64 `json:"category_scores,omitempty"`
 }
 
 var ModerationCategories = map[string]string{

api/core/types/oss.go

@@ -8,39 +8,39 @@ package types
 // * +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 
 type OSSConfig struct {
-	Active string             `json:"active"`
+	Active string             `json:"active,omitempty"`
-	Local  LocalStorageConfig `json:"local"`
+	Local  LocalStorageConfig `json:"local,omitempty"`
-	Minio  MiniOssConfig      `json:"minio"`
+	Minio  MiniOssConfig      `json:"minio,omitempty"`
-	QiNiu  QiNiuOssConfig     `json:"qiniu"`
+	QiNiu  QiNiuOssConfig     `json:"qiniu,omitempty"`
-	AliYun AliYunOssConfig    `json:"aliyun"`
+	AliYun AliYunOssConfig    `json:"aliyun,omitempty"`
 }
 
 type MiniOssConfig struct {
-	Endpoint     string `json:"endpoint"`
+	Endpoint     string `json:"endpoint,omitempty"`
-	AccessKey    string `json:"access_key"`
+	AccessKey    string `json:"access_key,omitempty"`
-	AccessSecret string `json:"access_secret"`
+	AccessSecret string `json:"access_secret,omitempty"`
-	Bucket       string `json:"bucket"`
+	Bucket       string `json:"bucket,omitempty"`
-	UseSSL       bool   `json:"use_ssl"`
+	UseSSL       bool   `json:"use_ssl,omitempty"`
-	Domain       string `json:"domain"`
+	Domain       string `json:"domain,omitempty"`
 }
 
 type QiNiuOssConfig struct {
-	Zone         string `json:"zone"`
+	Zone         string `json:"zone,omitempty"`
-	AccessKey    string `json:"access_key"`
+	AccessKey    string `json:"access_key,omitempty"`
-	AccessSecret string `json:"access_secret"`
+	AccessSecret string `json:"access_secret,omitempty"`
-	Bucket       string `json:"bucket"`
+	Bucket       string `json:"bucket,omitempty"`
-	Domain       string `json:"domain"`
+	Domain       string `json:"domain,omitempty"`
 }
 
 type AliYunOssConfig struct {
-	Endpoint     string `json:"endpoint"`
+	Endpoint     string `json:"endpoint,omitempty"`
-	AccessKey    string `json:"access_key"`
+	AccessKey    string `json:"access_key,omitempty"`
-	AccessSecret string `json:"access_secret"`
+	AccessSecret string `json:"access_secret,omitempty"`
-	Bucket       string `json:"bucket"`
+	Bucket       string `json:"bucket,omitempty"`
-	Domain       string `json:"domain"`
+	Domain       string `json:"domain,omitempty"`
 }
 
 type LocalStorageConfig struct {
-	BasePath string `json:"base_path"`
+	BasePath string `json:"base_path,omitempty"`
-	BaseURL  string `json:"base_url"`
+	BaseURL  string `json:"base_url,omitempty"`
 }

api/core/types/payment.go

@@ -1,19 +1,19 @@
 package types
 
 type PaymentConfig struct {
-	Alipay AlipayConfig `json:"alipay"` // 支付宝支付渠道配置
+	Alipay AlipayConfig `json:"alipay,omitempty"` // 支付宝支付渠道配置
-	Epay   EpayConfig   `json:"epay"`   // 易支付配置
+	Epay   EpayConfig   `json:"epay,omitempty"`   // 易支付配置
-	WxPay  WxPayConfig  `json:"wxpay"`  // 微信支付渠道配置
+	WxPay  WxPayConfig  `json:"wxpay,omitempty"`  // 微信支付渠道配置
 }
 
 // AlipayConfig 支付宝支付配置
 type AlipayConfig struct {
-	Enabled         bool   `json:"enabled"`           // 是否启用该支付通道
+	Enabled         bool   `json:"enabled,omitempty"`           // 是否启用该支付通道
-	SandBox         bool   `json:"sandbox"`           // 是否沙盒环境
+	SandBox         bool   `json:"sandbox,omitempty"`           // 是否沙盒环境
-	AppId           string `json:"app_id"`            // 应用 ID
+	AppId           string `json:"app_id,omitempty"`            // 应用 ID
-	PrivateKey      string `json:"private_key"`       // 应用私钥
+	PrivateKey      string `json:"private_key,omitempty"`       // 应用私钥
-	AlipayPublicKey string `json:"alipay_public_key"` // 支付宝公钥
+	AlipayPublicKey string `json:"alipay_public_key,omitempty"` // 支付宝公钥
-	Domain          string `json:"domain"`            // 支付回调域名
+	Domain          string `json:"domain,omitempty"`            // 支付回调域名
 }
 
 func (c *AlipayConfig) Equal(other *AlipayConfig) bool {
@@ -25,13 +25,13 @@ func (c *AlipayConfig) Equal(other *AlipayConfig) bool {
 
 // WxPayConfig 微信支付配置
 type WxPayConfig struct {
-	Enabled    bool   `json:"enabled"`     // 是否启用该支付通道
+	Enabled    bool   `json:"enabled,omitempty"`     // 是否启用该支付通道
-	AppId      string `json:"app_id"`      // 公众号的APPID,如：wxd678efh567hg6787
+	AppId      string `json:"app_id,omitempty"`      // 公众号的APPID,如：wxd678efh567hg6787
-	MchId      string `json:"mch_id"`      // 直连商户的商户号，由微信支付生成并下发
+	MchId      string `json:"mch_id,omitempty"`      // 直连商户的商户号，由微信支付生成并下发
-	SerialNo   string `json:"serial_no"`   // 商户证书的证书序列号
+	SerialNo   string `json:"serial_no,omitempty"`   // 商户证书的证书序列号
-	PrivateKey string `json:"private_key"` // 商户证书私钥
+	PrivateKey string `json:"private_key,omitempty"` // 商户证书私钥
-	ApiV3Key   string `json:"api_v3_key"`  // API V3 秘钥
+	ApiV3Key   string `json:"api_v3_key,omitempty"`  // API V3 秘钥
-	Domain     string `json:"domain"`      // 支付回调域名
+	Domain     string `json:"domain,omitempty"`      // 支付回调域名
 }
 
 func (c *WxPayConfig) Equal(other *WxPayConfig) bool {
@@ -45,11 +45,11 @@ func (c *WxPayConfig) Equal(other *WxPayConfig) bool {
 
 // EpayConfig 易支付配置
 type EpayConfig struct {
-	Enabled    bool   `json:"enabled"`     // 是否启用该支付通道
+	Enabled    bool   `json:"enabled,omitempty"`     // 是否启用该支付通道
-	AppId      string `json:"app_id"`      // 商户 ID
+	AppId      string `json:"app_id,omitempty"`      // 商户 ID
-	PrivateKey string `json:"private_key"` // 私钥
+	PrivateKey string `json:"private_key,omitempty"` // 私钥
-	ApiURL     string `json:"api_url"`     // z支付 API 网关
+	ApiURL     string `json:"api_url,omitempty"`     // z支付 API 网关
-	Domain     string `json:"domain"`      // 支付回调域名
+	Domain     string `json:"domain,omitempty"`      // 支付回调域名
 }
 
 func (c *EpayConfig) Equal(other *EpayConfig) bool {

api/core/types/sms.go

@@ -8,23 +8,23 @@ package types
 // * +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 
 type SMSConfig struct {
-	Active string       `json:"active"`
+	Active string       `json:"active,omitempty"`
-	Ali    SmsConfigAli `json:"aliyun"`
+	Ali    SmsConfigAli `json:"aliyun,omitempty"`
-	Bao    SmsConfigBao `json:"bao"`
+	Bao    SmsConfigBao `json:"bao,omitempty"`
 }
 
 // SmsConfigAli 阿里云短信平台配置
 type SmsConfigAli struct {
-	AccessKey    string `json:"access_key"`
+	AccessKey    string `json:"access_key,omitempty"`
-	AccessSecret string `json:"access_secret"`
+	AccessSecret string `json:"access_secret,omitempty"`
-	Sign         string `json:"sign"`         // 短信签名
+	Sign         string `json:"sign,omitempty"`         // 短信签名
-	CodeTempId   string `json:"code_temp_id"` // 验证码短信模板 ID
+	CodeTempId   string `json:"code_temp_id,omitempty"` // 验证码短信模板 ID
 }
 
 // SmsConfigBao 短信宝平台配置
 type SmsConfigBao struct {
-	Username     string `json:"username"`      //短信宝平台注册的用户名
+	Username     string `json:"username,omitempty"`      //短信宝平台注册的用户名
-	Password     string `json:"password"`      //短信宝平台注册的密码
+	Password     string `json:"password,omitempty"`      //短信宝平台注册的密码
-	Sign         string `json:"sign"`          // 短信签名
+	Sign         string `json:"sign,omitempty"`          // 短信签名
-	CodeTemplate string `json:"code_template"` // 验证码短信模板 匹配
+	CodeTemplate string `json:"code_template,omitempty"` // 验证码短信模板 匹配
 }

api/core/types/smtp.go

@@ -8,12 +8,12 @@ package types
 // * +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 
 type SmtpConfig struct {
-	UseTls   bool   `json:"use_tls"`  // 是否使用 TLS 发送
+	UseTls   bool   `json:"use_tls,omitempty"`  // 是否使用 TLS 发送
-	Host     string `json:"host"`     // 邮件服务器地址
+	Host     string `json:"host,omitempty"`     // 邮件服务器地址
-	Port     int    `json:"port"`     // 邮件服务器端口
+	Port     int    `json:"port,omitempty"`     // 邮件服务器端口
-	AppName  string `json:"app_name"` // 应用名称
+	AppName  string `json:"app_name,omitempty"` // 应用名称
-	From     string `json:"from"`     // 发件人邮箱地址
+	From     string `json:"from,omitempty"`     // 发件人邮箱地址
-	Password string `json:"password"` // 发件人邮箱密码
+	Password string `json:"password,omitempty"` // 发件人邮箱密码
 }
 
 func (s *SmtpConfig) Equal(other *SmtpConfig) bool {

api/core/types/user_lock.go

@@ -0,0 +1,45 @@
+package types
+
+// * +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
+// * Copyright 2023 The Geek-AI Authors. All rights reserved.
+// * Use of this source code is governed by a Apache-2.0 license
+// * that can be found in the LICENSE file.
+// * @Author yangjian102621@163.com
+// * +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
+
+import "sync"
+
+// UserLockManager 提供基于用户ID的TryLock功能，确保同一用户并发请求串行化
+type UserLockManager struct {
+	mu    sync.Mutex
+	locks map[uint]bool
+}
+
+func NewUserLockManager() *UserLockManager {
+	return &UserLockManager{mu: sync.Mutex{}, locks: make(map[uint]bool)}
+}
+
+// TryLock 尝试为指定用户加锁。若已被占用返回 false
+func (m *UserLockManager) TryLock(userId uint) bool {
+	if userId == 0 {
+		return true
+	}
+	m.mu.Lock()
+	defer m.mu.Unlock()
+
+	if m.locks[userId] {
+		return false
+	}
+	m.locks[userId] = true
+	return true
+}
+
+// Unlock 释放指定用户的锁
+func (m *UserLockManager) Unlock(userId uint) {
+	if userId == 0 {
+		return
+	}
+	m.mu.Lock()
+	delete(m.locks, userId)
+	m.mu.Unlock()
+}

api/go.mod

@@ -33,6 +33,7 @@ require (
 	github.com/shirou/gopsutil v3.21.11+incompatible
 	github.com/shopspring/decimal v1.3.1
 	github.com/syndtr/goleveldb v1.0.0
+	github.com/volcengine/volcengine-go-sdk v1.1.34
 	golang.org/x/image v0.15.0
 )
 
@@ -50,6 +51,7 @@ require (
 	github.com/tklauser/numcpus v0.7.0 // indirect
 	github.com/yusufpapurcu/wmi v1.2.4 // indirect
 	go.uber.org/mock v0.4.0 // indirect
+	gopkg.in/yaml.v2 v2.4.0 // indirect
 )
 
 require (

api/go.sum

@@ -100,6 +100,7 @@ github.com/golang/protobuf v1.4.0-rc.4.0.20200313231945-b860323f09d0/go.mod h1:W
 github.com/golang/protobuf v1.4.0/go.mod h1:jodUvKwWbYaEsadDk5Fwe5c77LiNKVO9IDvqG2KuDX0=
 github.com/golang/protobuf v1.4.1/go.mod h1:U8fpvMrcmy5pZrNK1lt4xCsGvpyWQ/VVv6QDs8UjoX8=
 github.com/golang/protobuf v1.4.3/go.mod h1:oDoupMAO8OvCJWAcko0GGGIgR6R6ocIYbsSw735rRwI=
+github.com/golang/protobuf v1.5.0/go.mod h1:FsONVRAS9T7sI+LIUmWTfcYkHO4aIWwzhcaSAoJOfIk=
 github.com/golang/protobuf v1.5.3 h1:KhyjKVUg7Usr/dYsdSqoFveMYd5ko72D+zANwlG1mmg=
 github.com/golang/protobuf v1.5.3/go.mod h1:XVQd3VNwM+JqD3oG2Ue2ip4fOMUkwXdXDdiuN0vRsmY=
 github.com/golang/snappy v0.0.0-20180518054509-2e65f85255db/go.mod h1:/XxbfmMg8lxefKM7IXC3fBNl/7bRcc72aCRzEWrmP2Q=
@@ -110,6 +111,7 @@ github.com/google/go-cmp v0.3.0/go.mod h1:8QqcDgzrUqlUb/G2PQTWiueGozuR1884gddMyw
 github.com/google/go-cmp v0.3.1/go.mod h1:8QqcDgzrUqlUb/G2PQTWiueGozuR1884gddMywk6iLU=
 github.com/google/go-cmp v0.4.0/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=
 github.com/google/go-cmp v0.5.0/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=
+github.com/google/go-cmp v0.5.5/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=
 github.com/google/go-cmp v0.6.0 h1:ofyhxvXcZhMsU5ulbFiLKl/XBFqE1GSq7atu8tAmTRI=
 github.com/google/go-cmp v0.6.0/go.mod h1:17dUlkBOakJ0+DkrSSNjCkIjxS6bF9zb3elmeNGIjoY=
 github.com/google/go-tika v0.3.1 h1:l+jr10hDhZjcgxFRfcQChRLo1bPXQeLFluMyvDhXTTA=
@@ -259,6 +261,8 @@ github.com/ugorji/go/codec v1.2.11 h1:BMaWp1Bb6fHwEtbplGBGJ498wD+LKlNSl25MjdZY4d
 github.com/ugorji/go/codec v1.2.11/go.mod h1:UNopzCgEMSXjBc6AOMqYvWC1ktqTAfzJZUZgYf6w6lg=
 github.com/volcengine/volc-sdk-golang v1.0.23 h1:anOslb2Qp6ywnsbyq9jqR0ljuO63kg9PY+4OehIk5R8=
 github.com/volcengine/volc-sdk-golang v1.0.23/go.mod h1:AfG/PZRUkHJ9inETvbjNifTDgut25Wbkm2QoYBTbvyU=
+github.com/volcengine/volcengine-go-sdk v1.1.34 h1:ha90JycCCTJNCse0UDziBgBsuX98ITOrkwYlDWcm7NI=
+github.com/volcengine/volcengine-go-sdk v1.1.34/go.mod h1:oxoVo+A17kvkwPkIeIHPVLjSw7EQAm+l/Vau1YGHN+A=
 github.com/yuin/goldmark v1.4.13/go.mod h1:6yULJ656Px+3vBD8DxQVa3kxgyrAnzto9xy5taEt/CY=
 github.com/yusufpapurcu/wmi v1.2.4 h1:zFUKzehAFReQwLys1b/iSMl+JQGSCSjtVqQn9bBrPo0=
 github.com/yusufpapurcu/wmi v1.2.4/go.mod h1:SBZ9tNy3G9/m5Oi98Zks0QjeHVDvuK0qfxQmPyzfmi0=
@@ -390,6 +394,8 @@ google.golang.org/protobuf v1.22.0/go.mod h1:EGpADcykh3NcUnDUJcl1+ZksZNG86OlYog2
 google.golang.org/protobuf v1.23.0/go.mod h1:EGpADcykh3NcUnDUJcl1+ZksZNG86OlYog2l/sGQquU=
 google.golang.org/protobuf v1.23.1-0.20200526195155-81db48ad09cc/go.mod h1:EGpADcykh3NcUnDUJcl1+ZksZNG86OlYog2l/sGQquU=
 google.golang.org/protobuf v1.25.0/go.mod h1:9JNX74DMeImyA3h4bdi1ymwjUzf21/xIlbajtzgsN7c=
+google.golang.org/protobuf v1.26.0-rc.1/go.mod h1:jlhhOSvTdKEhbULTjvd4ARK9grFBp09yW+WbY/TyQbw=
+google.golang.org/protobuf v1.31.0/go.mod h1:HV8QOd/L58Z+nl8r43ehVNZIU/HEI6OcFqwMG9pJV4I=
 google.golang.org/protobuf v1.33.0 h1:uNO2rsAINq/JlFpSdYEKIZ0uKD/R9cpdv0T+yoGwGmI=
 google.golang.org/protobuf v1.33.0/go.mod h1:c6P6GXX6sHbq/GpV6MGZEdwhWPcYBgnhAHhKbcUYpos=
 gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=

api/handler/admin/admin_handler.go

@@ -31,7 +31,7 @@ import (
 
 var logger = logger2.GetLogger()
 
-const SuperManagerID = 1
+const SuperUsername = "admin"
 
 type ManagerHandler struct {
 	handler.BaseHandler
@@ -94,7 +94,7 @@ func (h *ManagerHandler) Login(c *gin.Context) {
 	}
 
 	// 超级管理员默认是ID:1
-	if manager.Id != SuperManagerID && manager.Status == false {
+	if manager.Username != SuperUsername && !manager.Status {
 		resp.ERROR(c, "该用户已被禁止登录，请联系超级管理员")
 		return
 	}
@@ -125,7 +125,7 @@ func (h *ManagerHandler) Login(c *gin.Context) {
 		IsSuperAdmin bool   `json:"is_super_admin"`
 		Token        string `json:"token"`
 	}{
-		IsSuperAdmin: manager.Id == 1,
+		IsSuperAdmin: manager.Username == SuperUsername,
 		Token:        tokenString,
 	}
 
@@ -227,12 +227,19 @@ func (h *ManagerHandler) Remove(c *gin.Context) {
 		return
 	}
 
-	if id == SuperManagerID {
+	var user model.AdminUser
+	res := h.DB.Where("id", id).First(&user)
+	if res.Error != nil {
+		resp.ERROR(c, res.Error.Error())
+		return
+	}
+
+	if user.Username == SuperUsername {
 		resp.ERROR(c, "超级管理员不能删除")
 		return
 	}
 
-	res := h.DB.Where("id", id).Delete(&model.AdminUser{})
+	res = h.DB.Where("id", id).Delete(&model.AdminUser{})
 	if res.Error != nil {
 		resp.ERROR(c, res.Error.Error())
 		return
@@ -263,8 +270,14 @@ func (h *ManagerHandler) Enable(c *gin.Context) {
 
 // ResetPass 重置密码
 func (h *ManagerHandler) ResetPass(c *gin.Context) {
-	id := h.GetLoginUserId(c)
+	id := h.GetAdminId(c)
-	if id != SuperManagerID {
+	var user model.AdminUser
+	res := h.DB.Where("id", id).First(&user)
+	if res.Error != nil {
+		resp.ERROR(c, res.Error.Error())
+		return
+	}
+	if user.Username != SuperUsername {
 		resp.ERROR(c, "只有超级管理员能够进行该操作")
 		return
 	}
@@ -278,13 +291,6 @@ func (h *ManagerHandler) ResetPass(c *gin.Context) {
 		return
 	}
 
-	var user model.AdminUser
-	res := h.DB.Where("id", data.Id).First(&user)
-	if res.Error != nil {
-		resp.ERROR(c, res.Error.Error())
-		return
-	}
-
 	password := utils.GenPassword(data.Password, user.Salt)
 	user.Password = password
 	res = h.DB.Updates(&user)

api/handler/admin/config_handler.go

@@ -368,10 +368,7 @@ func (h *ConfigHandler) UpdateWxLogin(c *gin.Context) {
 		return
 	}
 
-	if data.Enabled {
 	h.wxLoginService.UpdateConfig(data)
-	}
-
 	h.sysConfig.WxLogin = data
 	resp.SUCCESS(c, data)
 }

api/handler/admin/jimeng_handler.go

@@ -131,7 +131,7 @@ func (h *AdminJimengHandler) BatchRemove(c *gin.Context) {
 			continue // 跳过不存在的
 		}
 		tx := h.DB.Begin()
-		if job.Status != model.JMTaskStatusSuccess && job.Power > 0 {
+		if job.Status != types.JMTaskStatusSuccess && job.Power > 0 {
 			remark := fmt.Sprintf("任务未成功，退回算力。任务ID：%d，Err: %s", job.Id, job.ErrMsg)
 			err = h.userService.IncreasePower(job.UserId, job.Power, model.PowerLog{
 				Type:   types.PowerRefund,
@@ -172,7 +172,7 @@ func (h *AdminJimengHandler) BatchRemove(c *gin.Context) {
 // Stats 获取统计信息
 func (h *AdminJimengHandler) Stats(c *gin.Context) {
 	type StatResult struct {
-		Status model.JMTaskStatus `json:"status"`
+		Status types.JMTaskStatus `json:"status"`
 		Count  int64              `json:"count"`
 	}
 
@@ -198,13 +198,13 @@ func (h *AdminJimengHandler) Stats(c *gin.Context) {
 	for _, stat := range stats {
 		result["totalTasks"] = result["totalTasks"].(int64) + stat.Count
 		switch stat.Status {
-		case model.JMTaskStatusInQueue:
+		case types.JMTaskStatusInQueue:
 			result["pendingTasks"] = stat.Count
-		case model.JMTaskStatusSuccess:
+		case types.JMTaskStatusSuccess:
 			result["completedTasks"] = stat.Count
-		case model.JMTaskStatusGenerating:
+		case types.JMTaskStatusGenerating:
 			result["processingTasks"] = stat.Count
-		case model.JMTaskStatusFailed:
+		case types.JMTaskStatusFailed:
 			result["failedTasks"] = stat.Count
 		}
 	}
@@ -231,29 +231,15 @@ func (h *AdminJimengHandler) UpdateConfig(c *gin.Context) {
 	}
 
 	// 验证算力配置
-	if req.Power.TextToImage <= 0 {
+	if len(req.Powers) == 0 {
-		resp.ERROR(c, "文生图算力必须大于0")
+		resp.ERROR(c, "请至少配置一个模型的积分")
 		return
 	}
-	if req.Power.ImageToImage <= 0 {
+	for key, val := range req.Powers {
-		resp.ERROR(c, "图生图算力必须大于0")
+		if val <= 0 {
+			resp.ERROR(c, fmt.Sprintf("模型 %s 的积分必须大于0", key))
 			return
 		}
-	if req.Power.ImageEdit <= 0 {
-		resp.ERROR(c, "图片编辑算力必须大于0")
-		return
-	}
-	if req.Power.ImageEffects <= 0 {
-		resp.ERROR(c, "图片特效算力必须大于0")
-		return
-	}
-	if req.Power.TextToVideo <= 0 {
-		resp.ERROR(c, "文生视频算力必须大于0")
-		return
-	}
-	if req.Power.ImageToVideo <= 0 {
-		resp.ERROR(c, "图生视频算力必须大于0")
-		return
 	}
 
 	// 保存配置

api/handler/chat_handler.go

@@ -69,6 +69,7 @@ type ChatHandler struct {
 	ReqCancelFunc     *types.LMap[string, context.CancelFunc] // HttpClient 请求取消 handle function
 	userService       *service.UserService
 	moderationManager *moderation.ServiceManager
+	userLocks         *types.UserLockManager
 }
 
 func NewChatHandler(app *core.AppServer, db *gorm.DB, redis *redis.Client, manager *oss.UploaderManager, licenseService *service.LicenseService, userService *service.UserService, moderationManager *moderation.ServiceManager) *ChatHandler {
@@ -80,6 +81,7 @@ func NewChatHandler(app *core.AppServer, db *gorm.DB, redis *redis.Client, manag
 		ReqCancelFunc:     types.NewLMap[string, context.CancelFunc](),
 		userService:       userService,
 		moderationManager: moderationManager,
+		userLocks:         types.NewUserLockManager(),
 	}
 }
 
@@ -120,6 +122,14 @@ func (h *ChatHandler) Chat(c *gin.Context) {
 		return
 	}
 
+	// 用户级并发锁，确保同一用户同时只有一个对话请求
+	if !h.userLocks.TryLock(input.UserId) {
+		pushMessage(c, ChatEventError, "您有一个对话请求正在进行中，请稍后再试或先停止当前生成！")
+		c.Abort()
+		return
+	}
+	defer h.userLocks.Unlock(input.UserId)
+
 	ctx, cancel := context.WithCancel(c.Request.Context())
 	defer cancel()
 
@@ -262,9 +272,9 @@ func (h *ChatHandler) sendMessage(ctx context.Context, input ChatInput, c *gin.C
 		if h.App.SysConfig.Base.ContextDeep > 0 {
 			var historyMessages []model.ChatMessage
 			dbSession := h.DB.Session(&gorm.Session{}).Where("chat_id", input.ChatId)
-			if input.LastMsgId > 0 { // 重新生成逻辑
+			if input.LastMsgId > 0 { // 重新生成和编辑逻辑
 				var lastMessage model.ChatMessage
-				err = dbSession.Where("id <= ?", input.LastMsgId).Where("type", types.PromptMsg).First(&lastMessage).Error
+				err = dbSession.Where("id < ?", input.LastMsgId).Where("type", types.ReplyMsg).Order("id DESC").First(&lastMessage).Error
 				if err != nil {
 					input.LastMsgId = 0
 				} else {
@@ -272,7 +282,7 @@ func (h *ChatHandler) sendMessage(ctx context.Context, input ChatInput, c *gin.C
 				}
 				dbSession = dbSession.Where("id < ?", input.LastMsgId)
 				// 删除对应的聊天记录
-				h.DB.Debug().Where("chat_id", input.ChatId).Where("id >= ?", input.LastMsgId).Delete(&model.ChatMessage{})
+				h.DB.Debug().Where("chat_id", input.ChatId).Where("id > ?", input.LastMsgId).Delete(&model.ChatMessage{})
 			}
 			err = dbSession.Limit(h.App.SysConfig.Base.ContextDeep).Order("id DESC").Find(&historyMessages).Error
 			if err == nil {

api/handler/jimeng_handler.go

@@ -1,6 +1,7 @@
 package handler
 
 import (
+	"errors"
 	"fmt"
 	"geekai/core"
 	"geekai/core/middleware"
@@ -38,49 +39,28 @@ func NewJimengHandler(app *core.AppServer, jimengService *jimeng.Service, db *go
 // RegisterRoutes 注册路由，新增统一任务接口
 func (h *JimengHandler) RegisterRoutes() {
 	group := h.App.Engine.Group("/api/jimeng/")
+	group.GET("power-config", h.GetPowerConfig)
 
 	// 需要用户授权的接口
 	group.Use(middleware.UserAuthMiddleware(h.App.Config.Session.SecretKey, h.App.Redis))
 	{
 		group.POST("task", h.CreateTask)
-		group.GET("power-config", h.GetPowerConfig)
 		group.POST("jobs", h.Jobs)
 		group.GET("remove", h.Remove)
 		group.GET("retry", h.Retry)
 	}
 }
 
-// JimengTaskRequest 统一任务请求结构体
-// 支持所有生图和生成视频类型
-type JimengTaskRequest struct {
-	TaskType         string   `json:"task_type" binding:"required"`
-	Prompt           string   `json:"prompt"`
-	ImageInput       string   `json:"image_input"`
-	ImageUrls        []string `json:"image_urls"`
-	BinaryDataBase64 []string `json:"binary_data_base64"`
-	Scale            float64  `json:"scale"`
-	Width            int      `json:"width"`
-	Height           int      `json:"height"`
-	Gpen             float64  `json:"gpen"`
-	Skin             float64  `json:"skin"`
-	SkinUnifi        float64  `json:"skin_unifi"`
-	GenMode          string   `json:"gen_mode"`
-	Seed             int64    `json:"seed"`
-	UsePreLLM        bool     `json:"use_pre_llm"`
-	TemplateId       string   `json:"template_id"`
-	AspectRatio      string   `json:"aspect_ratio"`
-}
-
 // CreateTask 统一任务创建接口
 func (h *JimengHandler) CreateTask(c *gin.Context) {
-	var req JimengTaskRequest
+	var req types.JimengTaskRequest
 	if err := c.ShouldBindJSON(&req); err != nil {
 		resp.ERROR(c, types.InvalidArgs)
 		return
 	}
 
 	// 文本审核
-	if h.App.SysConfig.Moderation.Enable {
+	if h.App.SysConfig.Moderation.Enable && req.Prompt != "" {
 		moderationResult, err := h.moderationManager.GetService().Moderate(req.Prompt)
 		if err != nil {
 			logger.Error("failed to moderate content: ", err)
@@ -103,136 +83,21 @@ func (h *JimengHandler) CreateTask(c *gin.Context) {
 
 	}
 
-	// 新增：除图像特效外，其他任务类型必须有提示词
+	if req.Prompt == "" && len(req.ImageUrls) == 0 {
-	if req.TaskType != "image_effects" && req.Prompt == "" {
+		resp.ERROR(c, "提示词和图片不能同时为空")
-		resp.ERROR(c, "提示词不能为空")
 		return
 	}
+
 	user, err := h.GetLoginUser(c)
 	if err != nil {
 		resp.NotAuth(c)
 		return
 	}
 
-	if req.Width == 0 {
+	// 获取算力消耗
-		req.Width = 1328
+	powerCost, err := h.getTaskPower(req)
-	}
+	if err != nil {
-	if req.Height == 0 {
+		resp.ERROR(c, "计算任务消耗积分失败: "+err.Error())
-		req.Height = 1328
-	}
-	if req.Seed == 0 {
-		req.Seed = -1
-	}
-
-	var powerCost int
-	var taskType model.JMTaskType
-	var params map[string]any
-	var reqKey string
-	var modelName string
-
-	switch req.TaskType {
-	case "text_to_image":
-		powerCost = h.getPowerFromConfig(model.JMTaskTypeTextToImage)
-		taskType = model.JMTaskTypeTextToImage
-		reqKey = jimeng.ReqKeyTextToImage
-		modelName = "即梦文生图"
-		if req.Scale == 0 {
-			req.Scale = 2.5
-		}
-		params = map[string]any{
-			"seed":        req.Seed,
-			"scale":       req.Scale,
-			"width":       req.Width,
-			"height":      req.Height,
-			"use_pre_llm": req.UsePreLLM,
-		}
-	case "image_to_image":
-		powerCost = h.getPowerFromConfig(model.JMTaskTypeImageToImage)
-		taskType = model.JMTaskTypeImageToImage
-		reqKey = jimeng.ReqKeyImageToImagePortrait
-		modelName = "即梦图生图"
-		if req.Gpen == 0 {
-			req.Gpen = 0.4
-		}
-		if req.Skin == 0 {
-			req.Skin = 0.3
-		}
-		if req.GenMode == "" {
-			if req.Prompt != "" {
-				req.GenMode = jimeng.GenModeCreative
-			} else {
-				req.GenMode = jimeng.GenModeReference
-			}
-		}
-		params = map[string]any{
-			"image_input": req.ImageInput,
-			"width":       req.Width,
-			"height":      req.Height,
-			"gpen":        req.Gpen,
-			"skin":        req.Skin,
-			"skin_unifi":  req.SkinUnifi,
-			"gen_mode":    req.GenMode,
-			"seed":        req.Seed,
-		}
-	case "image_edit":
-		powerCost = h.getPowerFromConfig(model.JMTaskTypeImageEdit)
-		taskType = model.JMTaskTypeImageEdit
-		reqKey = jimeng.ReqKeyImageEdit
-		modelName = "即梦图像编辑"
-		if req.Scale == 0 {
-			req.Scale = 0.5
-		}
-		params = map[string]any{
-			"seed":  req.Seed,
-			"scale": req.Scale,
-		}
-		params["image_urls"] = []string{req.ImageInput}
-	case "image_effects":
-		powerCost = h.getPowerFromConfig(model.JMTaskTypeImageEffects)
-		taskType = model.JMTaskTypeImageEffects
-		reqKey = jimeng.ReqKeyImageEffects
-		modelName = "即梦图像特效"
-		if req.Width == 0 {
-			req.Width = 1328
-		}
-		if req.Height == 0 {
-			req.Height = 1328
-		}
-		params = map[string]any{
-			"image_input1": req.ImageInput,
-			"template_id":  req.TemplateId,
-			"width":        req.Width,
-			"height":       req.Height,
-		}
-	case "text_to_video":
-		powerCost = h.getPowerFromConfig(model.JMTaskTypeTextToVideo)
-		taskType = model.JMTaskTypeTextToVideo
-		reqKey = jimeng.ReqKeyTextToVideo
-		modelName = "即梦文生视频"
-		if req.AspectRatio == "" {
-			req.AspectRatio = jimeng.AspectRatio16_9
-		}
-		params = map[string]any{
-			"seed":         req.Seed,
-			"aspect_ratio": req.AspectRatio,
-		}
-	case "image_to_video":
-		powerCost = h.getPowerFromConfig(model.JMTaskTypeImageToVideo)
-		taskType = model.JMTaskTypeImageToVideo
-		reqKey = jimeng.ReqKeyImageToVideo
-		modelName = "即梦图生视频"
-		params = map[string]any{
-			"seed":         req.Seed,
-			"aspect_ratio": req.AspectRatio,
-		}
-		if len(req.ImageUrls) > 0 {
-			params["image_urls"] = req.ImageUrls
-		}
-		if len(req.BinaryDataBase64) > 0 {
-			params["binary_data_base64"] = req.BinaryDataBase64
-		}
-	default:
-		resp.ERROR(c, "不支持的任务类型")
 		return
 	}
 
@@ -240,16 +105,9 @@ func (h *JimengHandler) CreateTask(c *gin.Context) {
 		resp.ERROR(c, fmt.Sprintf("算力不足，需要%d算力", powerCost))
 		return
 	}
+	req.Power = powerCost
 
-	taskReq := &jimeng.CreateTaskRequest{
+	job, err := h.jimengService.CreateTask(user.Id, &req)
-		Type:   taskType,
-		Prompt: req.Prompt,
-		Params: params,
-		ReqKey: reqKey,
-		Power:  powerCost,
-	}
-
-	job, err := h.jimengService.CreateTask(user.Id, taskReq)
 	if err != nil {
 		logger.Errorf("create jimeng task failed: %v", err)
 		resp.ERROR(c, "创建任务失败")
@@ -258,11 +116,42 @@ func (h *JimengHandler) CreateTask(c *gin.Context) {
 
 	h.userService.DecreasePower(user.Id, powerCost, model.PowerLog{
 		Type:   types.PowerConsume,
-		Model:  "jimeng",
+		Model:  job.ReqKey,
-		Remark: fmt.Sprintf("%s，任务ID：%d", modelName, job.Id),
+		Remark: h.getTaskRemark(req, job.Id),
 	})
 
-	resp.SUCCESS(c, job)
+	resp.SUCCESS(c)
+}
+
+func (h *JimengHandler) getTaskRemark(req types.JimengTaskRequest, jobId uint) string {
+	remark := fmt.Sprintf("即梦任务%s，任务ID：%d", req.ReqKey, jobId)
+	perUnit, ok := h.App.SysConfig.Jimeng.Powers[req.ReqKey]
+	if !ok || perUnit <= 0 {
+		return remark // Fallback if power not found or invalid
+	}
+	switch req.TaskType {
+	case types.JMTaskTypeImage:
+		remark = fmt.Sprintf("即梦图片生成，任务ID：%d，%d积分/张", jobId, perUnit)
+	case types.JMTaskTypeVideo:
+		seconds := 0
+		if perUnit > 0 {
+			seconds = req.Power / perUnit
+		}
+		remark = fmt.Sprintf("即梦视频生成，任务ID：%d，%d积分/秒, %d秒", jobId, perUnit, seconds)
+	case types.JMTaskTypeVirtualHuman:
+		seconds := 0
+		if perUnit > 0 {
+			seconds = req.Power / perUnit
+		}
+		remark = fmt.Sprintf("即梦数字人视频生成，任务ID：%d，%d积分/秒, %d秒", jobId, perUnit, seconds)
+	case types.JMTaskTypeActionTransfer:
+		seconds := 0
+		if perUnit > 0 {
+			seconds = req.Power / perUnit
+		}
+		remark = fmt.Sprintf("即梦视频动作迁移，任务ID：%d，%d积分/秒, %d秒", jobId, perUnit, seconds)
+	}
+	return remark
 }
 
 // Jobs 获取任务列表
@@ -287,17 +176,13 @@ func (h *JimengHandler) Jobs(c *gin.Context) {
 
 	switch req.Filter {
 	case "image":
-		query = query.Where("type IN (?)", []model.JMTaskType{
+		query = query.Where("type = ?", types.JMTaskTypeImage)
-			model.JMTaskTypeTextToImage,
-			model.JMTaskTypeImageToImage,
-			model.JMTaskTypeImageEdit,
-			model.JMTaskTypeImageEffects,
-		})
 	case "video":
-		query = query.Where("type IN (?)", []model.JMTaskType{
+		query = query.Where("type = ?", types.JMTaskTypeVideo)
-			model.JMTaskTypeTextToVideo,
+	case "virtual_human":
-			model.JMTaskTypeImageToVideo,
+		query = query.Where("type = ?", types.JMTaskTypeVirtualHuman)
-		})
+	case "action_transfer":
+		query = query.Where("type = ?", types.JMTaskTypeActionTransfer)
 	}
 
 	if len(req.Ids) > 0 {
@@ -357,7 +242,7 @@ func (h *JimengHandler) Remove(c *gin.Context) {
 	}
 
 	// 正在运行中的任务不能删除
-	if job.Status == model.JMTaskStatusGenerating || job.Status == model.JMTaskStatusInQueue {
+	if job.Status == types.JMTaskStatusGenerating || job.Status == types.JMTaskStatusInQueue {
 		resp.ERROR(c, "正在运行中的任务不能删除，否则无法退回算力")
 		return
 	}
@@ -370,10 +255,11 @@ func (h *JimengHandler) Remove(c *gin.Context) {
 	}
 
 	// 失败任务删除后退回算力
-	if job.Status != model.JMTaskStatusFailed {
+	if job.Status == types.JMTaskStatusFailed {
+		logger.Infof("delete jimeng job failed, refund power: %d", job.Power)
 		err = h.userService.IncreasePower(user.Id, job.Power, model.PowerLog{
 			Type:   types.PowerRefund,
-			Model:  "jimeng",
+			Model:  job.ReqKey,
 			Remark: fmt.Sprintf("删除任务，退回%d算力", job.Power),
 		})
 		if err != nil {
@@ -411,13 +297,13 @@ func (h *JimengHandler) Retry(c *gin.Context) {
 	}
 
 	// 只有失败的任务才能重试
-	if job.Status != model.JMTaskStatusFailed {
+	if job.Status != types.JMTaskStatusFailed {
 		resp.ERROR(c, "只有失败的任务才能重试")
 		return
 	}
 
 	// 重置任务状态
-	if err := h.jimengService.UpdateJobStatus(uint(jobId), model.JMTaskStatusInQueue, ""); err != nil {
+	if err := h.jimengService.UpdateJobStatus(uint(jobId), types.JMTaskStatusInQueue, ""); err != nil {
 		logger.Errorf("reset job status failed: %v", err)
 		resp.ERROR(c, "重置任务状态失败")
 		return
@@ -433,25 +319,49 @@ func (h *JimengHandler) Retry(c *gin.Context) {
 	resp.SUCCESS(c, gin.H{"message": "重试任务已提交"})
 }
 
-// getPowerFromConfig 从配置中获取指定类型的算力消耗
+func (h *JimengHandler) getTaskPower(req types.JimengTaskRequest) (int, error) {
-func (h *JimengHandler) getPowerFromConfig(taskType model.JMTaskType) int {
+	logger.Debugf("getTaskPower req: %+v", req)
 	config := h.App.SysConfig.Jimeng
-
+	basePower, ok := config.Powers[req.ReqKey]
-	switch taskType {
+	if !ok || basePower <= 0 {
-	case model.JMTaskTypeTextToImage:
+		return 0, errors.New("未配置模型积分或配置不合法")
-		return config.Power.TextToImage
+	}
-	case model.JMTaskTypeImageToImage:
+	switch req.TaskType {
-		return config.Power.ImageToImage
+	case types.JMTaskTypeImage:
-	case model.JMTaskTypeImageEdit:
+		return basePower, nil
-		return config.Power.ImageEdit
+	case types.JMTaskTypeVideo:
-	case model.JMTaskTypeImageEffects:
+		if req.Duration == 0 {
-		return config.Power.ImageEffects
+			return 0, errors.New("视频时长不能为0")
-	case model.JMTaskTypeTextToVideo:
+		}
-		return config.Power.TextToVideo
+		return basePower * req.Duration, nil
-	case model.JMTaskTypeImageToVideo:
+	case types.JMTaskTypeVirtualHuman:
-		return config.Power.ImageToVideo
+		if req.AudioURL == "" {
+			return 0, errors.New("音频URL不能为空")
+		}
+		audioDuration, err := utils.AudioDurationFromURL(req.AudioURL)
+		if err != nil {
+			return 0, err
+		}
+		seconds := int(audioDuration.Seconds())
+		if seconds <= 0 {
+			return 0, errors.New("音频时长无效")
+		}
+		return basePower * seconds, nil
+	case types.JMTaskTypeActionTransfer:
+		if req.VideoURL == "" {
+			return 0, errors.New("视频URL不能为空")
+		}
+		videoDuration, err := utils.VideoDurationMP4FromURL(req.VideoURL)
+		if err != nil {
+			return 0, err
+		}
+		seconds := int(videoDuration.Seconds())
+		if seconds <= 0 {
+			return 0, errors.New("视频时长无效")
+		}
+		return basePower * seconds, nil
 	default:
-		return 10
+		return 0, errors.New("任务类型不支持")
 	}
 }
 
@@ -459,11 +369,6 @@ func (h *JimengHandler) getPowerFromConfig(taskType model.JMTaskType) int {
 func (h *JimengHandler) GetPowerConfig(c *gin.Context) {
 	config := h.App.SysConfig.Jimeng
 	resp.SUCCESS(c, gin.H{
-		"text_to_image":  config.Power.TextToImage,
+		"powers": config.Powers,
-		"image_to_image": config.Power.ImageToImage,
-		"image_edit":     config.Power.ImageEdit,
-		"image_effects":  config.Power.ImageEffects,
-		"text_to_video":  config.Power.TextToVideo,
-		"image_to_video": config.Power.ImageToVideo,
 	})
 }

api/res/certs/alipay/alipayPublicCert.crt

@@ -1,38 +0,0 @@
------BEGIN CERTIFICATE-----
-MIIDszCCApugAwIBAgIQICMRB0rBU2/rZJbfJGMYIzANBgkqhkiG9w0BAQsFADCBkTELMAkGA1UE
-BhMCQ04xGzAZBgNVBAoMEkFudCBGaW5hbmNpYWwgdGVzdDElMCMGA1UECwwcQ2VydGlmaWNhdGlv
-biBBdXRob3JpdHkgdGVzdDE+MDwGA1UEAww1QW50IEZpbmFuY2lhbCBDZXJ0aWZpY2F0aW9uIEF1
-dGhvcml0eSBDbGFzcyAyIFIxIHRlc3QwHhcNMjMxMTA3MDYzNTQxWhcNMjQxMTA2MDYzNTQxWjCB
-hDELMAkGA1UEBhMCQ04xHzAdBgNVBAoMFm1ib25meTkwMTVAc2FuZGJveC5jb20xDzANBgNVBAsM
-BkFsaXBheTFDMEEGA1UEAww65pSv5LuY5a6dKOS4reWbvSnnvZHnu5zmioDmnK/mnInpmZDlhazl
-j7gtMjA4ODcyMTAyMDc1MDU4MTCCASIwDQYJKoZIhvcNAQEBBQADggEPADCCAQoCggEBAKsoKcw5
-sxaiyV7mpWzDtnQ1K518eQLP0+dJlZAf06aBep/Aj9DIqrba/k7DHt8dKQvILMLAMpN1+2IRxbaO
-yxMa/laj3lZ1eHrB6F077O3D62oHcE3noZtXL0N1zZAxpmkNmYIHeLZS2oLMS4ANu47O/wpDC7BV
-HjdpZugtdPJ4mxdCpM9GDdLs7W4s5QI4PUPK4skFNMFoKI+0cYP/9ju87UP//IHC/K510GWNl+Gn
-Cvgag3AmiIB0utJNsGhxm6zT1T9tUWjW9iz/BxBKiPatsCX9VpPQzGnW7ZonRQtiZSokIlP2IPvl
-H5DcwpWUz3/LUY0SmKxnKOEYeOOqCW8CAwEAAaMSMBAwDgYDVR0PAQH/BAQDAgTwMA0GCSqGSIb3
-DQEBCwUAA4IBAQAtgxF2EzjOndEFxBUD9tFwcSt6XKGggOp52oft1pvynPg4ALTLafOtfEPDrFBH
-PwpYrSu9s9C8NJtaA2HrlCfBjIuwEFTXiN+HPvS0SwSPKt9AXEiTcOF8vDcGamEen8QI4fo5Jia7
-2VRKkerkww5/+FzSaVO7ZUKuL80M1QJStmAZc8kPPwdYOTTW2bGf8BcmSDL6SPElBkt7tCCRd4sn
-+jq4cZ0yb2i77rBZCwHcTvfTqIBblPwLv4uGvg3+83BxIB5w6Kqp06bKEAPmobFY5IVHa+ON0/qi
-BXxXr+WQ3piKRVQEN64+PTAjSc67Ix1umvpLl3Ko6Ry7NJmpDcUn
------END CERTIFICATE-----
------BEGIN CERTIFICATE-----
-MIIDszCCApugAwIBAgIQIBkIGbgVxq210KxLJ+YA/TANBgkqhkiG9w0BAQsFADCBhDELMAkGA1UE
-BhMCQ04xFjAUBgNVBAoMDUFudCBGaW5hbmNpYWwxJTAjBgNVBAsMHENlcnRpZmljYXRpb24gQXV0
-aG9yaXR5IHRlc3QxNjA0BgNVBAMMLUFudCBGaW5hbmNpYWwgQ2VydGlmaWNhdGlvbiBBdXRob3Jp
-dHkgUjEgdGVzdDAeFw0xOTA4MTkxMTE2MDBaFw0yNDA4MDExMTE2MDBaMIGRMQswCQYDVQQGEwJD
-TjEbMBkGA1UECgwSQW50IEZpbmFuY2lhbCB0ZXN0MSUwIwYDVQQLDBxDZXJ0aWZpY2F0aW9uIEF1
-dGhvcml0eSB0ZXN0MT4wPAYDVQQDDDVBbnQgRmluYW5jaWFsIENlcnRpZmljYXRpb24gQXV0aG9y
-aXR5IENsYXNzIDIgUjEgdGVzdDCCASIwDQYJKoZIhvcNAQEBBQADggEPADCCAQoCggEBAMh4FKYO
-ZyRQHD6eFbPKZeSAnrfjfU7xmS9Yoozuu+iuqZlb6Z0SPLUqqTZAFZejOcmr07ln/pwZxluqplxC
-5+B48End4nclDMlT5HPrDr3W0frs6Xsa2ZNcyil/iKNB5MbGll8LRAxntsKvZZj6vUTMb705gYgm
-VUMILwi/ZxKTQqBtkT/kQQ5y6nOZsj7XI5rYdz6qqOROrpvS/d7iypdHOMIM9Iz9DlL1mrCykbBi
-t25y+gTeXmuisHUwqaRpwtCGK4BayCqxRGbNipe6W73EK9lBrrzNtTr9NaysesT/v+l25JHCL9tG
-wpNr1oWFzk4IHVOg0ORiQ6SUgxZUTYcCAwEAAaMSMBAwDgYDVR0PAQH/BAQDAgTwMA0GCSqGSIb3
-DQEBCwUAA4IBAQBWThEoIaQoBX2YeRY/I8gu6TYnFXtyuCljANnXnM38ft+ikhE5mMNgKmJYLHvT
-yWWWgwHoSAWEuml7EGbE/2AK2h3k0MdfiWLzdmpPCRG/RJHk6UB1pMHPilI+c0MVu16OPpKbg5Vf
-LTv7dsAB40AzKsvyYw88/Ezi1osTXo6QQwda7uefvudirtb8FcQM9R66cJxl3kt1FXbpYwheIm/p
-j1mq64swCoIYu4NrsUYtn6CV542DTQMI5QdXkn+PzUUly8F6kDp+KpMNd0avfWNL5+O++z+F5Szy
-1CPta1D7EQ/eYmMP+mOQ35oifWIoFCpN6qQVBS/Hob1J/UUyg7BW
------END CERTIFICATE-----

api/res/certs/alipay/alipayRootCert.crt

@@ -1,88 +0,0 @@
------BEGIN CERTIFICATE-----
-MIIBszCCAVegAwIBAgIIaeL+wBcKxnswDAYIKoEcz1UBg3UFADAuMQswCQYDVQQG
-EwJDTjEOMAwGA1UECgwFTlJDQUMxDzANBgNVBAMMBlJPT1RDQTAeFw0xMjA3MTQw
-MzExNTlaFw00MjA3MDcwMzExNTlaMC4xCzAJBgNVBAYTAkNOMQ4wDAYDVQQKDAVO
-UkNBQzEPMA0GA1UEAwwGUk9PVENBMFkwEwYHKoZIzj0CAQYIKoEcz1UBgi0DQgAE
-MPCca6pmgcchsTf2UnBeL9rtp4nw+itk1Kzrmbnqo05lUwkwlWK+4OIrtFdAqnRT
-V7Q9v1htkv42TsIutzd126NdMFswHwYDVR0jBBgwFoAUTDKxl9kzG8SmBcHG5Yti
-W/CXdlgwDAYDVR0TBAUwAwEB/zALBgNVHQ8EBAMCAQYwHQYDVR0OBBYEFEwysZfZ
-MxvEpgXBxuWLYlvwl3ZYMAwGCCqBHM9VAYN1BQADSAAwRQIgG1bSLeOXp3oB8H7b
-53W+CKOPl2PknmWEq/lMhtn25HkCIQDaHDgWxWFtnCrBjH16/W3Ezn7/U/Vjo5xI
-pDoiVhsLwg==
------END CERTIFICATE-----
-
------BEGIN CERTIFICATE-----
-MIIF0zCCA7ugAwIBAgIIH8+hjWpIDREwDQYJKoZIhvcNAQELBQAwejELMAkGA1UE
-BhMCQ04xFjAUBgNVBAoMDUFudCBGaW5hbmNpYWwxIDAeBgNVBAsMF0NlcnRpZmlj
-YXRpb24gQXV0aG9yaXR5MTEwLwYDVQQDDChBbnQgRmluYW5jaWFsIENlcnRpZmlj
-YXRpb24gQXV0aG9yaXR5IFIxMB4XDTE4MDMyMTEzNDg0MFoXDTM4MDIyODEzNDg0
-MFowejELMAkGA1UEBhMCQ04xFjAUBgNVBAoMDUFudCBGaW5hbmNpYWwxIDAeBgNV
-BAsMF0NlcnRpZmljYXRpb24gQXV0aG9yaXR5MTEwLwYDVQQDDChBbnQgRmluYW5j
-aWFsIENlcnRpZmljYXRpb24gQXV0aG9yaXR5IFIxMIICIjANBgkqhkiG9w0BAQEF
-AAOCAg8AMIICCgKCAgEAtytTRcBNuur5h8xuxnlKJetT65cHGemGi8oD+beHFPTk
-rUTlFt9Xn7fAVGo6QSsPb9uGLpUFGEdGmbsQ2q9cV4P89qkH04VzIPwT7AywJdt2
-xAvMs+MgHFJzOYfL1QkdOOVO7NwKxH8IvlQgFabWomWk2Ei9WfUyxFjVO1LVh0Bp
-dRBeWLMkdudx0tl3+21t1apnReFNQ5nfX29xeSxIhesaMHDZFViO/DXDNW2BcTs6
-vSWKyJ4YIIIzStumD8K1xMsoaZBMDxg4itjWFaKRgNuPiIn4kjDY3kC66Sl/6yTl
-YUz8AybbEsICZzssdZh7jcNb1VRfk79lgAprm/Ktl+mgrU1gaMGP1OE25JCbqli1
-Pbw/BpPynyP9+XulE+2mxFwTYhKAwpDIDKuYsFUXuo8t261pCovI1CXFzAQM2w7H
-DtA2nOXSW6q0jGDJ5+WauH+K8ZSvA6x4sFo4u0KNCx0ROTBpLif6GTngqo3sj+98
-SZiMNLFMQoQkjkdN5Q5g9N6CFZPVZ6QpO0JcIc7S1le/g9z5iBKnifrKxy0TQjtG
-PsDwc8ubPnRm/F82RReCoyNyx63indpgFfhN7+KxUIQ9cOwwTvemmor0A+ZQamRe
-9LMuiEfEaWUDK+6O0Gl8lO571uI5onYdN1VIgOmwFbe+D8TcuzVjIZ/zvHrAGUcC
-AwEAAaNdMFswCwYDVR0PBAQDAgEGMAwGA1UdEwQFMAMBAf8wHQYDVR0OBBYEFF90
-tATATwda6uWx2yKjh0GynOEBMB8GA1UdIwQYMBaAFF90tATATwda6uWx2yKjh0Gy
-nOEBMA0GCSqGSIb3DQEBCwUAA4ICAQCVYaOtqOLIpsrEikE5lb+UARNSFJg6tpkf
-tJ2U8QF/DejemEHx5IClQu6ajxjtu0Aie4/3UnIXop8nH/Q57l+Wyt9T7N2WPiNq
-JSlYKYbJpPF8LXbuKYG3BTFTdOVFIeRe2NUyYh/xs6bXGr4WKTXb3qBmzR02FSy3
-IODQw5Q6zpXj8prYqFHYsOvGCEc1CwJaSaYwRhTkFedJUxiyhyB5GQwoFfExCVHW
-05ZFCAVYFldCJvUzfzrWubN6wX0DD2dwultgmldOn/W/n8at52mpPNvIdbZb2F41
-T0YZeoWnCJrYXjq/32oc1cmifIHqySnyMnavi75DxPCdZsCOpSAT4j4lAQRGsfgI
-kkLPGQieMfNNkMCKh7qjwdXAVtdqhf0RVtFILH3OyEodlk1HYXqX5iE5wlaKzDop
-PKwf2Q3BErq1xChYGGVS+dEvyXc/2nIBlt7uLWKp4XFjqekKbaGaLJdjYP5b2s7N
-1dM0MXQ/f8XoXKBkJNzEiM3hfsU6DOREgMc1DIsFKxfuMwX3EkVQM1If8ghb6x5Y
-jXayv+NLbidOSzk4vl5QwngO/JYFMkoc6i9LNwEaEtR9PhnrdubxmrtM+RjfBm02
-77q3dSWFESFQ4QxYWew4pHE0DpWbWy/iMIKQ6UZ5RLvB8GEcgt8ON7BBJeMc+Dyi
-kT9qhqn+lw==
------END CERTIFICATE-----
-
------BEGIN CERTIFICATE-----
-MIICiDCCAgygAwIBAgIIQX76UsB/30owDAYIKoZIzj0EAwMFADB6MQswCQYDVQQG
-EwJDTjEWMBQGA1UECgwNQW50IEZpbmFuY2lhbDEgMB4GA1UECwwXQ2VydGlmaWNh
-dGlvbiBBdXRob3JpdHkxMTAvBgNVBAMMKEFudCBGaW5hbmNpYWwgQ2VydGlmaWNh
-dGlvbiBBdXRob3JpdHkgRTEwHhcNMTkwNDI4MTYyMDQ0WhcNNDkwNDIwMTYyMDQ0
-WjB6MQswCQYDVQQGEwJDTjEWMBQGA1UECgwNQW50IEZpbmFuY2lhbDEgMB4GA1UE
-CwwXQ2VydGlmaWNhdGlvbiBBdXRob3JpdHkxMTAvBgNVBAMMKEFudCBGaW5hbmNp
-YWwgQ2VydGlmaWNhdGlvbiBBdXRob3JpdHkgRTEwdjAQBgcqhkjOPQIBBgUrgQQA
-IgNiAASCCRa94QI0vR5Up9Yr9HEupz6hSoyjySYqo7v837KnmjveUIUNiuC9pWAU
-WP3jwLX3HkzeiNdeg22a0IZPoSUCpasufiLAnfXh6NInLiWBrjLJXDSGaY7vaokt
-rpZvAdmjXTBbMAsGA1UdDwQEAwIBBjAMBgNVHRMEBTADAQH/MB0GA1UdDgQWBBRZ
-4ZTgDpksHL2qcpkFkxD2zVd16TAfBgNVHSMEGDAWgBRZ4ZTgDpksHL2qcpkFkxD2
-zVd16TAMBggqhkjOPQQDAwUAA2gAMGUCMQD4IoqT2hTUn0jt7oXLdMJ8q4vLp6sg
-wHfPiOr9gxreb+e6Oidwd2LDnC4OUqCWiF8CMAzwKs4SnDJYcMLf2vpkbuVE4dTH
-Rglz+HGcTLWsFs4KxLsq7MuU+vJTBUeDJeDjdA==
------END CERTIFICATE-----
-
------BEGIN CERTIFICATE-----
-MIIDxTCCAq2gAwIBAgIUEMdk6dVgOEIS2cCP0Q43P90Ps5YwDQYJKoZIhvcNAQEF
-BQAwajELMAkGA1UEBhMCQ04xEzARBgNVBAoMCmlUcnVzQ2hpbmExHDAaBgNVBAsM
-E0NoaW5hIFRydXN0IE5ldHdvcmsxKDAmBgNVBAMMH2lUcnVzQ2hpbmEgQ2xhc3Mg
-MiBSb290IENBIC0gRzMwHhcNMTMwNDE4MDkzNjU2WhcNMzMwNDE4MDkzNjU2WjBq
-MQswCQYDVQQGEwJDTjETMBEGA1UECgwKaVRydXNDaGluYTEcMBoGA1UECwwTQ2hp
-bmEgVHJ1c3QgTmV0d29yazEoMCYGA1UEAwwfaVRydXNDaGluYSBDbGFzcyAyIFJv
-b3QgQ0EgLSBHMzCCASIwDQYJKoZIhvcNAQEBBQADggEPADCCAQoCggEBAOPPShpV
-nJbMqqCw6Bz1kehnoPst9pkr0V9idOwU2oyS47/HjJXk9Rd5a9xfwkPO88trUpz5
-4GmmwspDXjVFu9L0eFaRuH3KMha1Ak01citbF7cQLJlS7XI+tpkTGHEY5pt3EsQg
-wykfZl/A1jrnSkspMS997r2Gim54cwz+mTMgDRhZsKK/lbOeBPpWtcFizjXYCqhw
-WktvQfZBYi6o4sHCshnOswi4yV1p+LuFcQ2ciYdWvULh1eZhLxHbGXyznYHi0dGN
-z+I9H8aXxqAQfHVhbdHNzi77hCxFjOy+hHrGsyzjrd2swVQ2iUWP8BfEQqGLqM1g
-KgWKYfcTGdbPB1MCAwEAAaNjMGEwHQYDVR0OBBYEFG/oAMxTVe7y0+408CTAK8hA
-uTyRMB8GA1UdIwQYMBaAFG/oAMxTVe7y0+408CTAK8hAuTyRMA8GA1UdEwEB/wQF
-MAMBAf8wDgYDVR0PAQH/BAQDAgEGMA0GCSqGSIb3DQEBBQUAA4IBAQBLnUTfW7hp
-emMbuUGCk7RBswzOT83bDM6824EkUnf+X0iKS95SUNGeeSWK2o/3ALJo5hi7GZr3
-U8eLaWAcYizfO99UXMRBPw5PRR+gXGEronGUugLpxsjuynoLQu8GQAeysSXKbN1I
-UugDo9u8igJORYA+5ms0s5sCUySqbQ2R5z/GoceyI9LdxIVa1RjVX8pYOj8JFwtn
-DJN3ftSFvNMYwRuILKuqUYSHc2GPYiHVflDh5nDymCMOQFcFG3WsEuB+EYQPFgIU
-1DHmdZcz7Llx8UOZXX2JupWCYzK1XhJb+r4hK5ncf/w8qGtYlmyJpxk3hr1TfUJX
-Yf4Zr0fJsGuv
------END CERTIFICATE-----

api/res/certs/alipay/appPublicCert.crt

@@ -1,19 +0,0 @@
------BEGIN CERTIFICATE-----
-MIIDmTCCAoGgAwIBAgIQICMRB2LW76yahgdg3IFNPDANBgkqhkiG9w0BAQsFADCBkTELMAkGA1UE
-BhMCQ04xGzAZBgNVBAoMEkFudCBGaW5hbmNpYWwgdGVzdDElMCMGA1UECwwcQ2VydGlmaWNhdGlv
-biBBdXRob3JpdHkgdGVzdDE+MDwGA1UEAww1QW50IEZpbmFuY2lhbCBDZXJ0aWZpY2F0aW9uIEF1
-dGhvcml0eSBDbGFzcyAyIFIxIHRlc3QwHhcNMjMxMTA3MDU0NjE5WhcNMjQxMTExMDU0NjE5WjBr
-MQswCQYDVQQGEwJDTjEfMB0GA1UECgwWbWJvbmZ5OTAxNUBzYW5kYm94LmNvbTEPMA0GA1UECwwG
-QWxpcGF5MSowKAYDVQQDDCEyMDg4NzIxMDIwNzUwNTgxLTkwMjEwMDAxMzE2NTgwMjMwggEiMA0G
-CSqGSIb3DQEBAQUAA4IBDwAwggEKAoIBAQCxihQPf1Q+g9ArgM46shVqL5sbRha/df95D1PsWyEq
-ANmWmG4zZ+ksYDVQrc4KzhSRoi56sm/7TDFYTmM6bW99e/nKW58WxyZB4ie5qA3F4n17psPyDqb8
-IokcQmCphSFDaXQD6AoXoLNtTM0vAI2cWxAgebZ/vsrdj5Ntjt+Rp3NYMCk1i5xovHcfILzLEGbX
-QXoT9fo5AhHotTWa6xHVLPUGY9qwLzQxHzBmvy5ZMfnOfJkm/mDisTSqAUB59F3dzU/1ARVkEZ1w
-Mgb4XohWBw6iurQfbMnH2mIomAAwwZVFv+sXDbL9yMbSMo/SjVsTQprn0Q0EnwLo7nmmOM6HAgMB
-AAGjEjAQMA4GA1UdDwEB/wQEAwIE8DANBgkqhkiG9w0BAQsFAAOCAQEAn3Y4/C1h9R6ONsBqX3/q
-XfHX7yX1FM0Y1x48X3/Yxk6HivAkTukhhhVYVKJsbrbzRqHDp9vhAP/FR6o6pAevaYMmLov0VMXU
-7oAuetgkaYEYkDuNen5/Hpdhqi2vTtdT+q9w8zHJd6MDQ0aoHgIxpLKw5vof2R1N4fwSgNXMiXE5
-kmllKQMem/+on2p+Sj80/2asxryHIGlH87qPzkffv+kIOkZthbTApTFLLjdVri2QHGe8/cc4xy01
-/9iR3IUzNahotT41lJ4bMevBY7XMAS3n5ekyABN/9ZRJqhWdXgmFCRN/u56qd6lDgu7R2M2QUoyc
-LuW5DfgRItKlmUB7sw==
------END CERTIFICATE-----

api/res/certs/alipay/privateKey.txt

@@ -1 +0,0 @@
-MIIEpQIBAAKCAQEAsYoUD39UPoPQK4DOOrIVai+bG0YWv3X/eQ9T7FshKgDZlphuM2fpLGA1UK3OCs4UkaIuerJv+0wxWE5jOm1vfXv5ylufFscmQeInuagNxeJ9e6bD8g6m/CKJHEJgqYUhQ2l0A+gKF6CzbUzNLwCNnFsQIHm2f77K3Y+TbY7fkadzWDApNYucaLx3HyC8yxBm10F6E/X6OQIR6LU1musR1Sz1BmPasC80MR8wZr8uWTH5znyZJv5g4rE0qgFAefRd3c1P9QEVZBGdcDIG+F6IVgcOorq0H2zJx9piKJgAMMGVRb/rFw2y/cjG0jKP0o1bE0Ka59ENBJ8C6O55pjjOhwIDAQABAoIBAFetNfz1R7hbxjlFshMAkVzQR8wvT9qbvl+dtzdZRcaFhu89NecDIP7+QDYor0FcxoGpU0TazDyRQyk2BQD8vHt+9zv9BVLtZLJSqoWgPbUFBi1DjS8EF2ka8RVYnn35NhUhhd7L//ftL88Bh673mfembQ9srDjoEy1Z01feoABAnCMkNFl986DmEwnarvEufXSDIgeN4ioMxha4NvfIPuI0zpVdV1O9sv+SGC+VEWZBtN3GNsaf4zS/f8FVGvTiU/Abz0gSw/iwSPHclDWQDTN3yFHf/tfqlzh0mH0WfhnuOBFWXzK+R7fbnM+asI9ttvzRcfpzgRGXdPcNcOv/6cECgYEA3DVqpi1k8MYfJixju6SG5gfyhM4VFksFmCMaNPgtatDMBKLMTgV/Ej6LXREojcy29uZl83F09pVlpd41eG39ULIPktixA/BqErQ2UaWh6kOxifycpu22Jh0r09hax6UgVrcBrrnCJEjcFsuJlrZvXQSzc3PBxjWy5gjabS5h9iECgYEAzmVAIh2frF01Y95zsLueAhhZwCtPanm6kf7ivR4r1plIX3b2sNRhWGmEHFgaCE6Braa0ogQ73Hd26kw4ZW+D6QMGC/zjCBEzDLLf++SjdVUHiY5AR4WHqXzq1jdAlsVyo9R661oAOp3lhiJVGLNXkHyEfEVPHsaxJh4osYSbX6cCgYEAx32Qx0i6eDFTyLZQB46uMrgiaVN04QRH5iJuvGvUYT8UhGKjaU8rZfDJOh+wOH2rhxMEaz1uc3C2bERY9mfWI4Ob/jFWc7YZsiYWS3Mcsuhubw4tMECLUg39RWZsHw8ls8kIuixIh6yFzhTH6YQOcRswIrhMZG8DScfdcSmiz2ECgYEAkWP1t5KSpkLKl11etcKUXfl1T8+yk9jIOowIgRw92WAFAWq2AH67TCKYM7dEL1HOO9tRJ0hAOt/U3ttuZtYVYBEHM26jJ02mXm2rJrA7DS4mrxmL4lYH6LbcXqZxU0Qnq4zEQgIWYzRTORf6Rfof1uJAGaJhR9bDd4yLMfGt2cUCgYEAo216Y61xOHUTA4AF1eekk+r+uOcQgQDvLXfs9FkDdJLk0mPG48/+eIYpPFnANJ/riF/DWOp8WGEe2IzA9yUFexzDbNQK8ha9kGcxaSAyiCwzjZ/t9/+hScDSV8kNqWSRSisu/YOFleEHbokT6mbLZ+gdqES8mUUanaEBzRQYGxo=

api/service/jimeng/client.go

@@ -1,15 +1,21 @@
 package jimeng
 
 import (
+	"context"
 	"encoding/json"
+	"errors"
 	"fmt"
 	"geekai/core/types"
 	"net/http"
 	"net/url"
 	"strings"
+	"time"
 
 	"github.com/volcengine/volc-sdk-golang/base"
 	"github.com/volcengine/volc-sdk-golang/service/visual"
+	"github.com/volcengine/volcengine-go-sdk/service/arkruntime"
+	"github.com/volcengine/volcengine-go-sdk/service/arkruntime/model"
+	"github.com/volcengine/volcengine-go-sdk/volcengine"
 )
 
 // Client 即梦API客户端
@@ -50,6 +56,22 @@ func (c *Client) UpdateConfig(config types.JimengConfig) error {
 				"Version": []string{"2022-08-31"},
 			},
 		},
+		"CVSubmitTask": {
+			Method: http.MethodPost,
+			Path:   "/",
+			Query: url.Values{
+				"Action":  []string{"CVSubmitTask"},
+				"Version": []string{"2022-08-31"},
+			},
+		},
+		"CVGetResult": {
+			Method: http.MethodPost,
+			Path:   "/",
+			Query: url.Values{
+				"Action":  []string{"CVGetResult"},
+				"Version": []string{"2022-08-31"},
+			},
+		},
 		"CVProcess": {
 			Method: http.MethodPost,
 			Path:   "/",
@@ -71,6 +93,22 @@ func (c *Client) UpdateConfig(config types.JimengConfig) error {
 	return c.testConnection()
 }
 
+// GetErrorMessage 根据错误代码获取对应的错误信息
+func GetErrorMessage(code int) string {
+	if message, exists := errorCodeMessages[code]; exists {
+		return message
+	}
+	return fmt.Sprintf("未知错误代码: %d", code)
+}
+
+// HandleResponseError 处理响应错误，根据错误代码返回详细的错误信息
+func HandleResponseError(code int, message string) error {
+	if code == ECSuccess {
+		return nil
+	}
+	return errors.New(GetErrorMessage(code))
+}
+
 // testConnection 测试即梦AI连接
 func (c *Client) testConnection() error {
 
@@ -80,7 +118,7 @@ func (c *Client) testConnection() error {
 		TaskId: "test_task_id_12345",
 	}
 
-	_, err := c.QueryTask(testReq)
+	_, err := c.QueryTask(testReq, ASyncActionGetResult)
 	// 即使任务不存在，只要不是认证错误就说明连接正常
 	if err != nil {
 		// 检查是否是认证错误
@@ -94,7 +132,7 @@ func (c *Client) testConnection() error {
 }
 
 // SubmitTask 提交异步任务
-func (c *Client) SubmitTask(req *SubmitTaskRequest) (*SubmitTaskResponse, error) {
+func (c *Client) SubmitTask(req map[string]any) (*SubmitTaskResponse, error) {
 	// 直接将请求转为map[string]interface{}
 	reqBodyBytes, err := json.Marshal(req)
 	if err != nil {
@@ -103,9 +141,14 @@ func (c *Client) SubmitTask(req *SubmitTaskRequest) (*SubmitTaskResponse, error)
 
 	// 直接使用序列化后的字节
 	jsonBody := reqBodyBytes
+	action := ASyncActionSubmit
+	if v, ok := req["action"]; ok {
+		action = v.(string)
+		delete(req, "action")
+	}
 
 	// 调用SDK的JSON方法
-	respBody, statusCode, err := c.visual.Client.Json("CVSync2AsyncSubmitTask", nil, string(jsonBody))
+	respBody, statusCode, err := c.visual.Client.Json(action, nil, string(jsonBody))
 	if err != nil {
 		return nil, fmt.Errorf("submit task failed (status: %d): %w", statusCode, err)
 	}
@@ -118,11 +161,70 @@ func (c *Client) SubmitTask(req *SubmitTaskRequest) (*SubmitTaskResponse, error)
 		return nil, fmt.Errorf("unmarshal response failed: %w", err)
 	}
 
+	// 检查响应错误代码
+	if err := HandleResponseError(result.Code, result.Message); err != nil {
+		return nil, err
+	}
+
 	return &result, nil
 }
 
+// 识别数字人主体
+func (c *Client) AvatarRecognition(imgUrl string, reqKey string) error {
+	params := map[string]any{
+		"image_url": imgUrl,
+		"req_key":   reqKey,
+	}
+	reqBodyBytes, err := json.Marshal(params)
+	if err != nil {
+		return fmt.Errorf("marshal request failed: %w", err)
+	}
+	// 调用SDK的JSON方法
+	respBody, statusCode, err := c.visual.Client.Json(SyncActionSubmit, nil, string(reqBodyBytes))
+	if err != nil {
+		return fmt.Errorf("submit task failed (status: %d): %w", statusCode, err)
+	}
+
+	// 解析响应
+	var result SubmitTaskResponse
+	if err := json.Unmarshal(respBody, &result); err != nil {
+		return fmt.Errorf("unmarshal response failed: %w", err)
+	}
+
+	// 检查响应错误代码
+	if err := HandleResponseError(result.Code, result.Message); err != nil {
+		return err
+	}
+
+	// 等待任务完成
+	for {
+		resp, err := c.QueryTask(&QueryTaskRequest{
+			ReqKey: reqKey,
+			TaskId: result.Data.TaskId,
+		}, SyncActionGetResult)
+		if err != nil {
+			return fmt.Errorf("query task failed: %w", err)
+		}
+		if resp.Data.Status != types.JMTaskStatusDone {
+			time.Sleep(time.Second * 3)
+			continue
+		}
+		var respData map[string]int
+		if err := json.Unmarshal([]byte(resp.Data.RespData), &respData); err != nil {
+			return fmt.Errorf("unmarshal response failed: %w", err)
+		}
+		logger.Debugf("Jimeng AvatarRecognition Response: %+v", resp)
+		if respData["status"] == 1 {
+			return nil
+		} else {
+			return errors.New("不包含人、类人、拟人等主体")
+		}
+
+	}
+}
+
 // QueryTask 查询任务结果
-func (c *Client) QueryTask(req *QueryTaskRequest) (*QueryTaskResponse, error) {
+func (c *Client) QueryTask(req *QueryTaskRequest, action string) (*QueryTaskResponse, error) {
 	// 序列化请求
 	jsonBody, err := json.Marshal(req)
 	if err != nil {
@@ -130,7 +232,7 @@ func (c *Client) QueryTask(req *QueryTaskRequest) (*QueryTaskResponse, error) {
 	}
 
 	// 调用SDK的JSON方法
-	respBody, statusCode, err := c.visual.Client.Json("CVSync2AsyncGetResult", nil, string(jsonBody))
+	respBody, statusCode, err := c.visual.Client.Json(action, nil, string(jsonBody))
 	if err != nil {
 		return nil, fmt.Errorf("query task failed (status: %d): %w", statusCode, err)
 	}
@@ -143,30 +245,37 @@ func (c *Client) QueryTask(req *QueryTaskRequest) (*QueryTaskResponse, error) {
 		return nil, fmt.Errorf("unmarshal response failed: %w", err)
 	}
 
-	return &result, nil
+	// 检查响应错误代码
-}
+	if err := HandleResponseError(result.Code, result.Message); err != nil {
-
+		return nil, err
-// SubmitSyncTask 提交同步任务（仅用于文生图）
-func (c *Client) SubmitSyncTask(req *SubmitTaskRequest) (*QueryTaskResponse, error) {
-	// 序列化请求
-	jsonBody, err := json.Marshal(req)
-	if err != nil {
-		return nil, fmt.Errorf("marshal request failed: %w", err)
-	}
-
-	// 调用SDK的JSON方法
-	respBody, statusCode, err := c.visual.Client.Json("CVProcess", nil, string(jsonBody))
-	if err != nil {
-		return nil, fmt.Errorf("submit sync task failed (status: %d): %w", statusCode, err)
-	}
-
-	logger.Infof("Jimeng SubmitSyncTask Response: %s", string(respBody))
-
-	// 解析响应，同步任务直接返回结果
-	var result QueryTaskResponse
-	if err := json.Unmarshal(respBody, &result); err != nil {
-		return nil, fmt.Errorf("unmarshal response failed: %w", err)
 	}
 
 	return &result, nil
 }
+
+// SubmitSyncImageTask 提交同步生图任务
+func (c *Client) SubmitSyncImageTask(req types.JimengTaskRequest) (*model.ImagesResponse, error) {
+	// 配置火山引擎访问密钥，目前只支持API Key验证
+	client := arkruntime.NewClientWithApiKey(c.config.ApiKey)
+	// 构造生图请求
+	sequentialImageGeneration := model.SequentialImageGeneration("disabled")
+	generateReq := model.GenerateImagesRequest{
+		Model:                     req.ReqKey,                                               // 模型名称
+		Prompt:                    req.Prompt,                                               // 提示词
+		Size:                      volcengine.String(req.Size),                              // 图片尺寸
+		SequentialImageGeneration: &sequentialImageGeneration,                               // 禁用序列生成
+		ResponseFormat:            volcengine.String(model.GenerateImagesResponseFormatURL), // 响应格式为 URL
+		Watermark:                 volcengine.Bool(false),                                   // 不添加水印
+		OptimizePrompt:            volcengine.Bool(true),                                    // 优化提示词
+	}
+	if len(req.ImageUrls) > 0 {
+		generateReq.Image = req.ImageUrls
+	}
+	// 调用生图 API
+	resp, err := client.GenerateImages(context.Background(), generateReq)
+	if err != nil {
+		return nil, err
+	}
+
+	return &resp, nil
+}

api/service/jimeng/service.go

@@ -4,11 +4,12 @@ import (
 	"context"
 	"encoding/json"
 	"fmt"
-	"strconv"
+	"strings"
 	"time"
 
 	"gorm.io/gorm"
 
+	"geekai/core/types"
 	logger2 "geekai/logger"
 	"geekai/service/oss"
 	"geekai/store"
@@ -95,32 +96,26 @@ func (s *Service) processNextTask() {
 
 	if err := s.ProcessTask(jobId); err != nil {
 		logger.Errorf("process jimeng task failed: job_id=%d, error=%v", jobId, err)
-		s.UpdateJobStatus(jobId, model.JMTaskStatusFailed, err.Error())
+		s.UpdateJobStatus(jobId, types.JMTaskStatusFailed, err.Error())
 	} else {
 		logger.Infof("Jimeng task processed successfully: job_id=%d", jobId)
 	}
 }
 
 // CreateTask 创建任务
-func (s *Service) CreateTask(userId uint, req *CreateTaskRequest) (*model.JimengJob, error) {
+func (s *Service) CreateTask(userId uint, req *types.JimengTaskRequest) (*model.JimengJob, error) {
 	// 生成任务ID
 	taskId := utils.RandString(20)
 
-	// 序列化任务参数
-	paramsJson, err := json.Marshal(req.Params)
-	if err != nil {
-		return nil, fmt.Errorf("marshal task params failed: %w", err)
-	}
-
 	// 创建任务记录
 	job := &model.JimengJob{
 		UserId:    userId,
 		TaskId:    taskId,
-		Type:       req.Type,
+		Type:      req.TaskType,
 		ReqKey:    req.ReqKey,
 		Prompt:    req.Prompt,
-		TaskParams: string(paramsJson),
+		Params:    utils.JsonEncode(req),
-		Status:     model.JMTaskStatusInQueue,
+		Status:    types.JMTaskStatusInQueue,
 		Power:     req.Power,
 		CreatedAt: time.Now(),
 		UpdatedAt: time.Now(),
@@ -148,25 +143,71 @@ func (s *Service) ProcessTask(jobId uint) error {
 	}
 
 	// 更新任务状态为处理中
-	if err := s.UpdateJobStatus(job.Id, model.JMTaskStatusGenerating, ""); err != nil {
+	if err := s.UpdateJobStatus(job.Id, types.JMTaskStatusGenerating, ""); err != nil {
 		return fmt.Errorf("update job status failed: %w", err)
 	}
 
+	// 解析任务参数
+	var req types.JimengTaskRequest
+	err := utils.JsonDecode(job.Params, &req)
+	if err != nil {
+		return fmt.Errorf("parse task params failed: %w", err)
+	}
+
 	// 构建请求并提交任务
-	req, err := s.buildTaskRequest(&job)
+	params, err := s.buildTaskRequest(&req)
 	if err != nil {
 		return s.handleTaskError(job.Id, fmt.Sprintf("build task request failed: %v", err))
 	}
 
-	logger.Infof("提交即梦任务: %+v", req)
+	// 数字人任务，先识别主体
+	if req.TaskType == types.JMTaskTypeVirtualHuman {
+		if err := s.client.AvatarRecognition(req.ImageUrls[0], req.RecognizeKey); err != nil {
+			return s.handleTaskError(job.Id, fmt.Sprintf("avatar recognition failed: %v", err))
+		}
+	}
 
-	// 提交异步任务
+	// 同步任务 ，后台执行
-	resp, err := s.client.SubmitTask(req)
+	if req.ReqKey == DoubaoSeedream40ReqKey {
+		go func() {
+			resp, err := s.client.SubmitSyncImageTask(req)
+			if err != nil {
+				_ = s.handleTaskError(job.Id, fmt.Sprintf("submit task failed: %v", err))
+				return
+			}
+			logger.Infof("同步任务提交成功: %+v", resp)
+			// 更新原始数据
+			rawData, _ := json.Marshal(resp)
+			updates := map[string]any{
+				"raw_data": string(rawData),
+			}
+			if resp.Error != nil {
+				updates["status"] = types.JMTaskStatusFailed
+				updates["err_msg"] = resp.Error.Message
+				s.db.Model(&model.JimengJob{}).Where("id = ?", job.Id).Updates(updates)
+				return
+			}
+
+			// 更新任务状态
+			updates["status"] = types.JMTaskStatusSuccess
+			// 下载图片
+			imgUrl, err := s.uploader.GetUploadHandler().PutUrlFile(*resp.Data[0].Url, ".png", false)
+			if err == nil {
+				updates["img_url"] = imgUrl
+			}
+			s.db.Model(&model.JimengJob{}).Where("id = ?", job.Id).Updates(updates)
+		}()
+		return nil
+	}
+
+	logger.Debugf("提交即梦任务: %+v", params)
+	// 异步任务 ，前台执行
+	resp, err := s.client.SubmitTask(params)
 	if err != nil {
 		return s.handleTaskError(job.Id, fmt.Sprintf("submit task failed: %v", err))
 	}
 
-	if resp.Code != 10000 {
+	if resp.Code != CodeSuccess {
 		return s.handleTaskError(job.Id, fmt.Sprintf("submit task failed: %s", resp.Message))
 	}
 
@@ -184,172 +225,51 @@ func (s *Service) ProcessTask(jobId uint) error {
 }
 
 // buildTaskRequest 构建任务请求（统一的参数解析）
-func (s *Service) buildTaskRequest(job *model.JimengJob) (*SubmitTaskRequest, error) {
+func (s *Service) buildTaskRequest(req *types.JimengTaskRequest) (map[string]any, error) {
-	// 解析任务参数
 	var params map[string]any
-	if err := json.Unmarshal([]byte(job.TaskParams), &params); err != nil {
+	err := utils.JsonDecode(utils.JsonEncode(req), &params)
+	if err != nil {
 		return nil, fmt.Errorf("parse task params failed: %w", err)
 	}
-
+	// 把 size 转成 width 和 height
-	// 构建基础请求
+	if size, ok := params["size"]; ok {
-	req := &SubmitTaskRequest{
+		if sizeStr, ok := size.(string); ok {
-		ReqKey: job.ReqKey,
+			if strings.Contains(sizeStr, "x") {
-		Prompt: job.Prompt,
+				sizes := strings.Split(sizeStr, "x")
+				params["width"] = sizes[0]
+				params["height"] = sizes[1]
+			}
+		}
+		delete(params, "size")
 	}
 
-	// 根据任务类型设置特定参数
+	// duration 转成 frames
-	switch job.Type {
+	if duration, ok := params["duration"]; ok {
-	case model.JMTaskTypeTextToImage:
+		if secs, ok := duration.(int); ok {
-		s.setTextToImageParams(req, params)
+			params["frames"] = secs*24 + 1
-	case model.JMTaskTypeImageToImage:
+		}
-		s.setImageToImageParams(req, params)
+		delete(params, "duration")
-	case model.JMTaskTypeImageEdit:
-		s.setImageEditParams(req, params)
-	case model.JMTaskTypeImageEffects:
-		s.setImageEffectsParams(req, params)
-	case model.JMTaskTypeTextToVideo:
-		s.setTextToVideoParams(req, params)
-	case model.JMTaskTypeImageToVideo:
-		s.setImageToVideoParams(req, params)
-	default:
-		return nil, fmt.Errorf("unsupported task type: %s", job.Type)
 	}
 
-	return req, nil
+	// 单独处理图片特效任务
+	if req.ReqKey == ImageEffectReqKey {
+		params["image_input1"] = req.ImageUrls[0]
+		delete(params, "image_urls")
 	}
 
-// setTextToImageParams 设置文生图参数
+	// 动作迁移，数字人任务参数处理
-func (s *Service) setTextToImageParams(req *SubmitTaskRequest, params map[string]any) {
+	if req.TaskType == types.JMTaskTypeVirtualHuman || req.TaskType == types.JMTaskTypeActionTransfer {
-	if seed, ok := params["seed"]; ok {
+		params["image_url"] = req.ImageUrls[0]
-		if seedVal, err := strconv.ParseInt(fmt.Sprintf("%.0f", seed), 10, 64); err == nil {
+		delete(params, "image_urls")
-			req.Seed = seedVal
-		}
-	}
-	if scale, ok := params["scale"]; ok {
-		if scaleVal, ok := scale.(float64); ok {
-			req.Scale = scaleVal
-		}
-	}
-	if width, ok := params["width"]; ok {
-		if widthVal, ok := width.(float64); ok {
-			req.Width = int(widthVal)
-		}
-	}
-	if height, ok := params["height"]; ok {
-		if heightVal, ok := height.(float64); ok {
-			req.Height = int(heightVal)
-		}
-	}
-	if usePreLlm, ok := params["use_pre_llm"]; ok {
-		if usePreLlmVal, ok := usePreLlm.(bool); ok {
-			req.UsePreLLM = usePreLlmVal
-		}
 	}
+	if req.RecognizeKey != "" {
+		delete(params, "recognize_key")
 	}
 
-// setImageToImageParams 设置图生图参数
+	// 删除多余参数，剩下的就是各个任务自己专有参数了
-func (s *Service) setImageToImageParams(req *SubmitTaskRequest, params map[string]any) {
+	delete(params, "type")
-	if imageInput, ok := params["image_input"].(string); ok {
+	delete(params, "power")
-		req.ImageInput = imageInput
+	return params, nil
-	}
-	if gpen, ok := params["gpen"]; ok {
-		if gpenVal, ok := gpen.(float64); ok {
-			req.Gpen = gpenVal
-		}
-	}
-	if skin, ok := params["skin"]; ok {
-		if skinVal, ok := skin.(float64); ok {
-			req.Skin = skinVal
-		}
-	}
-	if skinUnifi, ok := params["skin_unifi"]; ok {
-		if skinUnifiVal, ok := skinUnifi.(float64); ok {
-			req.SkinUnifi = skinUnifiVal
-		}
-	}
-	if genMode, ok := params["gen_mode"].(string); ok {
-		req.GenMode = genMode
-	}
-	s.setCommonParams(req, params) // 复用通用参数
-}
-
-// setImageEditParams 设置图像编辑参数
-func (s *Service) setImageEditParams(req *SubmitTaskRequest, params map[string]any) {
-	if imageUrls, ok := params["image_urls"].([]any); ok {
-		for _, url := range imageUrls {
-			if urlStr, ok := url.(string); ok {
-				req.ImageUrls = append(req.ImageUrls, urlStr)
-			}
-		}
-	}
-	if binaryData, ok := params["binary_data_base64"].([]any); ok {
-		for _, data := range binaryData {
-			if dataStr, ok := data.(string); ok {
-				req.BinaryDataBase64 = append(req.BinaryDataBase64, dataStr)
-			}
-		}
-	}
-	if scale, ok := params["scale"]; ok {
-		if scaleVal, ok := scale.(float64); ok {
-			req.Scale = scaleVal
-		}
-	}
-	s.setCommonParams(req, params)
-}
-
-// setImageEffectsParams 设置图像特效参数
-func (s *Service) setImageEffectsParams(req *SubmitTaskRequest, params map[string]any) {
-	if imageInput1, ok := params["image_input1"].(string); ok {
-		req.ImageInput1 = imageInput1
-	}
-	if templateId, ok := params["template_id"].(string); ok {
-		req.TemplateId = templateId
-	}
-	if width, ok := params["width"]; ok {
-		if widthVal, ok := width.(float64); ok {
-			req.Width = int(widthVal)
-		}
-	}
-	if height, ok := params["height"]; ok {
-		if heightVal, ok := height.(float64); ok {
-			req.Height = int(heightVal)
-		}
-	}
-}
-
-// setTextToVideoParams 设置文生视频参数
-func (s *Service) setTextToVideoParams(req *SubmitTaskRequest, params map[string]any) {
-	if aspectRatio, ok := params["aspect_ratio"].(string); ok {
-		req.AspectRatio = aspectRatio
-	}
-	s.setCommonParams(req, params)
-}
-
-// setImageToVideoParams 设置图生视频参数
-func (s *Service) setImageToVideoParams(req *SubmitTaskRequest, params map[string]any) {
-	s.setImageEditParams(req, params) // 复用图像编辑的参数设置
-	if aspectRatio, ok := params["aspect_ratio"].(string); ok {
-		req.AspectRatio = aspectRatio
-	}
-}
-
-// setCommonParams 设置通用参数（seed, width, height等）
-func (s *Service) setCommonParams(req *SubmitTaskRequest, params map[string]any) {
-	if seed, ok := params["seed"]; ok {
-		if seedVal, err := strconv.ParseInt(fmt.Sprintf("%.0f", seed), 10, 64); err == nil {
-			req.Seed = seedVal
-		}
-	}
-	if width, ok := params["width"]; ok {
-		if widthVal, ok := width.(float64); ok {
-			req.Width = int(widthVal)
-		}
-	}
-	if height, ok := params["height"]; ok {
-		if heightVal, ok := height.(float64); ok {
-			req.Height = int(heightVal)
-		}
-	}
 }
 
 // pollTaskStatus 轮询任务状态
@@ -357,7 +277,7 @@ func (s *Service) pollTaskStatus() {
 
 	for {
 		var jobs []model.JimengJob
-		s.db.Where("status IN (?)", []model.JMTaskStatus{model.JMTaskStatusGenerating, model.JMTaskStatusInQueue}).Find(&jobs)
+		s.db.Where("status IN (?)", []types.JMTaskStatus{types.JMTaskStatusGenerating, types.JMTaskStatusInQueue}).Find(&jobs)
 		if len(jobs) == 0 {
 			logger.Debugf("no jimeng task to poll, sleep 10s")
 			time.Sleep(10 * time.Second)
@@ -371,12 +291,17 @@ func (s *Service) pollTaskStatus() {
 				continue
 			}
 
+			// 豆包生图 4.0 是同步任务，不需要轮询
+			if job.ReqKey == DoubaoSeedream40ReqKey {
+				continue
+			}
+
 			// 查询任务状态
 			resp, err := s.client.QueryTask(&QueryTaskRequest{
 				ReqKey:  job.ReqKey,
 				TaskId:  job.TaskId,
 				ReqJson: `{"return_url":true}`,
-			})
+			}, ASyncActionGetResult)
 
 			if err != nil {
 				s.handleTaskError(job.Id, fmt.Sprintf("query task failed: %s", err.Error()))
@@ -387,13 +312,13 @@ func (s *Service) pollTaskStatus() {
 			rawData, _ := json.Marshal(resp)
 			s.db.Model(&model.JimengJob{}).Where("id = ?", job.Id).Update("raw_data", string(rawData))
 
-			if resp.Code != 10000 {
+			if resp.Code != CodeSuccess {
 				s.handleTaskError(job.Id, fmt.Sprintf("query task failed: %s", resp.Message))
 				continue
 			}
 
 			switch resp.Data.Status {
-			case model.JMTaskStatusDone:
+			case types.JMTaskStatusDone:
 				// 判断任务是否成功
 				if resp.Message != "Success" {
 					s.handleTaskError(job.Id, fmt.Sprintf("task failed: %s", resp.Data.AlgorithmBaseResp.StatusMessage))
@@ -402,7 +327,7 @@ func (s *Service) pollTaskStatus() {
 
 				// 任务完成，更新结果
 				updates := map[string]any{
-					"status":     model.JMTaskStatusSuccess,
+					"status":     types.JMTaskStatusSuccess,
 					"updated_at": time.Now(),
 				}
 
@@ -425,15 +350,15 @@ func (s *Service) pollTaskStatus() {
 				}
 
 				s.db.Model(&model.JimengJob{}).Where("id = ?", job.Id).Updates(updates)
-			case model.JMTaskStatusInQueue, model.JMTaskStatusGenerating:
+			case types.JMTaskStatusInQueue, types.JMTaskStatusGenerating:
 				// 任务处理中
-				s.UpdateJobStatus(job.Id, model.JMTaskStatusGenerating, "")
+				s.UpdateJobStatus(job.Id, types.JMTaskStatusGenerating, "")
 
-			case model.JMTaskStatusNotFound:
+			case types.JMTaskStatusNotFound:
 				// 任务未找到
 				s.handleTaskError(job.Id, "task not found")
 
-			case model.JMTaskStatusExpired:
+			case types.JMTaskStatusExpired:
 				continue
 			default:
 				logger.Warnf("unknown task status: %s", resp.Data.Status)
@@ -448,7 +373,7 @@ func (s *Service) pollTaskStatus() {
 }
 
 // UpdateJobStatus 更新任务状态
-func (s *Service) UpdateJobStatus(jobId uint, status model.JMTaskStatus, errMsg string) error {
+func (s *Service) UpdateJobStatus(jobId uint, status types.JMTaskStatus, errMsg string) error {
 	updates := map[string]any{
 		"status":     status,
 		"updated_at": time.Now(),
@@ -462,7 +387,7 @@ func (s *Service) UpdateJobStatus(jobId uint, status model.JMTaskStatus, errMsg
 // handleTaskError 处理任务错误
 func (s *Service) handleTaskError(jobId uint, errMsg string) error {
 	logger.Errorf("Jimeng task error (job_id: %d): %s", jobId, errMsg)
-	return s.UpdateJobStatus(jobId, model.JMTaskStatusFailed, errMsg)
+	return s.UpdateJobStatus(jobId, types.JMTaskStatusFailed, errMsg)
 }
 
 // PushTaskToQueue 推送任务到队列（用于手动重试）
@@ -473,7 +398,7 @@ func (s *Service) PushTaskToQueue(jobId uint) error {
 // GetTaskStats 获取任务统计信息
 func (s *Service) GetTaskStats() (map[string]any, error) {
 	type StatResult struct {
-		Status string `json:"status"`
+		Status types.JMTaskStatus `json:"status"`
 		Count  int64              `json:"count"`
 	}
 
@@ -496,7 +421,7 @@ func (s *Service) GetTaskStats() (map[string]any, error) {
 
 	for _, stat := range stats {
 		result["total"] = result["total"].(int64) + stat.Count
-		result[stat.Status] = stat.Count
+		result[string(stat.Status)] = stat.Count
 	}
 
 	return result, nil

api/service/jimeng/types.go

@@ -1,43 +1,9 @@
 package jimeng
 
-import "geekai/store/model"
+import (
-
+	"geekai/core/types"
-// ReqKey 常量定义
-const (
-	ReqKeyTextToImage          = "high_aes_general_v30l_zt2i" // 文生图
-	ReqKeyImageToImagePortrait = "i2i_portrait_photo"         // 图生图人像写真
-	ReqKeyImageEdit            = "seededit_v3.0"              // 图像编辑
-	ReqKeyImageEffects         = "i2i_multi_style_zx2x"       // 图像特效
-	ReqKeyTextToVideo          = "jimeng_vgfm_t2v_l20"        // 文生视频
-	ReqKeyImageToVideo         = "jimeng_vgfm_i2v_l20"        // 图生视频
 )
 
-// SubmitTaskRequest 提交任务请求
-type SubmitTaskRequest struct {
-	ReqKey string `json:"req_key"`
-	// 文生图参数
-	Prompt    string  `json:"prompt,omitempty"`
-	Seed      int64   `json:"seed,omitempty"`
-	Scale     float64 `json:"scale,omitempty"`
-	Width     int     `json:"width,omitempty"`
-	Height    int     `json:"height,omitempty"`
-	UsePreLLM bool    `json:"use_pre_llm,omitempty"`
-	// 图生图参数
-	ImageInput       string   `json:"image_input,omitempty"`
-	ImageUrls        []string `json:"image_urls,omitempty"`
-	BinaryDataBase64 []string `json:"binary_data_base64,omitempty"`
-	Gpen             float64  `json:"gpen,omitempty"`
-	Skin             float64  `json:"skin,omitempty"`
-	SkinUnifi        float64  `json:"skin_unifi,omitempty"`
-	GenMode          string   `json:"gen_mode,omitempty"`
-	// 图像编辑参数
-	// 图像特效参数
-	ImageInput1 string `json:"image_input1,omitempty"`
-	TemplateId  string `json:"template_id,omitempty"`
-	// 视频生成参数
-	AspectRatio string `json:"aspect_ratio,omitempty"`
-}
-
 // SubmitTaskResponse 提交任务响应
 type SubmitTaskResponse struct {
 	Code        int    `json:"code"`
@@ -73,7 +39,7 @@ type QueryTaskResponse struct {
 		ImageUrls         []string           `json:"image_urls"`
 		VideoUrl          string             `json:"video_url"`
 		RespData          string             `json:"resp_data"`
-		Status            model.JMTaskStatus `json:"status"`
+		Status            types.JMTaskStatus `json:"status"`
 		LlmResult         string             `json:"llm_result"`
 		PeResult          string             `json:"pe_result"`
 		PredictTagsResult string             `json:"predict_tags_result"`
@@ -83,9 +49,73 @@ type QueryTaskResponse struct {
 	} `json:"data"`
 }
 
+const CodeSuccess = 10000
+
+// 即梦AI错误代码常量
+const (
+	// 成功
+	ECSuccess = 10000
+
+	// 请求参数错误 (50200-50215)
+	ECReqInvalidArgs     = 50200 // 参数错误
+	ECReqMissingArgs     = 50201 // 缺少参数
+	ECParseArgs          = 50204 // 参数类型错误/参数缺失
+	ECImageSizeLimited   = 50205 // 图像尺寸超过限制
+	ECImageEmpty         = 50206 // 请求参数中没有获取到图像
+	ECImageDecodeError   = 50207 // 图像解码错误
+	ECVideoEmpty         = 50209 // 请求参数中没有获取到视频
+	ECVideoDecodeError   = 50210 // 视频解码错误
+	ECVideoSizeLimited   = 50211 // 视频尺寸超过限制
+	ECReqBodySizeLimited = 50213 // 请求Body过大
+	ECVideoTimeTooLong   = 50214 // 输入视频时长过大
+	ECRPCProcess         = 50215 // 请求处理失败
+
+	// 算法服务错误 (60102-60208)
+	ECJPFaceDetect      = 60102 // 算法服务需要输入人脸图，但未检测到
+	ECFSLeaderRiskError = 60208 // 输入图片中包含敏感信息，未通过审核
+
+	// 权限和系统错误 (50400-50501)
+	ECAuth        = 50400 // 权限校验失败
+	ECReqMethod   = 50402 // 访问的接口不存在
+	ECReqLimit    = 50429 // 超过调用QPS限制
+	ECInternal    = 50500 // 服务器内部错误
+	ECRPCInternal = 50501 // 服务器内部RPC错误
+)
+
+// 错误代码到错误信息的映射
+var errorCodeMessages = map[int]string{
+	// 成功
+	ECSuccess: "请求成功",
+
+	// 请求参数错误
+	ECReqInvalidArgs:     "参数错误，检查入参及MIME类型",
+	ECReqMissingArgs:     "缺少参数，检查入参及MIME类型",
+	ECParseArgs:          "参数类型错误/参数缺失，检查入参及MIME类型",
+	ECImageSizeLimited:   "图像尺寸超过限制，参考接口文档入参要求部分",
+	ECImageEmpty:         "请求参数中没有获取到图像，检查入参",
+	ECImageDecodeError:   "图像解码错误：没有获取到图像或者通过image_base64参数传递图像是base64解码错误，检查输出图片或检查base64是否错误携带前缀",
+	ECVideoEmpty:         "请求参数中没有获取到视频。输入为视频时可能返回此错误，检查入参",
+	ECVideoDecodeError:   "视频解码错误。输入为视频时可能返回此错误，检查输入视频是否不正确",
+	ECVideoSizeLimited:   "视频尺寸超过限制。输入为视频时可能返回此错误，检查输入视频大小",
+	ECReqBodySizeLimited: "请求Body过大，超出接口限制，检查请求Body大小",
+	ECVideoTimeTooLong:   "输入视频时长过大，检查输入视频时长",
+	ECRPCProcess:         "由于输入的图片、视频、参数等不满足要求，导致请求处理失败。若接口文档中有具体说明，优先参考其具体含义，按照具体服务说明进行检查",
+
+	// 算法服务错误
+	ECJPFaceDetect:      "算法服务需要输入人脸图，但未检测到，检查输入图片是否包含人脸",
+	ECFSLeaderRiskError: "输入图片中包含敏感信息，未通过审核",
+
+	// 权限和系统错误
+	ECAuth:        "权限校验失败，请检查是否已创建应用并开通服务或签名，参考接入指南及快速接入",
+	ECReqMethod:   "访问的接口不存在，检查入参",
+	ECReqLimit:    "超过调用QPS限制，购买QPS增项包",
+	ECInternal:    "服务器内部错误，提工单",
+	ECRPCInternal: "服务器内部RPC错误，提工单",
+}
+
 // CreateTaskRequest 创建任务请求
 type CreateTaskRequest struct {
-	Type      model.JMTaskType `json:"type"`
+	Type      types.JMTaskType `json:"type"`
 	Prompt    string           `json:"prompt"`
 	Params    map[string]any   `json:"params"`
 	ReqKey    string           `json:"req_key"`
@@ -93,53 +123,14 @@ type CreateTaskRequest struct {
 	Power     int              `json:"power,omitempty"`
 }
 
-// LogoInfo 水印信息
-type LogoInfo struct {
-	AddLogo         bool    `json:"add_logo"`
-	Position        int     `json:"position"`
-	Language        int     `json:"language"`
-	Opacity         float64 `json:"opacity"`
-	LogoTextContent string  `json:"logo_text_content"`
-}
-
-// ReqJsonConfig 查询配置
-type ReqJsonConfig struct {
-	ReturnUrl bool      `json:"return_url"`
-	LogoInfo  *LogoInfo `json:"logo_info,omitempty"`
-}
-
-// ImageEffectTemplate 图像特效模板
 const (
-	TemplateIdFelt3DPolaroid             = "felt_3d_polaroid"                        // 毛毡3d拍立得风格
+	ImageEffectReqKey      = "i2i_multi_style_zx2x"
-	TemplateIdMyWorld                    = "my_world"                                // 像素世界风
+	DoubaoSeedream40ReqKey = "doubao-seedream-4-0-250828"
-	TemplateIdMyWorldUniversal           = "my_world_universal"                      // 像素世界-万物通用版
-	TemplateIdPlasticBubbleFigure        = "plastic_bubble_figure"                   // 盲盒玩偶风
-	TemplateIdPlasticBubbleFigureCartoon = "plastic_bubble_figure_cartoon_text"      // 塑料泡罩人偶-文字卡头版
-	TemplateIdFurryDreamDoll             = "furry_dream_doll"                        // 毛绒玩偶风
-	TemplateIdMicroLandscapeMiniWorld    = "micro_landscape_mini_world"              // 迷你世界玩偶风
-	TemplateIdMicroLandscapeProfessional = "micro_landscape_mini_world_professional" // 微型景观小世界-职业版
-	TemplateIdAcrylicOrnaments           = "acrylic_ornaments"                       // 亚克力挂饰
-	TemplateIdFeltKeychain               = "felt_keychain"                           // 毛毡钥匙扣
-	TemplateIdLofiPixelCharacter         = "lofi_pixel_character_mini_card"          // Lofi像素人物小卡
-	TemplateIdAngelFigurine              = "angel_figurine"                          // 天使形象手办
-	TemplateIdLyingInFluffyBelly         = "lying_in_fluffy_belly"                   // 躺在毛茸茸肚皮里
-	TemplateIdGlassBall                  = "glass_ball"                              // 玻璃球
 )
 
-// AspectRatio 视频宽高比
 const (
-	AspectRatio16_9 = "16:9" // 1280×720
+	ASyncActionSubmit    = "CVSync2AsyncSubmitTask" // 异步提交任务
-	AspectRatio9_16 = "9:16" // 720×1280
+	SyncActionSubmit     = "CVSubmitTask"           // 同步提交任务
-	AspectRatio1_1  = "1:1"  // 960×960
+	ASyncActionGetResult = "CVSync2AsyncGetResult"  // 异步获取结果
-	AspectRatio4_3  = "4:3"  // 960×720
+	SyncActionGetResult  = "CVGetResult"            // 同步获取结果
-	AspectRatio3_4  = "3:4"  // 720×960
-	AspectRatio21_9 = "21:9" // 1680×720
-	AspectRatio9_21 = "9:21" // 720×1680
-)
-
-// GenMode 生成模式
-const (
-	GenModeCreative      = "creative"       // 提示词模式
-	GenModeReference     = "reference"      // 全参考模式
-	GenModeReferenceChar = "reference_char" // 人物参考模式
 )

api/service/migration_service.go

@@ -159,8 +159,16 @@ func (s *MigrationService) MigrateConfigContent() error {
 
 // 数据表迁移
 func (s *MigrationService) TableMigration() {
+
+	// v4.2.7 数据表迁移
+	if s.db.Migrator().HasColumn(&model.JimengJob{}, "task_params") {
+		s.db.Migrator().RenameColumn(&model.JimengJob{}, "task_params", "params")
+	}
+
 	// 新数据表
+	if !s.db.Migrator().HasTable(&model.Moderation{}) {
 		s.db.AutoMigrate(&model.Moderation{})
+	}
 
 	// 订单字段整理
 	if s.db.Migrator().HasColumn(&model.Order{}, "pay_type") {

api/service/user_service.go

@@ -57,13 +57,19 @@ func (s *UserService) DecreasePower(userId uint, power int, log model.PowerLog)
 	defer s.lock.Unlock()
 
 	tx := s.db.Begin()
+	var user model.User
+	tx.Where("id", userId).First(&user)
+	if user.Power < power {
+		tx.Rollback()
+		return fmt.Errorf("用户算力不足")
+	}
+
 	err := tx.Model(&model.User{}).Where("id", userId).UpdateColumn("power", gorm.Expr("power - ?", power)).Error
 	if err != nil {
 		tx.Rollback()
 		return fmt.Errorf("扣减算力失败：%v", err)
 	}
-	var user model.User
+
-	tx.Where("id", userId).First(&user)
 	err = tx.Create(&model.PowerLog{
 		UserId:    user.Id,
 		Username:  user.Username,

api/store/model/jimeng_job.go

@@ -1,6 +1,7 @@
 package model
 
 import (
+	"geekai/core/types"
 	"time"
 )
 
@@ -9,46 +10,21 @@ type JimengJob struct {
 	Id        uint               `gorm:"column:id;primaryKey;autoIncrement" json:"id"`
 	UserId    uint               `gorm:"column:user_id;type:int(11);not null;index;comment:用户ID" json:"user_id"`
 	TaskId    string             `gorm:"column:task_id;type:varchar(100);not null;index;comment:任务ID" json:"task_id"`
-	Type       JMTaskType   `gorm:"column:type;type:varchar(50);not null;comment:任务类型" json:"type"`
+	Type      types.JMTaskType   `gorm:"column:type;type:varchar(50);not null;comment:任务类型" json:"type"`
 	ReqKey    string             `gorm:"column:req_key;type:varchar(100);comment:请求Key" json:"req_key"`
 	Prompt    string             `gorm:"column:prompt;type:text;comment:提示词" json:"prompt"`
-	TaskParams string       `gorm:"column:task_params;type:text;comment:任务参数JSON" json:"task_params"`
+	Params    string             `gorm:"column:params;type:text;comment:任务参数JSON" json:"params"`
 	ImgURL    string             `gorm:"column:img_url;type:varchar(1024);comment:图片或封面URL" json:"img_url"`
 	VideoURL  string             `gorm:"column:video_url;type:varchar(1024);comment:视频URL" json:"video_url"`
 	RawData   string             `gorm:"column:raw_data;type:text;comment:原始API响应" json:"raw_data"`
 	Progress  int                `gorm:"column:progress;type:int;default:0;comment:进度百分比" json:"progress"`
-	Status     JMTaskStatus `gorm:"column:status;type:varchar(20);default:'pending';comment:任务状态" json:"status"`
+	Status    types.JMTaskStatus `gorm:"column:status;type:varchar(20);default:'pending';comment:任务状态" json:"status"`
 	ErrMsg    string             `gorm:"column:err_msg;type:varchar(1024);comment:错误信息" json:"err_msg"`
 	Power     int                `gorm:"column:power;type:int(11);default:0;comment:消耗算力" json:"power"`
 	CreatedAt time.Time          `gorm:"column:created_at;type:datetime;not null;comment:创建时间" json:"created_at"`
 	UpdatedAt time.Time          `gorm:"column:updated_at;type:datetime;not null;comment:更新时间" json:"updated_at"`
 }
 
-// JMTaskStatus 任务状态
-type JMTaskStatus string
-
-const (
-	JMTaskStatusInQueue    = JMTaskStatus("in_queue")   // 任务已提交
-	JMTaskStatusGenerating = JMTaskStatus("generating") // 任务处理中
-	JMTaskStatusDone       = JMTaskStatus("done")       // 处理完成
-	JMTaskStatusNotFound   = JMTaskStatus("not_found")  // 任务未找到
-	JMTaskStatusSuccess    = JMTaskStatus("success")    // 任务成功
-	JMTaskStatusFailed     = JMTaskStatus("failed")     // 任务失败
-	JMTaskStatusExpired    = JMTaskStatus("expired")    // 任务过期
-)
-
-// JMTaskType 任务类型
-type JMTaskType string
-
-const (
-	JMTaskTypeTextToImage  = JMTaskType("text_to_image")  // 文生图
-	JMTaskTypeImageToImage = JMTaskType("image_to_image") // 图生图
-	JMTaskTypeImageEdit    = JMTaskType("image_edit")     // 图像编辑
-	JMTaskTypeImageEffects = JMTaskType("image_effects")  // 图像特效
-	JMTaskTypeTextToVideo  = JMTaskType("text_to_video")  // 文生视频
-	JMTaskTypeImageToVideo = JMTaskType("image_to_video") // 图生视频
-)
-
 // TableName 返回数据表名称
 func (JimengJob) TableName() string {
 	return "geekai_jimeng_jobs"

api/store/vo/jimeng_job.go

@@ -1,21 +1,21 @@
 package vo
 
-import "geekai/store/model"
+import "geekai/core/types"
 
 // JimengJob 即梦AI任务VO
 type JimengJob struct {
 	Id        uint               `json:"id"`
 	UserId    uint               `json:"user_id"`
 	TaskId    string             `json:"task_id"`
-	Type       model.JMTaskType   `json:"type"`
+	Type      types.JMTaskType   `json:"type"`
 	ReqKey    string             `json:"req_key"`
 	Prompt    string             `json:"prompt"`
-	TaskParams string             `json:"task_params"`
+	Params    map[string]any     `json:"params"`
 	ImgURL    string             `json:"img_url"`
 	VideoURL  string             `json:"video_url"`
 	RawData   string             `json:"raw_data"`
 	Progress  int                `json:"progress"`
-	Status     model.JMTaskStatus `json:"status"`
+	Status    types.JMTaskStatus `json:"status"`
 	ErrMsg    string             `json:"err_msg"`
 	Power     int                `json:"power"`
 	CreatedAt int64              `json:"created_at"` // 时间戳

api/test/test_test.go

@@ -0,0 +1,10 @@
+package test
+
+import (
+	"fmt"
+	"testing"
+)
+
+func Test(t *testing.T) {
+	fmt.Println("test")
+}

api/utils/media_duration.go

@@ -0,0 +1,817 @@
+package utils
+
+import (
+	"context"
+	"encoding/binary"
+	"errors"
+	"fmt"
+	"io"
+	"net/http"
+	"os"
+	"time"
+)
+
+// AudioDuration returns duration of an audio file.
+// Supported formats: MP3, WAV (auto-detected by header)
+func AudioDuration(path string) (time.Duration, error) {
+	f, err := os.Open(path)
+	if err != nil {
+		return 0, err
+	}
+	defer f.Close()
+
+	// Peek first 12 bytes to detect format
+	head := make([]byte, 12)
+	n, err := io.ReadFull(f, head)
+	if err != nil {
+		return 0, err
+	}
+	if n < 12 {
+		return 0, errors.New("file too small")
+	}
+
+	// WAV: RIFF....WAVE
+	if string(head[0:4]) == "RIFF" && string(head[8:12]) == "WAVE" {
+		if _, err := f.Seek(0, io.SeekStart); err != nil {
+			return 0, err
+		}
+		return wavDuration(f)
+	}
+
+	// MP3 can start with ID3 or frame sync 0xFFEx
+	if string(head[0:3]) == "ID3" || (head[0] == 0xFF && (head[1]&0xE0) == 0xE0) {
+		if _, err := f.Seek(0, io.SeekStart); err != nil {
+			return 0, err
+		}
+		return mp3Duration(f)
+	}
+
+	return 0, errors.New("unsupported audio format")
+}
+
+// AudioDurationFromURL downloads the url to a temp file and returns duration.
+func AudioDurationFromURL(url string) (time.Duration, error) {
+	path, err := fetchURLToTemp(url, 30*time.Second)
+	if err != nil {
+		return 0, err
+	}
+	defer os.Remove(path)
+	return AudioDuration(path)
+}
+
+// VideoDurationMP4 returns duration of an MP4 file (MOV/MP4 base media).
+func VideoDurationMP4(path string) (time.Duration, error) {
+	f, err := os.Open(path)
+	if err != nil {
+		return 0, err
+	}
+	defer f.Close()
+
+	return mp4Duration(f)
+}
+
+// VideoDurationMP4FromURL downloads the url to a temp file and returns duration.
+func VideoDurationMP4FromURL(url string) (time.Duration, error) {
+	path, err := fetchURLToTemp(url, 30*time.Second)
+	if err != nil {
+		return 0, err
+	}
+	defer os.Remove(path)
+	return VideoDurationMP4(path)
+}
+
+// ---------------------- helpers ----------------------
+
+func fetchURLToTemp(url string, timeout time.Duration) (string, error) {
+	ctx, cancel := context.WithTimeout(context.Background(), timeout)
+	defer cancel()
+
+	req, err := http.NewRequestWithContext(ctx, http.MethodGet, url, nil)
+	if err != nil {
+		return "", err
+	}
+	resp, err := http.DefaultClient.Do(req)
+	if err != nil {
+		return "", err
+	}
+	defer resp.Body.Close()
+	if resp.StatusCode != http.StatusOK {
+		return "", fmt.Errorf("http status: %d", resp.StatusCode)
+	}
+
+	tmp, err := os.CreateTemp("", "media-*")
+	if err != nil {
+		return "", err
+	}
+	defer tmp.Close()
+
+	if _, err := io.Copy(tmp, resp.Body); err != nil {
+		path := tmp.Name()
+		_ = os.Remove(path)
+		return "", err
+	}
+	return tmp.Name(), nil
+}
+
+// ---------------------- WAV ----------------------
+
+func wavDuration(r io.ReadSeeker) (time.Duration, error) {
+	// RIFF header already checked outside if needed. We parse chunks to get fmt and data.
+	// WAV little-endian
+	if _, err := r.Seek(0, io.SeekStart); err != nil {
+		return 0, err
+	}
+
+	// Read RIFF header (12 bytes)
+	head := make([]byte, 12)
+	if _, err := io.ReadFull(r, head); err != nil {
+		return 0, err
+	}
+	if string(head[0:4]) != "RIFF" || string(head[8:12]) != "WAVE" {
+		return 0, errors.New("invalid wav header")
+	}
+
+	var sampleRate uint32
+	var numChans uint16
+	var bitsPerSample uint16
+	var byteRate uint32
+	var dataSize uint32
+
+	for {
+		chunkHdr := make([]byte, 8)
+		if _, err := io.ReadFull(r, chunkHdr); err != nil {
+			if errors.Is(err, io.EOF) || errors.Is(err, io.ErrUnexpectedEOF) {
+				break
+			}
+			return 0, err
+		}
+		ckID := string(chunkHdr[0:4])
+		ckSize := binary.LittleEndian.Uint32(chunkHdr[4:8])
+
+		switch ckID {
+		case "fmt ":
+			fmtData := make([]byte, ckSize)
+			if _, err := io.ReadFull(r, fmtData); err != nil {
+				return 0, err
+			}
+			// audioFormat := binary.LittleEndian.Uint16(fmtData[0:2]) // 1 = PCM
+			numChans = binary.LittleEndian.Uint16(fmtData[2:4])
+			sampleRate = binary.LittleEndian.Uint32(fmtData[4:8])
+			byteRate = binary.LittleEndian.Uint32(fmtData[8:12])
+			// blockAlign := binary.LittleEndian.Uint16(fmtData[12:14])
+			if len(fmtData) >= 16 {
+				bitsPerSample = binary.LittleEndian.Uint16(fmtData[14:16])
+			}
+		case "data":
+			dataSize = ckSize
+			// Skip data content
+			if _, err := r.Seek(int64(ckSize), io.SeekCurrent); err != nil {
+				return 0, err
+			}
+		default:
+			// Skip other chunks
+			if _, err := r.Seek(int64(ckSize), io.SeekCurrent); err != nil {
+				return 0, err
+			}
+		}
+		// Chunks are word-aligned (pad byte if odd size)
+		if ckSize%2 == 1 {
+			if _, err := r.Seek(1, io.SeekCurrent); err != nil { // skip pad
+				return 0, err
+			}
+		}
+	}
+
+	if sampleRate == 0 || numChans == 0 {
+		return 0, errors.New("invalid wav fmt")
+	}
+
+	var durationSeconds float64
+	if byteRate != 0 {
+		durationSeconds = float64(dataSize) / float64(byteRate)
+	} else {
+		bytesPerSec := float64(sampleRate) * float64(numChans) * float64(bitsPerSample) / 8.0
+		if bytesPerSec == 0 {
+			return 0, errors.New("invalid wav parameters")
+		}
+		durationSeconds = float64(dataSize) / bytesPerSec
+	}
+	return time.Duration(durationSeconds * float64(time.Second)), nil
+}
+
+// ---------------------- MP3 ----------------------
+
+func mp3Duration(r io.ReadSeeker) (time.Duration, error) {
+	// Strategy:
+	// 1) Skip ID3v2 header if present.
+	// 2) Try read first frame and detect XING/Info or VBRI to get total frames and duration.
+	// 3) If VBR headers not present, fall back to CBR estimation: (audioDataBytes * 8) / bitrate.
+
+	// File size
+	fi, err := fileSizeFromSeeker(r)
+	if err != nil {
+		return 0, err
+	}
+
+	// Skip ID3v2
+	var id3v2Size int64
+	if _, err := r.Seek(0, io.SeekStart); err != nil {
+		return 0, err
+	}
+	id3v2Size, err = skipID3v2(r)
+	if err != nil {
+		return 0, err
+	}
+
+	// Remember audio start offset
+	startOffset, _ := r.Seek(0, io.SeekCurrent)
+
+	// Read first frame header (search sync)
+	off, fh, err := findNextMP3Frame(r)
+	if err != nil {
+		return 0, err
+	}
+	if _, err := r.Seek(off, io.SeekStart); err != nil {
+		return 0, err
+	}
+
+	// Check for XING/Info header in first frame (for VBR)
+	totalFrames, sampleRate, samplesPerFrame, bitrateKbps, vbrFound, err := parseFirstFrameForVBR(r, fh)
+	if err != nil {
+		return 0, err
+	}
+
+	if vbrFound && totalFrames > 0 && sampleRate > 0 && samplesPerFrame > 0 {
+		seconds := (float64(totalFrames) * float64(samplesPerFrame)) / float64(sampleRate)
+		return time.Duration(seconds * float64(time.Second)), nil
+	}
+
+	// Fall back to CBR estimate using bitrate and data size (excluding ID3v2 and ID3v1)
+	// Detect ID3v1 at end (128 bytes TAG)
+	var id3v1Size int64
+	if fi >= 128 {
+		if _, err := r.Seek(fi-128, io.SeekStart); err == nil {
+			buf := make([]byte, 3)
+			if _, err := io.ReadFull(r, buf); err == nil {
+				if string(buf) == "TAG" {
+					id3v1Size = 128
+				}
+			}
+		}
+	}
+
+	audioBytes := fi - id3v2Size - id3v1Size - startOffset
+	if audioBytes <= 0 || bitrateKbps == 0 {
+		return 0, errors.New("unable to estimate mp3 duration")
+	}
+	// bitrateKbps in kbps, bytes -> bits
+	seconds := float64(audioBytes*8) / float64(bitrateKbps*1000)
+	return time.Duration(seconds * float64(time.Second)), nil
+}
+
+type mp3FrameHeader struct {
+	Version     int // 1: MPEG1, 2: MPEG2, 25: MPEG2.5
+	Layer       int // 1,2,3
+	BitrateKbps int
+	SampleRate  int
+	Padding     int
+	ChannelMode int // 0:Stereo,1:Joint,2:Dual,3:Mono
+}
+
+func findNextMP3Frame(r io.ReadSeeker) (int64, mp3FrameHeader, error) {
+	var hdr mp3FrameHeader
+	// Start from current pos and scan up to 64KB
+	start, _ := r.Seek(0, io.SeekCurrent)
+	limit := int64(64 * 1024)
+	buf := make([]byte, limit)
+	n, err := r.Read(buf)
+	if err != nil && !errors.Is(err, io.EOF) {
+		return 0, hdr, err
+	}
+	for i := 0; i+4 <= n; i++ {
+		if buf[i] == 0xFF && (buf[i+1]&0xE0) == 0xE0 { // sync
+			if h, ok := parseMP3Header(buf[i : i+4]); ok {
+				offset := start + int64(i)
+				return offset, h, nil
+			}
+		}
+	}
+	return 0, hdr, errors.New("mp3 frame not found")
+}
+
+func parseMP3Header(b []byte) (mp3FrameHeader, bool) {
+	var h mp3FrameHeader
+	if len(b) < 4 {
+		return h, false
+	}
+	if b[0] != 0xFF || (b[1]&0xE0) != 0xE0 {
+		return h, false
+	}
+	versionBits := (b[1] >> 3) & 0x03
+	layerBits := (b[1] >> 1) & 0x03
+	bitrateBits := (b[2] >> 4) & 0x0F
+	sampleRateBits := (b[2] >> 2) & 0x03
+	paddingBit := (b[2] >> 1) & 0x01
+	channelMode := (b[3] >> 6) & 0x03
+
+	var version int
+	switch versionBits {
+	case 0x00:
+		version = 25 // MPEG 2.5
+	case 0x02:
+		version = 2 // MPEG 2
+	case 0x03:
+		version = 1 // MPEG 1
+	default:
+		return h, false
+	}
+
+	var layer int
+	switch layerBits {
+	case 0x01:
+		layer = 3
+	case 0x02:
+		layer = 2
+	case 0x03:
+		layer = 1
+	default:
+		return h, false
+	}
+
+	br := mp3BitrateKbps(version, layer, int(bitrateBits))
+	if br == 0 {
+		return h, false
+	}
+	sr := mp3SampleRate(version, int(sampleRateBits))
+	if sr == 0 {
+		return h, false
+	}
+
+	h = mp3FrameHeader{
+		Version:     version,
+		Layer:       layer,
+		BitrateKbps: br,
+		SampleRate:  sr,
+		Padding:     int(paddingBit),
+		ChannelMode: int(channelMode),
+	}
+	return h, true
+}
+
+func mp3BitrateKbps(version, layer, index int) int {
+	// index: 1..14 valid; 0,15 invalid
+	if index <= 0 || index == 15 {
+		return 0
+	}
+	// Tables per ISO/IEC 11172-3/13818-3 (common subset)
+	var tbl [15]int
+	if layer == 1 { // Layer I
+		if version == 1 { // MPEG1
+			tbl = [15]int{0, 32, 64, 96, 128, 160, 192, 224, 256, 288, 320, 352, 384, 416, 448}
+		} else { // MPEG2/2.5
+			tbl = [15]int{0, 32, 48, 56, 64, 80, 96, 112, 128, 144, 160, 176, 192, 224, 256}
+		}
+	} else if layer == 2 { // Layer II
+		if version == 1 {
+			tbl = [15]int{0, 32, 48, 56, 64, 80, 96, 112, 128, 160, 192, 224, 256, 320, 384}
+		} else {
+			tbl = [15]int{0, 8, 16, 24, 32, 40, 48, 56, 64, 80, 96, 112, 128, 144, 160}
+		}
+	} else { // Layer III
+		if version == 1 {
+			tbl = [15]int{0, 32, 40, 48, 56, 64, 80, 96, 112, 128, 160, 192, 224, 256, 320}
+		} else {
+			tbl = [15]int{0, 8, 16, 24, 32, 40, 48, 56, 64, 80, 96, 112, 128, 144, 160}
+		}
+	}
+	return tbl[index]
+}
+
+func mp3SampleRate(version, index int) int {
+	if index == 3 {
+		return 0
+	}
+	// base table for MPEG1
+	base := [3]int{44100, 48000, 32000}
+	sr := base[index]
+	if version == 2 { // MPEG2
+		sr /= 2
+	} else if version == 25 { // MPEG2.5
+		sr /= 4
+	}
+	return sr
+}
+
+func samplesPerMP3Frame(version, layer int) int {
+	switch layer {
+	case 1:
+		return 384
+	case 2:
+		return 1152
+	case 3:
+		if version == 1 {
+			return 1152
+		}
+		return 576 // MPEG2/2.5 Layer III
+	default:
+		return 0
+	}
+}
+
+func parseFirstFrameForVBR(r io.ReadSeeker, fh mp3FrameHeader) (totalFrames uint32, sampleRate int, samplesPerFrame int, bitrateKbps int, vbrFound bool, err error) {
+	// After the 4-byte header, possible side info and then XING/Info
+	if _, err = r.Seek(0, io.SeekCurrent); err != nil {
+		return
+	}
+	// Re-read header
+	hdr := make([]byte, 4)
+	if _, err = io.ReadFull(r, hdr); err != nil {
+		return
+	}
+
+	// side info size depends on MPEG version and channel mode (for Layer III)
+	sideInfoSize := 0
+	if fh.Layer == 3 { // Layer III
+		if fh.Version == 1 { // MPEG1
+			if fh.ChannelMode == 3 { // mono
+				sideInfoSize = 17
+			} else {
+				sideInfoSize = 32
+			}
+		} else { // MPEG2/2.5
+			if fh.ChannelMode == 3 {
+				sideInfoSize = 9
+			} else {
+				sideInfoSize = 17
+			}
+		}
+	}
+
+	// Read next up to 120 bytes to search for XING/Info or VBRI
+	buf := make([]byte, sideInfoSize+120)
+	if _, err = io.ReadFull(r, buf); err != nil {
+		// If short, still try within available
+		if !errors.Is(err, io.ErrUnexpectedEOF) && !errors.Is(err, io.EOF) {
+			return
+		}
+	}
+
+	// Search XING/Info signature
+	sigs := [][]byte{[]byte("Xing"), []byte("Info")}
+	for _, sig := range sigs {
+		idx := indexOf(buf, sig)
+		if idx >= 0 {
+			// flags after signature (4 bytes), then if frames flag set, 4 bytes frames
+			if len(buf) >= idx+4+4 {
+				flags := binary.BigEndian.Uint32(buf[idx+4 : idx+8])
+				var frames uint32
+				if (flags & 0x01) != 0 { // frames present
+					if len(buf) >= idx+8+4 {
+						frames = binary.BigEndian.Uint32(buf[idx+8 : idx+12])
+					}
+				}
+				if frames > 0 {
+					vbrFound = true
+					totalFrames = frames
+					sampleRate = fh.SampleRate
+					samplesPerFrame = samplesPerMP3Frame(fh.Version, fh.Layer)
+					bitrateKbps = fh.BitrateKbps
+					return
+				}
+			}
+		}
+	}
+
+	// Check VBRI (usually at 32 bytes after header for MPEG1 Layer III)
+	if len(buf) >= 4 {
+		idx := indexOf(buf, []byte("VBRI"))
+		if idx >= 0 {
+			if len(buf) >= idx+4+2+2+4+4 {
+				// VBRI layout: 'VBRI'(4) + version(2) + delay(2) + quality(2?) varies; but at offset 10 comes bytes: bytes (4), frames (4)
+				// Some docs: offset 10: bytes, offset 14: frames (big-endian)
+				bytesOffset := idx + 10
+				framesOffset := idx + 14
+				if len(buf) >= framesOffset+4 {
+					frames := binary.BigEndian.Uint32(buf[framesOffset : framesOffset+4])
+					if frames > 0 {
+						vbrFound = true
+						totalFrames = frames
+						sampleRate = fh.SampleRate
+						samplesPerFrame = samplesPerMP3Frame(fh.Version, fh.Layer)
+						bitrateKbps = fh.BitrateKbps
+						_ = bytesOffset // not used
+						return
+					}
+				}
+			}
+		}
+	}
+
+	// No VBR header. Provide header info for CBR fallback
+	sampleRate = fh.SampleRate
+	samplesPerFrame = samplesPerMP3Frame(fh.Version, fh.Layer)
+	bitrateKbps = fh.BitrateKbps
+	return
+}
+
+func indexOf(haystack []byte, needle []byte) int {
+	for i := 0; i+len(needle) <= len(haystack); i++ {
+		match := true
+		for j := 0; j < len(needle); j++ {
+			if haystack[i+j] != needle[j] {
+				match = false
+				break
+			}
+		}
+		if match {
+			return i
+		}
+	}
+	return -1
+}
+
+func skipID3v2(r io.ReadSeeker) (int64, error) {
+	if _, err := r.Seek(0, io.SeekStart); err != nil {
+		return 0, err
+	}
+	head := make([]byte, 10)
+	if _, err := io.ReadFull(r, head); err != nil {
+		return 0, nil // no header
+	}
+	if string(head[0:3]) != "ID3" {
+		if _, err := r.Seek(0, io.SeekStart); err != nil {
+			return 0, err
+		}
+		return 0, nil
+	}
+	// size: 4 synchsafe bytes
+	sz := int64((int(head[6]&0x7F) << 21) | (int(head[7]&0x7F) << 14) | (int(head[8]&0x7F) << 7) | int(head[9]&0x7F))
+	// total header size = 10 + sz (+ footer 10 if flag set)
+	footer := int64(0)
+	if (head[5] & 0x10) != 0 { // footer present
+		footer = 10
+	}
+	total := 10 + sz + footer
+	if _, err := r.Seek(total, io.SeekStart); err != nil {
+		return 0, err
+	}
+	return total, nil
+}
+
+func fileSizeFromSeeker(r io.ReadSeeker) (int64, error) {
+	cur, err := r.Seek(0, io.SeekCurrent)
+	if err != nil {
+		return 0, err
+	}
+	end, err := r.Seek(0, io.SeekEnd)
+	if err != nil {
+		return 0, err
+	}
+	if _, err := r.Seek(cur, io.SeekStart); err != nil {
+		return 0, err
+	}
+	return end, nil
+}
+
+// ---------------------- MP4 ----------------------
+
+type mp4BoxHeader struct {
+	Size uint64
+	Type [4]byte
+}
+
+func readBoxHeader(r io.ReadSeeker) (mp4BoxHeader, error) {
+	var h mp4BoxHeader
+	buf := make([]byte, 8)
+	if _, err := io.ReadFull(r, buf); err != nil {
+		return h, err
+	}
+	sz := binary.BigEndian.Uint32(buf[0:4])
+	copy(h.Type[:], buf[4:8])
+	if sz == 1 {
+		// 64-bit size follows
+		ext := make([]byte, 8)
+		if _, err := io.ReadFull(r, ext); err != nil {
+			return h, err
+		}
+		h.Size = binary.BigEndian.Uint64(ext)
+	} else {
+		h.Size = uint64(sz)
+	}
+	return h, nil
+}
+
+func skipBox(r io.ReadSeeker, boxSize uint64, alreadyRead int64) error {
+	toSkip := int64(boxSize) - alreadyRead
+	if toSkip < 0 {
+		return fmt.Errorf("invalid box size")
+	}
+	_, err := r.Seek(toSkip, io.SeekCurrent)
+	return err
+}
+
+func mp4Duration(r io.ReadSeeker) (time.Duration, error) {
+	if _, err := r.Seek(0, io.SeekStart); err != nil {
+		return 0, err
+	}
+	// Find moov box
+	var moovStart int64
+	var moovSize uint64
+	for {
+		pos, _ := r.Seek(0, io.SeekCurrent)
+		h, err := readBoxHeader(r)
+		if err != nil {
+			if errors.Is(err, io.EOF) || errors.Is(err, io.ErrUnexpectedEOF) {
+				break
+			}
+			return 0, err
+		}
+		if string(h.Type[:]) == "moov" {
+			moovStart = pos
+			moovSize = h.Size
+			break
+		}
+		if h.Size < 8 {
+			return 0, errors.New("invalid mp4 box size")
+		}
+		if err := skipBox(r, h.Size, 8); err != nil {
+			return 0, err
+		}
+	}
+	if moovStart == 0 && moovSize == 0 {
+		return 0, errors.New("moov not found")
+	}
+	// Parse inside moov for video trak mdhd, else mvhd
+	if _, err := r.Seek(moovStart+8, io.SeekStart); err != nil { // skip moov header
+		return 0, err
+	}
+	end := moovStart + int64(moovSize)
+	var movieTimescale uint32
+	var movieDuration uint64
+	var foundVideoMdhd bool
+	var mdhdTimescale uint32
+	var mdhdDuration uint64
+
+	for {
+		pos, _ := r.Seek(0, io.SeekCurrent)
+		if pos >= end {
+			break
+		}
+		h, err := readBoxHeader(r)
+		if err != nil {
+			return 0, err
+		}
+		switch string(h.Type[:]) {
+		case "mvhd":
+			// movie header
+			ver := make([]byte, 1)
+			if _, err := io.ReadFull(r, ver); err != nil {
+				return 0, err
+			}
+			if _, err := r.Seek(3, io.SeekCurrent); err != nil { // flags
+				return 0, err
+			}
+			if ver[0] == 1 {
+				// version 1: 64-bit duration
+				buf := make([]byte, 8+8+4+8) // ctime(8) mtime(8) timescale(4) duration(8)
+				if _, err := io.ReadFull(r, buf); err != nil {
+					return 0, err
+				}
+				movieTimescale = binary.BigEndian.Uint32(buf[16:20])
+				movieDuration = binary.BigEndian.Uint64(buf[20:28])
+			} else {
+				buf := make([]byte, 4+4+4+4) // ctime mtime timescale duration
+				if _, err := io.ReadFull(r, buf); err != nil {
+					return 0, err
+				}
+				movieTimescale = binary.BigEndian.Uint32(buf[8:12])
+				movieDuration = uint64(binary.BigEndian.Uint32(buf[12:16]))
+			}
+			// skip rest of mvhd
+			read := int64(1 + 3)
+			if ver[0] == 1 {
+				read += int64(8 + 8 + 4 + 8)
+			} else {
+				read += int64(4 + 4 + 4 + 4)
+			}
+			if err := skipBox(r, h.Size, 8+read); err != nil {
+				return 0, err
+			}
+		case "trak":
+			// parse trak for hdlr and mdhd
+			tEnd := int64(0)
+			if h.Size < 8 {
+				return 0, errors.New("invalid trak size")
+			}
+			tEnd = pos + int64(h.Size)
+			var isVideo bool
+			var tMdhdTimescale uint32
+			var tMdhdDuration uint64
+			for {
+				cpos, _ := r.Seek(0, io.SeekCurrent)
+				if cpos >= tEnd {
+					break
+				}
+				ch, err := readBoxHeader(r)
+				if err != nil {
+					return 0, err
+				}
+				switch string(ch.Type[:]) {
+				case "mdia":
+					mEnd := cpos + int64(ch.Size)
+					for {
+						mpos, _ := r.Seek(0, io.SeekCurrent)
+						if mpos >= mEnd {
+							break
+						}
+						mh, err := readBoxHeader(r)
+						if err != nil {
+							return 0, err
+						}
+						switch string(mh.Type[:]) {
+						case "hdlr":
+							// skip version+flags (4), pre_defined(4)
+							if _, err := r.Seek(8, io.SeekCurrent); err != nil {
+								return 0, err
+							}
+							handler := make([]byte, 4)
+							if _, err := io.ReadFull(r, handler); err != nil {
+								return 0, err
+							}
+							if string(handler) == "vide" {
+								isVideo = true
+							}
+							if err := skipBox(r, mh.Size, 8+8+4); err != nil { // header + skipped + read handler
+								return 0, err
+							}
+						case "mdhd":
+							ver := make([]byte, 1)
+							if _, err := io.ReadFull(r, ver); err != nil {
+								return 0, err
+							}
+							if _, err := r.Seek(3, io.SeekCurrent); err != nil { // flags
+								return 0, err
+							}
+							if ver[0] == 1 {
+								buf := make([]byte, 8+8+4+8)
+								if _, err := io.ReadFull(r, buf); err != nil {
+									return 0, err
+								}
+								tMdhdTimescale = binary.BigEndian.Uint32(buf[16:20])
+								tMdhdDuration = binary.BigEndian.Uint64(buf[20:28])
+							} else {
+								buf := make([]byte, 4+4+4+4)
+								if _, err := io.ReadFull(r, buf); err != nil {
+									return 0, err
+								}
+								tMdhdTimescale = binary.BigEndian.Uint32(buf[8:12])
+								tMdhdDuration = uint64(binary.BigEndian.Uint32(buf[12:16]))
+							}
+							if err := skipBox(r, mh.Size, 8+1+3+int64(lenVersionPayload(ver[0]))); err != nil {
+								return 0, err
+							}
+						default:
+							if err := skipBox(r, mh.Size, 8); err != nil {
+								return 0, err
+							}
+						}
+					}
+				default:
+					if err := skipBox(r, ch.Size, 8); err != nil {
+						return 0, err
+					}
+				}
+			}
+			if isVideo && tMdhdTimescale != 0 && tMdhdDuration != 0 {
+				foundVideoMdhd = true
+				mdhdTimescale = tMdhdTimescale
+				mdhdDuration = tMdhdDuration
+			}
+			// Skip remaining of trak if any
+			if _, err := r.Seek(tEnd, io.SeekStart); err != nil {
+				return 0, err
+			}
+		default:
+			if err := skipBox(r, h.Size, 8); err != nil {
+				return 0, err
+			}
+		}
+	}
+
+	if foundVideoMdhd && mdhdTimescale != 0 {
+		sec := float64(mdhdDuration) / float64(mdhdTimescale)
+		return time.Duration(sec * float64(time.Second)), nil
+	}
+	if movieTimescale != 0 {
+		sec := float64(movieDuration) / float64(movieTimescale)
+		return time.Duration(sec * float64(time.Second)), nil
+	}
+	return 0, errors.New("failed to read mp4 duration")
+}
+
+func lenVersionPayload(ver byte) int {
+	if ver == 1 {
+		return 8 + 8 + 4 + 8
+	}
+	return 4 + 4 + 4 + 4
+}

web/.env.development

@@ -6,7 +6,7 @@ VITE_ADMIN_USER=admin
 VITE_ADMIN_PASS=admin123
 VITE_KEY_PREFIX=GeekAI_DEV_
 VITE_TITLE="Geek-AI 创作系统"
-VITE_VERSION=v4.2.6
+VITE_VERSION=v4.2.7
 VITE_DOCS_URL=https://docs.geekai.me
 VITE_GITHUB_URL=https://github.com/yangjian102621/geekai
 VITE_GITEE_URL=https://gitee.com/blackfox/geekai

web/.env.production

@@ -1,7 +1,8 @@
 VITE_API_HOST=
 VITE_WS_HOST=
 VITE_KEY_PREFIX=GeekAI_
-VITE_VERSION=v4.2.6
+VITE_VERSION=v4.2.7
+VITE_TITLE="Geek-AI 创作系统"
 VITE_DOCS_URL=https://docs.geekai.me
 VITE_GITHUB_URL=https://github.com/yangjian102621/geekai
 VITE_GITEE_URL=https://gitee.com/blackfox/geekai

web/src/assets/css/jimeng.scss

@@ -49,16 +49,17 @@
 
         .category-btn {
           display: flex;
-          flex-direction: column;
+          flex-direction: row;
+          justify-content: center;
           align-items: center;
-          padding: 15px 10px;
+          padding: 10px 15px;
           border: 2px solid var(--border-color, #f0f0f0);
           border-radius: 12px;
           cursor: pointer;
           transition: all 0.3s ease;
           background: var(--card-bg-secondary, #fafafa);
           /* 暗色主题支持 */
-          [data-theme="dark"] & {
+          [data-theme='dark'] & {
             background: var(--card-bg-secondary-dark, #23242a);
             border-color: var(--border-color-dark, #33343a);
           }
@@ -66,7 +67,7 @@
           &:hover {
             border-color: var(--primary-color, #5865f2);
             background: var(--card-bg-hover, #f8f9ff);
-            [data-theme="dark"] & {
+            [data-theme='dark'] & {
               background: var(--card-bg-hover-dark, #2a2b31);
             }
             transform: translateY(-2px);
@@ -76,8 +77,11 @@
             border-color: var(--primary-color, #5865f2);
             background: var(--primary-gradient, linear-gradient(135deg, #5865f2 0%, #7289da 100%));
             color: var(--primary-text-on-primary, #fff);
-            [data-theme="dark"] & {
+            [data-theme='dark'] & {
-              background: var(--primary-gradient-dark, linear-gradient(135deg, #23242a 0%, #2a2b31 100%));
+              background: var(
+                --primary-gradient-dark,
+                linear-gradient(135deg, #23242a 0%, #2a2b31 100%)
+              );
               color: var(--primary-text-on-primary-dark, #fff);
             }
             transform: translateY(-2px);
@@ -96,108 +100,6 @@
         }
       }
     }
-
-    // 功能开关
-    .function-switch {
-      margin-bottom: 25px;
-
-      .switch-label {
-        display: flex;
-        align-items: center;
-        margin-bottom: 15px;
-        font-size: 16px;
-        font-weight: 600;
-        color: var(--text-theme-color);
-
-        .el-icon {
-          margin-right: 8px;
-          color: var(--primary-color, #5865f2);
-        }
-      }
-
-      .switch-container {
-        display: flex;
-        align-items: center;
-        justify-content: space-between;
-        padding: 5px 15px;
-        border: 1px solid var(--border-color, #e0e0e0);
-        border-radius: 10px;
-        background: var(--card-bg-secondary, #f9f9f9);
-        [data-theme="dark"] & {
-          background: var(--card-bg-secondary-dark, #23242a);
-          border-color: var(--border-color-dark, #33343a);
-        }
-
-        .switch-info {
-          flex: 1;
-
-          .switch-title {
-            font-size: 14px;
-            font-weight: 600;
-            color: var(--text-theme-color);
-            margin-bottom: 4px;
-          }
-
-          .switch-desc {
-            font-size: 12px;
-            color: var(--text-sub-color, #666);
-          }
-        }
-      }
-    }
-
-    // 参数容器
-    .params-container {
-      .function-panel {
-        .param-line {
-          margin-bottom: 15px;
-
-          &.pt {
-            margin-top: 20px;
-          }
-
-          .label {
-            display: flex;
-            align-items: center;
-            margin-bottom: 8px;
-            font-weight: 600;
-            color: var(--text-theme-color);
-          }
-        }
-
-        .item-group {
-          display: flex;
-          align-items: center;
-          margin-bottom: 15px;
-
-          .label {
-            margin-right: 15px;
-            font-weight: 600;
-            color: var(--text-theme-color);
-            min-width: 80px;
-          }
-        }
-
-        .text-info {
-          margin: 20px 0;
-          padding: 15px;
-          background: var(--info-bg, #f0f8ff);
-          border-radius: 8px;
-          border-left: 4px solid var(--primary-color, #5865f2);
-        }
-
-        .submit-btn {
-          margin-top: 30px;
-
-          .el-button {
-            width: 100%;
-            height: 50px;
-            font-size: 16px;
-            font-weight: 600;
-          }
-        }
-      }
-    }
   }
 
   // 右侧主要内容区域
@@ -239,6 +141,23 @@
         &:hover {
           box-shadow: 0 4px 24px rgba(88, 101, 242, 0.12);
         }
+
+        // 增强任务项悬停动画
+        transition: box-shadow 3s cubic-bezier(0.4, 0, 0.2, 1),
+          transform 0.5s cubic-bezier(0.4, 0, 0.2, 1), border-color 0.5s;
+        box-shadow: 0 2px 8px rgba(0, 0, 0, 0.04);
+        border: 1.5px solid transparent;
+        border-radius: 12px;
+        // background: #fff;
+        position: relative;
+        z-index: 1;
+
+        &:hover {
+          box-shadow: 0 8px 32px rgba(0, 0, 0, 0.18), 0 1.5px 8px rgba(0, 0, 0, 0.1);
+          border-color: #a259ff;
+          transform: scale(1.025) translateY(-2px);
+          z-index: 10;
+        }
         .task-left {
           width: 100%;
           flex: none;
@@ -253,18 +172,69 @@
             display: flex;
             align-items: center;
             justify-content: center;
-            .preview-image, .preview-video {
+            .preview-image,
+            .preview-video {
               width: 100%;
               height: 100%;
               object-fit: cover;
             }
+
+            // 视频预览包装器
+            .preview-video-wrapper {
+              position: relative;
+              width: 100%;
+              height: 100%;
+
+              .video-mask {
+                position: absolute;
+                top: 0;
+                left: 0;
+                width: 100%;
+                height: 100%;
+                background: rgba(0, 0, 0, 0.25);
+                display: flex;
+                justify-content: center;
+                align-items: center;
+                opacity: 0;
+                transition: opacity 0.2s;
+                z-index: 2;
+              }
+
+              &:hover .video-mask {
+                opacity: 1;
+              }
+
+              .play-btn {
+                width: 64px;
+                height: 64px;
+                background: rgba(255, 255, 255, 0.3);
+                border-radius: 50%;
+                display: flex;
+                justify-content: center;
+                align-items: center;
+                box-shadow: 0 2px 8px rgba(0, 0, 0, 0.15);
+                cursor: pointer;
+                z-index: 3;
+                transition: background 0.2s;
+
+                &:hover {
+                  background: rgba(255, 255, 255, 0.4);
+                }
+
+                img {
+                  width: 36px;
+                  height: 36px;
+                }
+              }
+            }
             .preview-placeholder {
               display: flex;
               flex-direction: column;
               align-items: center;
               color: var(--text-disabled-color, #999);
               font-size: 16px;
-              .el-icon, .iconfont {
+              .el-icon,
+              .iconfont {
                 font-size: 32px;
                 margin-bottom: 5px;
               }
@@ -347,3 +317,74 @@
     }
   }
 }
+
+// 错误信息样式
+.err-msg-clip {
+  display: -webkit-box;
+  -webkit-line-clamp: 2;
+  line-clamp: 2;
+  -webkit-box-orient: vertical;
+  overflow: hidden;
+  text-overflow: ellipsis;
+  word-break: break-all;
+  white-space: normal;
+}
+
+// 模板选择器样式
+.jimeng-template-select {
+  .el-select-dropdown__item {
+    height: 60px;
+    line-height: 60px;
+  }
+}
+
+// 提示词指南样式
+.prompt-guide {
+  margin: 12px 0 16px;
+  background-color: var(--el-fill-color-blank);
+
+  .el-collapse {
+    --el-collapse-border-color: none;
+  }
+
+  .guide-title {
+    display: flex;
+    align-items: center;
+    font-weight: 600;
+    color: #666;
+  }
+
+  .guide-content {
+    max-height: 220px;
+    overflow: auto;
+    line-height: 1.6;
+    font-size: 12px;
+    color: #555;
+    padding-right: 4px;
+  }
+
+  .guide-section {
+    margin-bottom: 10px;
+  }
+
+  .guide-subtitle {
+    font-weight: 600;
+    margin-bottom: 6px;
+    color: #333;
+  }
+
+  ul {
+    list-style: disc;
+    padding-left: 18px;
+    margin: 4px 0;
+  }
+
+  .quote {
+    margin: 8px 0;
+    padding: 8px 10px;
+    border-left: 3px solid #a3a3a3;
+    background: #f8f8f8;
+    border-radius: 4px;
+    color: #444;
+  }
+}

web/src/assets/css/mobile/jimeng.scss

@@ -887,3 +887,284 @@
     transform: rotate(360deg);
   }
 }
+
+/* Dark 主题样式 - 按照 theme-dark.scss 的模式 */
+:root[data-theme='dark'] .jimeng-create {
+  background-color: rgb(13, 20, 53);
+
+  /* 页面头部样式 */
+  .sticky {
+    background-color: rgb(31, 41, 55) !important;
+    box-shadow: 0 1px 3px rgba(0, 0, 0, 0.3);
+
+    h1 {
+      color: rgb(255, 255, 255) !important;
+    }
+
+    .iconfont {
+      color: rgb(156, 163, 175) !important;
+    }
+
+    button:hover {
+      background-color: rgb(75, 85, 99) !important;
+    }
+  }
+
+  /* 功能分类选择 */
+  .jimeng-create__content {
+    .bg-white {
+      background-color: rgb(55, 65, 81) !important;
+      box-shadow: 0 1px 3px rgba(0, 0, 0, 0.3);
+    }
+
+    .text-gray-700 {
+      color: rgb(209, 213, 219) !important;
+    }
+
+    .text-gray-900 {
+      color: rgb(255, 255, 255) !important;
+    }
+
+    .text-gray-600 {
+      color: rgb(156, 163, 175) !important;
+    }
+
+    .text-gray-500 {
+      color: rgb(156, 163, 175) !important;
+    }
+
+    .bg-gray-100:hover {
+      background-color: rgb(75, 85, 99) !important;
+    }
+
+    /* Element Plus 组件样式覆盖 */
+    :deep(.el-input__wrapper) {
+      background-color: rgb(31, 41, 55) !important;
+      border-color: rgb(75, 85, 99) !important;
+      box-shadow: none !important;
+    }
+
+    :deep(.el-input__inner) {
+      color: rgb(209, 213, 219) !important;
+      background-color: transparent !important;
+    }
+
+    :deep(.el-input__inner::placeholder) {
+      color: rgb(156, 163, 175) !important;
+    }
+
+    :deep(.el-textarea__inner) {
+      color: rgb(209, 213, 219) !important;
+      background-color: transparent !important;
+    }
+
+    :deep(.el-textarea__inner::placeholder) {
+      color: rgb(156, 163, 175) !important;
+    }
+
+    :deep(.el-switch__core) {
+      background-color: rgb(75, 85, 99) !important;
+      border-color: rgb(75, 85, 99) !important;
+    }
+
+    :deep(.el-switch.is-checked .el-switch__core) {
+      background-color: rgb(139, 92, 246) !important;
+      border-color: rgb(139, 92, 246) !important;
+    }
+
+    :deep(.el-slider__runway) {
+      background-color: rgb(75, 85, 99) !important;
+    }
+
+    :deep(.el-slider__bar) {
+      background-color: rgb(139, 92, 246) !important;
+    }
+
+    :deep(.el-slider__button) {
+      border-color: rgb(139, 92, 246) !important;
+    }
+
+    :deep(.el-tooltip__trigger) {
+      color: rgb(156, 163, 175) !important;
+    }
+  }
+
+  /* 提交按钮 */
+  .bg-gradient-to-r {
+    background: linear-gradient(88deg, #af61f0 1.44%, #5b62ce) !important;
+
+    &:hover {
+      background: linear-gradient(88deg, #9f51e0 1.44%, #4b52be) !important;
+    }
+
+    &:disabled {
+      background: linear-gradient(88deg, #6b7280 1.44%, #4b5563) !important;
+    }
+  }
+
+  /* 作品列表 */
+  .jimeng-create__works {
+    &-title {
+      color: rgb(255, 255, 255) !important;
+    }
+
+    &-item {
+      background-color: rgb(55, 65, 81) !important;
+      box-shadow: 0 1px 3px rgba(0, 0, 0, 0.3);
+
+      &-content {
+        .jimeng-create__works-item-info {
+          &-title {
+            color: rgb(255, 255, 255) !important;
+          }
+
+          &-prompt {
+            color: rgb(209, 213, 219) !important;
+          }
+
+          &-tags {
+            &-item {
+              background-color: rgb(75, 85, 99) !important;
+              color: rgb(209, 213, 219) !important;
+
+              &--warning {
+                background-color: rgb(239, 68, 68) !important;
+                color: rgb(255, 255, 255) !important;
+              }
+
+              &--primary {
+                background-color: rgb(59, 130, 246) !important;
+                color: rgb(255, 255, 255) !important;
+              }
+
+              &--power {
+                background-color: rgb(139, 92, 246) !important;
+                color: rgb(255, 255, 255) !important;
+              }
+            }
+          }
+        }
+      }
+
+      &-quick-actions {
+        button {
+          color: rgb(156, 163, 175) !important;
+
+          &:hover {
+            color: rgb(209, 213, 219) !important;
+          }
+        }
+      }
+
+      &-error {
+        &-content {
+          background-color: rgb(31, 41, 55) !important;
+          border-color: rgb(239, 68, 68) !important;
+
+          .jimeng-create__works-item-error-text {
+            color: rgb(239, 68, 68) !important;
+          }
+
+          .jimeng-create__works-item-error-copy-btn {
+            color: rgb(156, 163, 175) !important;
+
+            &:hover {
+              color: rgb(209, 213, 219) !important;
+            }
+          }
+        }
+      }
+    }
+
+    &-loading {
+      color: rgb(156, 163, 175) !important;
+    }
+
+    &-finished {
+      color: rgb(156, 163, 175) !important;
+    }
+  }
+
+  /* 媒体预览弹窗 */
+  .jimeng-create__media-dialog {
+    background-color: rgba(0, 0, 0, 0.8) !important;
+
+    &-content {
+      background-color: rgb(55, 65, 81) !important;
+      box-shadow: 0 0 15px rgba(107, 80, 225, 0.8) !important;
+    }
+
+    &-header {
+      background-color: rgb(31, 41, 55) !important;
+      border-bottom-color: rgb(75, 85, 99) !important;
+
+      h3 {
+        color: rgb(255, 255, 255) !important;
+      }
+
+      button {
+        color: rgb(156, 163, 175) !important;
+
+        &:hover {
+          color: rgb(209, 213, 219) !important;
+        }
+      }
+    }
+  }
+
+  /* 图片上传组件 */
+  :deep(.image-upload) {
+    .upload-area {
+      background-color: rgb(31, 41, 55) !important;
+      border-color: rgb(75, 85, 99) !important;
+
+      &:hover {
+        border-color: rgb(139, 92, 246) !important;
+        background-color: rgb(55, 65, 81) !important;
+      }
+    }
+
+    .upload-text {
+      color: rgb(209, 213, 219) !important;
+    }
+
+    .upload-icon {
+      color: rgb(139, 92, 246) !important;
+    }
+  }
+
+  /* 自定义选择组件 */
+  :deep(.custom-select) {
+    .select-trigger {
+      background-color: rgb(31, 41, 55) !important;
+      border-color: rgb(75, 85, 99) !important;
+      color: rgb(209, 213, 219) !important;
+    }
+
+    .select-dropdown {
+      background-color: rgb(55, 65, 81) !important;
+      border-color: rgb(75, 85, 99) !important;
+      box-shadow: 0 0 15px rgba(107, 80, 225, 0.8) !important;
+    }
+
+    .select-option {
+      color: rgb(209, 213, 219) !important;
+
+      &:hover {
+        background-color: rgb(75, 85, 99) !important;
+      }
+
+      &.selected {
+        background-color: rgb(139, 92, 246) !important;
+        color: rgb(255, 255, 255) !important;
+      }
+    }
+  }
+
+  /* 空状态组件 */
+  :deep(.van-empty) {
+    .van-empty__description {
+      color: rgb(156, 163, 175) !important;
+    }
+  }
+}

web/src/assets/css/theme-dark.scss

@@ -26,6 +26,7 @@
   --btnColor: linear-gradient(88deg, #af61f0 1.44%, #5b62ce);
   --border-active: rgba(255, 255, 255, 0.1);
   --card-bg: #252d58;
+  --card-bg-secondary: #313a6b;
   --chat-bg: #1f243f;
   --chat-wel-bg: #2d2f38;
   --card-bg-table: rgba(17, 28, 68, 1);

web/src/assets/css/theme-light.scss

@@ -24,7 +24,8 @@
   --btnColor: linear-gradient(88deg, #af61f0 1.44%, #5b62ce);
   --border-active: rgba(134, 140, 255, 1);
   --code-btnColor: linear-gradient(88deg, #af61f0 1.44%, #5b62ce);
-  --card-bg: #fff;
+  --card-bg: #f5f5f5;
+  --card-bg-secondary: #e5e5e5;
   --chat-bg: #fff;
   --theme-bg: linear-gradient(88deg, #fff3f3 1.44%, #e7e8ff);
   --theme-bg-all: #f5f7fd;

web/src/assets/css/video.scss

@@ -14,7 +14,7 @@
   }
 
   :deep(.el-textarea__inner) {
-    background: transparent;
+    // background: transparent;
     color: var(--text-theme-color);
   }
 
@@ -104,7 +104,8 @@
         color: var(--text-theme-color);
       }
 
-      .el-input, .el-slider {
+      .el-input,
+      .el-slider {
         width: 100%;
       }
 
@@ -422,7 +423,8 @@
             flex-flow: column;
             padding: 0 20px;
 
-            .prompt, .failed {
+            .prompt,
+            .failed {
               padding: 0;
               font-size: 16px;
               max-height: 80px;
@@ -547,7 +549,8 @@
         .left .container {
           width: 120px;
 
-          .video, .el-image {
+          .video,
+          .el-image {
             width: 120px;
           }
         }

web/src/assets/iconfont/iconfont.css

@@ -1,8 +1,8 @@
 @font-face {
   font-family: "iconfont"; /* Project id 4125778 */
-  src: url('iconfont.woff2?t=1756954977612') format('woff2'),
+  src: url('iconfont.woff2?t=1757571432313') format('woff2'),
-       url('iconfont.woff?t=1756954977612') format('woff'),
+       url('iconfont.woff?t=1757571432313') format('woff'),
-       url('iconfont.ttf?t=1756954977612') format('truetype');
+       url('iconfont.ttf?t=1757571432313') format('truetype');
 }
 
 .iconfont {
@@ -13,6 +13,42 @@
   -moz-osx-font-smoothing: grayscale;
 }
 
+.icon-resize:before {
+  content: "\e718";
+}
+
+.icon-template:before {
+  content: "\e8a6";
+}
+
+.icon-error-line:before {
+  content: "\e868";
+}
+
+.icon-success-line:before {
+  content: "\e88c";
+}
+
+.icon-yunjing:before {
+  content: "\e69b";
+}
+
+.icon-action:before {
+  content: "\e658";
+}
+
+.icon-dancing:before {
+  content: "\e659";
+}
+
+.icon-running:before {
+  content: "\e65e";
+}
+
+.icon-shuziren:before {
+  content: "\e6df";
+}
+
 .icon-cube:before {
   content: "\e72c";
 }

web/src/assets/iconfont/iconfont.json

@@ -5,6 +5,69 @@
   "css_prefix_text": "icon-",
   "description": "",
   "glyphs": [
+    {
+      "icon_id": "10564356",
+      "name": "resize",
+      "font_class": "resize",
+      "unicode": "e718",
+      "unicode_decimal": 59160
+    },
+    {
+      "icon_id": "1727381",
+      "name": "34模板、框架",
+      "font_class": "template",
+      "unicode": "e8a6",
+      "unicode_decimal": 59558
+    },
+    {
+      "icon_id": "9626841",
+      "name": "错误",
+      "font_class": "error-line",
+      "unicode": "e868",
+      "unicode_decimal": 59496
+    },
+    {
+      "icon_id": "9626990",
+      "name": "正确",
+      "font_class": "success-line",
+      "unicode": "e88c",
+      "unicode_decimal": 59532
+    },
+    {
+      "icon_id": "40613765",
+      "name": "运镜控制",
+      "font_class": "yunjing",
+      "unicode": "e69b",
+      "unicode_decimal": 59035
+    },
+    {
+      "icon_id": "5215282",
+      "name": "动作",
+      "font_class": "action",
+      "unicode": "e658",
+      "unicode_decimal": 58968
+    },
+    {
+      "icon_id": "7250581",
+      "name": "跳舞",
+      "font_class": "dancing",
+      "unicode": "e659",
+      "unicode_decimal": 58969
+    },
+    {
+      "icon_id": "8153037",
+      "name": "动作",
+      "font_class": "running",
+      "unicode": "e65e",
+      "unicode_decimal": 58974
+    },
+    {
+      "icon_id": "42680536",
+      "name": "数字人",
+      "font_class": "shuziren",
+      "unicode": "e6df",
+      "unicode_decimal": 59103
+    },
     {
       "icon_id": "544492",
       "name": "cube",

web/src/components/AccountBg.vue

@@ -1,84 +0,0 @@
-<template>
-  <div class="right flex-center">
-    <div class="logo">
-      <el-image
-        :src="logo"
-        alt=""
-        style="max-width: 300px; max-height: 300px"
-        class="rounded-full"
-      />
-    </div>
-    <div>welcome</div>
-    <footer-bar />
-  </div>
-</template>
-
-<script setup>
-import FooterBar from '@/components/FooterBar.vue'
-import { getSystemInfo } from '@/store/cache'
-import { ref } from 'vue'
-
-const logo = ref('')
-const title = ref('')
-
-getSystemInfo()
-  .then((res) => {
-    logo.value = res.data.logo
-    title.value = res.data.title
-  })
-  .catch((err) => {
-    console.log(err)
-    logo.value = '/images/logo.png'
-    title.value = 'Geek-AI'
-  })
-</script>
-
-<style lang="scss" scoped>
-.right {
-  font-size: 40px;
-  font-weight: bold;
-  color: #fff;
-  flex-direction: column;
-  background-image: url('~@/assets/img/login-bg.png');
-  background-size: cover;
-  background-position: center;
-  width: 50%;
-  min-height: 100vh;
-  max-height: 100vh;
-  background-repeat: no-repeat;
-  position: relative;
-  overflow: hidden;
-  z-index: 1;
-
-  :deep(.foot-container) {
-    position: absolute;
-    bottom: 20px;
-    width: 100%;
-    background: none;
-    color: var(--sm-txt);
-    font-size: 12px;
-    text-align: center;
-
-    .footer {
-      a,
-      span {
-        color: var(--text-fff);
-      }
-    }
-  }
-}
-
-.logo {
-  margin-bottom: 26px;
-  width: 200px;
-  height: 200px;
-  background: #fff;
-  border-radius: 50%;
-
-  img {
-    width: 100%;
-    object-fit: cover;
-    height: 100%;
-  }
-}
-</style>

web/src/components/AccountTop.vue

@@ -1,102 +0,0 @@
-<template>
-  <div>
-    <ThemeChange />
-    <div @click="goBack" class="flex back animate__animated animate__pulse animate__infinite">
-      <el-icon><ArrowLeftBold /></el-icon>{{ title === '注册' ? '首页' : '返回' }}
-    </div>
-    <div class="title">{{ title }}</div>
-    <div class="smTitle" v-if="title !== '重置密码'">
-      {{ title === '登录' ? '没有账号？' : '已有账号？'
-      }}<span @click="goPageFun" class="text-color-primary sign"
-        >赶紧{{ title === '登录' ? '注册' : '登录' }}</span
-      >
-    </div>
-    <slot></slot>
-    <div class="flex orline" v-if="title !== '重置密码'">
-      <div class="lineor"></div>
-      <span>或</span>
-      <div class="lineor"></div>
-    </div>
-  </div>
-</template>
-
-<script setup>
-import ThemeChange from '@/components/ThemeChange.vue'
-import { ArrowLeftBold } from '@element-plus/icons-vue'
-import { defineProps } from 'vue'
-import { useRouter } from 'vue-router'
-
-const props = defineProps({
-  title: {
-    type: String,
-    default: '登录',
-  },
-  smTitle: { type: String, default: '没有账号？' },
-  goPage: {
-    type: String,
-    default: '/register',
-  },
-})
-
-const router = useRouter()
-const goBack = () => {
-  if (props.title === '注册') {
-    router.push('/')
-  } else {
-    router.go(-1)
-  }
-}
-const goPageFun = () => {
-  if (props.title === '登录') {
-    router.push('/register')
-  } else {
-    router.push('/login')
-  }
-}
-</script>
-
-<style lang="scss" scoped>
-.back {
-  color: var(--sm-txt);
-  font-size: 14px;
-  margin-bottom: 140px;
-  margin-top: 18px;
-  cursor: pointer;
-
-  .el-icon {
-    margin-right: 6px;
-  }
-}
-
-.title {
-  font-size: 36px;
-  margin-bottom: 16px;
-  color: var(--text-color);
-}
-
-.smTitle {
-  color: var(--text-color);
-  font-size: 14px;
-  margin-bottom: 36px;
-}
-
-.sign {
-  text-decoration: underline;
-  cursor: pointer;
-}
-
-.orline {
-  color: var(--text-secondary);
-
-  span {
-    font-size: 14px;
-    margin: 0 10px;
-  }
-
-  .lineor {
-    width: 182px;
-    height: 1px;
-    background: var(--text-secondary);
-  }
-}
-</style>

web/src/components/ChatPrompt.vue

@@ -1,5 +1,5 @@
 <template>
-  <div class="chat-line chat-line-prompt-list" v-if="listStyle === 'list'">
+  <div class="chat-line chat-line-prompt-chat">
     <div class="chat-line-inner">
       <div class="chat-icon">
         <img :src="data.icon" alt="User" />
@@ -29,73 +29,68 @@
             </div>
           </div>
         </div>
-        <div class="content position-relative">
+
-          <div v-html="content"></div>
+        <!-- 编辑模式 -->
-        </div>
+        <div v-if="isEditing" class="edit-mode">
-        <div class="bar" v-if="data.created_at > 0">
+          <div class="flex flex-row space-x-2 w-full">
-          <span class="bar-item"
+            <div>
-            ><el-icon><Clock /></el-icon> {{ dateFormat(data.created_at) }}</span
+              <el-tooltip class="box-item" effect="dark" content="取消" placement="top">
-          >
-          <span class="bar-item">
-            <el-tooltip class="box-item" effect="dark" content="复制" placement="bottom">
                 <i
-                class="iconfont icon-copy cursor-pointer"
+                  class="iconfont icon-error-line !text-lg mr-1 cursor-pointer"
-                @click="copyContent(data.content.text)"
+                  @click="cancelEdit"
                 ></i>
               </el-tooltip>
-          </span>
             </div>
+            <div class="w-full">
+              <textarea
+                v-model="editText"
+                :rows="3"
+                placeholder="请输入修改后的内容..."
+                class="w-full p-3 border-2 border-purple-500 rounded-md text-sm"
+                resize="vertical"
+                style="background-color: var(--chat-content-bg); color: var(--theme-text-primary)"
+              ></textarea>
+            </div>
+            <div>
+              <el-tooltip class="box-item" effect="dark" content="提交" placement="top">
+                <i
+                  class="iconfont icon-back-circle cursor-pointer !text-3xl text-purple-500 mr-1 hover:text-purple-700 mb-2"
+                  style="transform: rotate(90deg); display: inline-block"
+                  @click="submitEdit"
+                ></i>
+              </el-tooltip>
             </div>
           </div>
         </div>
 
-  <div class="chat-line chat-line-prompt-chat" v-else>
+        <!-- 显示模式 -->
-    <div class="chat-line-inner">
+        <div v-else>
-      <div class="chat-icon">
-        <img :src="data.icon" alt="User" />
-      </div>
-
-      <div class="chat-item">
-        <div v-if="files && files.length > 0" class="file-list-box">
-          <div v-for="file in files" :key="file.url">
-            <div class="image" v-if="isImage(file.ext)">
-              <el-image :src="file.url" fit="cover" />
-            </div>
-            <div class="item" v-else>
-              <div class="icon">
-                <el-image :src="GetFileIcon(file.ext)" fit="cover" />
-              </div>
-              <div class="body">
-                <div class="title">
-                  <el-link :href="file.url" target="_blank" style="--el-font-weight-primary: bold"
-                    >{{ file.name }}
-                  </el-link>
-                </div>
-                <div class="info">
-                  <span>{{ GetFileType(file.ext) }}</span>
-                  <span>{{ FormatFileSize(file.size) }}</span>
-                </div>
-              </div>
-            </div>
-          </div>
-        </div>
           <div class="content-wrapper">
             <div class="content position-relative">
               <div v-html="content"></div>
             </div>
           </div>
-        <div class="bar" v-if="data.created_at > 0">
+          <div
-          <span class="bar-item"
+            class="flex text-gray-500 text-sm py-2 justify-end items-center space-x-2"
-            ><el-icon><Clock /></el-icon> {{ dateFormat(data.created_at) }}</span
+            v-if="data.created_at > 0"
           >
-          <span class="bar-item">
+            <span class="flex items-center"
-            <el-tooltip class="box-item" effect="dark" content="复制" placement="bottom">
+              ><i class="iconfont icon-clock mr-1"></i> {{ dateFormat(data.created_at) }}</span
+            >
+            <span class="flex items-center">
+              <el-tooltip class="box-item" effect="dark" content="复制" placement="top">
                 <i
-                class="iconfont icon-copy cursor-pointer"
+                  class="iconfont icon-copy cursor-pointer !text-sm"
                   @click="copyContent(data.content.text)"
                 ></i>
               </el-tooltip>
             </span>
+            <span class="flex items-center">
+              <el-tooltip class="box-item" effect="dark" content="修改提问" placement="top">
+                <i class="iconfont icon-edit cursor-pointer !text-sm" @click="startEdit"></i>
+              </el-tooltip>
+            </span>
+          </div>
         </div>
       </div>
     </div>
@@ -106,7 +101,7 @@
 import { FormatFileSize, GetFileIcon, GetFileType } from '@/store/system'
 import { showMessageSuccess } from '@/utils/dialog'
 import { dateFormat, isImage, processPrompt } from '@/utils/libs'
-import { Clock } from '@element-plus/icons-vue'
+import { ElMessage } from 'element-plus'
 import hl from 'highlight.js'
 import MarkdownIt from 'markdown-it'
 import emoji from 'markdown-it-emoji'
@@ -157,15 +152,19 @@ const props = defineProps({
       icon: '',
     },
   },
-  listStyle: {
+  messageIndex: {
-    type: String,
+    type: Number,
-    default: 'list',
+    default: -1,
   },
 })
 const finalTokens = ref(props.data.tokens)
 const content = ref(processPrompt(props.data.content.text))
 const files = ref(props.data.content.files)
 
+// 编辑相关状态
+const isEditing = ref(false)
+const editText = ref('')
+
 // 定义emit事件
 const emit = defineEmits(['edit'])
 
@@ -186,148 +185,40 @@ const copyContent = (text) => {
   navigator.clipboard.writeText(text)
   showMessageSuccess('复制成功')
 }
+
+// 开始编辑
+const startEdit = () => {
+  isEditing.value = true
+  editText.value = props.data.content.text
+}
+
+// 取消编辑
+const cancelEdit = () => {
+  isEditing.value = false
+  editText.value = ''
+}
+
+// 提交编辑
+const submitEdit = () => {
+  if (!editText.value.trim()) {
+    ElMessage.warning('内容不能为空')
+    return
+  }
+  // 发送重新提交事件，传递修改后的内容
+  emit('edit', {
+    messageIndex: props.messageIndex,
+    newContent: editText.value.trim(),
+  })
+
+  // 退出编辑模式
+  isEditing.value = false
+  editText.value = ''
+}
 </script>
 
 <style lang="scss">
 @use '@/assets/css/markdown/vue.css' as *;
-.chat-page,
+.chat-page {
-.chat-export {
-  .chat-line-prompt-list {
-    background-color: var(--chat-content-bg-list);
-    color: var(--theme-text-color-primary);
-    justify-content: center;
-    width: 100%;
-    padding-bottom: 1.5rem;
-    padding-top: 1.5rem;
-    // border-bottom: 0.5px solid var(--el-border-color);
-
-    .chat-line-inner {
-      display: flex;
-      width: 100%;
-      max-width: 900px;
-      padding-left: 10px;
-
-      .chat-icon {
-        margin-right: 20px;
-
-        img {
-          width: 36px;
-          height: 36px;
-          border-radius: 50%;
-          padding: 1px;
-        }
-      }
-
-      .chat-item {
-        width: 100%;
-        padding: 0 5px 0 0;
-        overflow: hidden;
-
-        .file-list-box {
-          display: flex;
-          flex-flow: column;
-
-          .image {
-            display: flex;
-            flex-flow: row;
-            margin-right: 10px;
-            position: relative;
-            justify-content: start;
-
-            .el-image {
-              border: 1px solid #e3e3e3;
-              border-radius: 10px;
-              margin-bottom: 10px;
-              max-width: 150px;
-              max-height: 150px;
-            }
-          }
-
-          .item {
-            display: flex;
-            flex-flow: row;
-            border-radius: 10px;
-            background-color: var(--chat-content-bg);
-            border: 1px solid #e3e3e3;
-            color: var(--theme-text-color-primary);
-            padding: 6px;
-            margin-bottom: 10px;
-
-            .icon {
-              .el-image {
-                width: 40px;
-                height: 40px;
-              }
-            }
-
-            .body {
-              margin-left: 8px;
-              font-size: 14px;
-
-              .title {
-                font-weight: bold;
-                line-height: 24px;
-                color: #0d0d0d;
-              }
-
-              .info {
-                color: #b4b4b4;
-
-                span {
-                  margin-right: 10px;
-                }
-              }
-            }
-          }
-        }
-
-        .content {
-          word-break: break-word;
-          padding: 0;
-          color: var(--theme-text-color-primary);
-          font-size: var(--content-font-size);
-          border-radius: 5px;
-          overflow: auto;
-
-          img {
-            max-width: 600px;
-            border-radius: 10px;
-            margin: 10px 0;
-          }
-
-          p {
-            line-height: 1.5;
-          }
-
-          p:last-child {
-            margin-bottom: 0;
-          }
-
-          p:first-child {
-            margin-top: 0;
-          }
-        }
-
-        .bar {
-          padding: 10px 10px 10px 0;
-
-          .bar-item {
-            // background-color #f7f7f8;
-            color: #888;
-            padding: 3px 5px;
-            margin-right: 10px;
-            border-radius: 5px;
-
-            .el-icon {
-              position: relative;
-              top: 2px;
-            }
-          }
-        }
-      }
-    }
-  }
-
   .chat-line-prompt-chat {
     background: var(--chat-bg);
     justify-content: center;
@@ -356,6 +247,7 @@ const copyContent = (text) => {
         padding: 0;
         overflow: hidden;
         max-width: calc(100% - 110px);
+        width: 100%;
 
         .file-list-box {
           display: flex;
@@ -448,20 +340,9 @@ const copyContent = (text) => {
           }
         }
 
-        .bar {
+        .edit-mode {
-          padding: 10px 10px 10px 0;
+          width: 100%;
-
+          margin-top: 15px;
-          .bar-item {
-            color: #888;
-            padding: 3px 5px;
-            margin-right: 10px;
-            border-radius: 5px;
-
-            .el-icon {
-              position: relative;
-              top: 2px;
-            }
-          }
         }
       }
     }

web/src/components/ChatPromptLine.vue

@@ -0,0 +1,265 @@
+<template>
+  <div class="chat-line chat-line-prompt-list">
+    <div class="chat-line-inner">
+      <div class="chat-icon">
+        <img :src="data.icon" alt="User" />
+      </div>
+
+      <div class="chat-item">
+        <div v-if="files && files.length > 0" class="file-list-box">
+          <div v-for="file in files" :key="file.url">
+            <div class="image" v-if="isImage(file.ext)">
+              <el-image :src="file.url" fit="cover" />
+            </div>
+            <div class="item" v-else>
+              <div class="icon">
+                <el-image :src="GetFileIcon(file.ext)" fit="cover" />
+              </div>
+              <div class="body">
+                <div class="title">
+                  <el-link :href="file.url" target="_blank" style="--el-font-weight-primary: bold"
+                    >{{ file.name }}
+                  </el-link>
+                </div>
+                <div class="info">
+                  <span>{{ GetFileType(file.ext) }}</span>
+                  <span>{{ FormatFileSize(file.size) }}</span>
+                </div>
+              </div>
+            </div>
+          </div>
+        </div>
+        <div class="content position-relative">
+          <div v-html="content"></div>
+        </div>
+        <div
+          class="flex text-gray-500 text-sm py-2 items-center space-x-2"
+          v-if="data.created_at > 0"
+        >
+          <span class="flex items-center"
+            ><i class="iconfont icon-clock mr-1"></i> {{ dateFormat(data.created_at) }}</span
+          >
+          <span class="flex items-center">
+            <el-tooltip class="box-item" effect="dark" content="复制" placement="top">
+              <i
+                class="iconfont icon-copy cursor-pointer !text-sm"
+                @click="copyContent(data.content.text)"
+              ></i>
+            </el-tooltip>
+          </span>
+        </div>
+      </div>
+    </div>
+  </div>
+</template>
+
+<script setup>
+import { FormatFileSize, GetFileIcon, GetFileType } from '@/store/system'
+import { showMessageSuccess } from '@/utils/dialog'
+import { dateFormat, isImage, processPrompt } from '@/utils/libs'
+import hl from 'highlight.js'
+import MarkdownIt from 'markdown-it'
+import emoji from 'markdown-it-emoji'
+import mathjaxPlugin from 'markdown-it-mathjax3'
+import { onMounted, ref } from 'vue'
+
+const md = new MarkdownIt({
+  breaks: true,
+  html: true,
+  linkify: true,
+  typographer: true,
+  highlight: function (str, lang) {
+    const codeIndex = parseInt(Date.now()) + Math.floor(Math.random() * 10000000)
+    // 显示复制代码按钮
+    const copyBtn = `<span class="copy-code-btn" data-clipboard-action="copy" data-clipboard-target="#copy-target-${codeIndex}">复制</span>
+<textarea style="position: absolute;top: -9999px;left: -9999px;z-index: -9999;" id="copy-target-${codeIndex}">${str.replace(
+      /<\/textarea>/g,
+      '&lt;/textarea>'
+    )}</textarea>`
+    if (lang && hl.getLanguage(lang)) {
+      const langHtml = `<span class="lang-name">${lang}</span>`
+      // 处理代码高亮
+      const preCode = hl.highlight(str, { language: lang, ignoreIllegals: true }).value
+      // 将代码包裹在 pre 中
+      return `<pre class="code-container"><code class="language-${lang} hljs">${preCode}</code>${copyBtn} ${langHtml}</pre>`
+    }
+
+    // 处理代码高亮
+    const preCode = md.utils.escapeHtml(str)
+    // 将代码包裹在 pre 中
+    return `<pre class="code-container"><code class="language-${lang} hljs">${preCode}</code>${copyBtn}</pre>`
+  },
+})
+md.use(mathjaxPlugin)
+md.use(emoji)
+
+const props = defineProps({
+  data: {
+    type: Object,
+    default: {
+      content: {
+        text: '',
+        files: [],
+      },
+      created_at: '',
+      tokens: 0,
+      model: '',
+      icon: '',
+    },
+  },
+  listStyle: {
+    type: String,
+    default: 'list',
+  },
+})
+const finalTokens = ref(props.data.tokens)
+const content = ref(processPrompt(props.data.content.text))
+const files = ref(props.data.content.files)
+
+// 定义emit事件
+const emit = defineEmits(['edit'])
+
+onMounted(() => {
+  processFiles()
+})
+
+const processFiles = () => {
+  if (!props.data.content) {
+    return
+  }
+  content.value = md.render(content.value.trim())
+}
+const isExternalImg = (link, files) => {
+  return isImage(link) && !files.find((file) => file.url === link)
+}
+const copyContent = (text) => {
+  navigator.clipboard.writeText(text)
+  showMessageSuccess('复制成功')
+}
+</script>
+
+<style lang="scss">
+@use '@/assets/css/markdown/vue.css' as *;
+.chat-page,
+.chat-export {
+  .chat-line-prompt-list {
+    background-color: var(--chat-content-bg-list);
+    color: var(--theme-text-color-primary);
+    justify-content: center;
+    width: 100%;
+    padding-bottom: 1.5rem;
+    padding-top: 1.5rem;
+    // border-bottom: 0.5px solid var(--el-border-color);
+
+    .chat-line-inner {
+      display: flex;
+      width: 100%;
+      max-width: 900px;
+      padding-left: 10px;
+
+      .chat-icon {
+        margin-right: 20px;
+        min-width: 36px;
+
+        img {
+          width: 36px;
+          height: 36px;
+          border-radius: 50%;
+          padding: 1px;
+        }
+      }
+
+      .chat-item {
+        width: 100%;
+        padding: 0 5px 0 0;
+        overflow: hidden;
+
+        .file-list-box {
+          display: flex;
+          flex-flow: column;
+
+          .image {
+            display: flex;
+            flex-flow: row;
+            margin-right: 10px;
+            position: relative;
+            justify-content: start;
+
+            .el-image {
+              border: 1px solid #e3e3e3;
+              border-radius: 10px;
+              margin-bottom: 10px;
+              max-width: 150px;
+              max-height: 150px;
+            }
+          }
+
+          .item {
+            display: flex;
+            flex-flow: row;
+            border-radius: 10px;
+            background-color: var(--chat-content-bg);
+            border: 1px solid #e3e3e3;
+            color: var(--theme-text-color-primary);
+            padding: 6px;
+            margin-bottom: 10px;
+
+            .icon {
+              .el-image {
+                width: 40px;
+                height: 40px;
+              }
+            }
+
+            .body {
+              margin-left: 8px;
+              font-size: 14px;
+
+              .title {
+                font-weight: bold;
+                line-height: 24px;
+                color: #0d0d0d;
+              }
+
+              .info {
+                color: #b4b4b4;
+
+                span {
+                  margin-right: 10px;
+                }
+              }
+            }
+          }
+        }
+
+        .content {
+          word-break: break-word;
+          padding: 0;
+          color: var(--theme-text-color-primary);
+          font-size: var(--content-font-size);
+          border-radius: 5px;
+          overflow: auto;
+
+          img {
+            max-width: 600px;
+            border-radius: 10px;
+            margin: 10px 0;
+          }
+
+          p {
+            line-height: 1.5;
+          }
+
+          p:last-child {
+            margin-bottom: 0;
+          }
+
+          p:first-child {
+            margin-top: 0;
+          }
+        }
+      }
+    }
+  }
+}
+</style>

web/src/components/ChatReply.vue

@@ -1,59 +1,6 @@
 <template>
   <div class="chat-reply">
-    <div class="chat-line chat-line-reply-list" v-if="listStyle === 'list'">
+    <div class="chat-line chat-line-reply-chat">
-      <div class="chat-line-inner">
-        <div class="chat-icon">
-          <img :src="data.icon" alt="ChatGPT" />
-        </div>
-
-        <div class="chat-item">
-          <div
-            class="content-wrapper"
-            v-html="md.render(processContent(data.content.text))"
-            v-if="data.content.text"
-          ></div>
-          <div class="content-wrapper flex justify-start items-center" v-else>
-            <span class="mr-2">AI 思考中</span> <Thinking :duration="1.5" />
-          </div>
-          <div class="bar flex text-gray-500" v-if="data.created_at">
-            <span class="bar-item text-sm">{{ dateFormat(data.created_at) }}</span>
-            <!-- <span class="bar-item">tokens: {{ data.tokens }}</span> -->
-            <span class="bar-item">
-              <el-tooltip class="box-item" effect="dark" content="复制回答" placement="bottom">
-                <el-icon class="copy-reply" :data-clipboard-text="data.content">
-                  <DocumentCopy />
-                </el-icon>
-              </el-tooltip>
-            </span>
-            <span v-if="!readOnly" class="flex">
-              <span class="bar-item" @click="reGenerate(data.id)">
-                <el-tooltip class="box-item" effect="dark" content="重新生成" placement="bottom">
-                  <el-icon><Refresh /></el-icon>
-                </el-tooltip>
-              </span>
-
-              <span class="bar-item">
-                <el-tooltip
-                  class="box-item"
-                  effect="dark"
-                  content="生成语音朗读"
-                  placement="bottom"
-                >
-                  <i
-                    class="iconfont icon-speaker"
-                    v-if="!isPlaying"
-                    @click="synthesis(data.content)"
-                  ></i>
-                  <el-image class="voice-icon" :src="playIcon" v-else />
-                </el-tooltip>
-              </span>
-            </span>
-          </div>
-        </div>
-      </div>
-    </div>
-
-    <div class="chat-line chat-line-reply-chat" v-else>
       <div class="chat-line-inner">
         <div class="chat-icon">
           <img :src="data.icon" alt="ChatGPT" />
@@ -69,23 +16,27 @@
               <span class="mr-2">AI 思考中</span> <Thinking :duration="1.5" />
             </div>
           </div>
-          <div class="bar text-gray-500" v-if="data.created_at">
+          <div
-            <span class="bar-item text-sm"> {{ dateFormat(data.created_at) }}</span>
+            class="flex text-gray-500 text-sm py-2 items-center space-x-2"
-            <span class="bar-item bg">
+            v-if="data.created_at"
-              <el-tooltip class="box-item" effect="dark" content="复制回答" placement="bottom">
+          >
+            <span class="flex items-center"
+              ><i class="iconfont icon-clock mr-1"></i> {{ dateFormat(data.created_at) }}</span
+            >
+            <span class="flex items-center">
+              <el-tooltip class="box-item" effect="dark" content="复制回答" placement="top">
                 <el-icon class="copy-reply" :data-clipboard-text="data.content.text">
                   <DocumentCopy />
                 </el-icon>
               </el-tooltip>
             </span>
-            <span v-if="!readOnly" class="flex">
+            <span class="flex items-center" @click="reGenerate">
-              <span class="bar-item bg" @click="reGenerate(data.id)">
               <el-tooltip class="box-item" effect="dark" content="重新生成" placement="bottom">
-                  <el-icon><Refresh /></el-icon>
+                <i class="iconfont icon-refresh cursor-pointer !text-sm"></i>
               </el-tooltip>
             </span>
 
-              <span class="bar-item bg">
+            <span class="flex items-center">
               <el-tooltip
                 class="box-item"
                 effect="dark"
@@ -93,7 +44,10 @@
                 placement="bottom"
                 v-if="!isPlaying"
               >
-                  <i class="iconfont icon-speaker" @click="synthesis(data.content.text)"></i>
+                <i
+                  class="iconfont icon-speaker !text-sm cursor-pointer"
+                  @click="synthesis(data.content.text)"
+                ></i>
               </el-tooltip>
               <el-tooltip
                 class="box-item"
@@ -105,7 +59,6 @@
                 <el-image class="voice-icon" :src="playIcon" @click="stopSynthesis()" />
               </el-tooltip>
             </span>
-            </span>
           </div>
         </div>
       </div>
@@ -118,7 +71,7 @@
 import { useSharedStore } from '@/store/sharedata'
 import { httpPost } from '@/utils/http'
 import { dateFormat, processContent } from '@/utils/libs'
-import { DocumentCopy, Refresh } from '@element-plus/icons-vue'
+import { DocumentCopy } from '@element-plus/icons-vue'
 import { ElMessage } from 'element-plus'
 import hl from 'highlight.js'
 import MarkdownIt from 'markdown-it'
@@ -141,13 +94,9 @@ const props = defineProps({
       tokens: 0,
     },
   },
-  readOnly: {
+  messageIndex: {
-    type: Boolean,
+    type: Number,
-    default: false,
+    default: -1,
-  },
-  listStyle: {
-    type: String,
-    default: 'list',
   },
 })
 
@@ -232,8 +181,8 @@ const stopSynthesis = () => {
 }
 
 // 重新生成
-const reGenerate = (messageId) => {
+const reGenerate = () => {
-  emits('regen', messageId)
+  emits('regen', props.messageIndex)
 }
 
 // 添加代码块展开/收起功能
@@ -322,227 +271,6 @@ const setupCodeBlockEvents = () => {
     sans-serif;
   font-family: var(--font-family);
 
-  .chat-line {
-    .boxed {
-      border: 1px solid var(--el-border-color);
-      border-radius: 5px;
-      padding: 0 5px;
-    }
-    .chat-item {
-      .content-wrapper {
-        img {
-          max-width: 600px;
-          border-radius: 10px;
-        }
-
-        p {
-          line-height: 1.5;
-
-          code {
-            color: var(--theme-text-color-primary);
-            font-weight: 600;
-          }
-        }
-
-        p:last-child {
-          margin-bottom: 0;
-        }
-
-        p:first-child {
-          margin-top: 0;
-        }
-
-        .code-container {
-          background-color: #2b2b2b;
-          border-radius: 10px;
-          position: relative;
-
-          .hljs {
-            border-radius: 10px;
-            width: 100%;
-          }
-
-          .copy-code-btn {
-            cursor: pointer;
-            font-size: 12px;
-            color: #c1c1c1;
-
-            &:hover {
-              color: #20a0ff;
-            }
-          }
-        }
-
-        // 添加代码块展开/收起样式
-        .code-collapsed {
-          .hljs {
-            max-height: 200px;
-            overflow: hidden;
-            position: relative;
-            transition: max-height 0.3s ease;
-
-            &::after {
-              content: '';
-              position: absolute;
-              bottom: 0;
-              left: 0;
-              right: 0;
-              height: 30px;
-              background: linear-gradient(transparent, #2b2b2b);
-              pointer-events: none;
-            }
-          }
-        }
-
-        .code-expanded {
-          .hljs {
-            max-height: none;
-            overflow: auto;
-            transition: max-height 0.3s ease;
-
-            &::after {
-              display: none;
-            }
-          }
-        }
-
-        .expand-btn {
-          transition: color 0.2s ease;
-
-          &:hover {
-            color: #20a0ff !important;
-          }
-        }
-
-        .lang-name {
-          color: #00e0e0;
-        }
-
-        // 设置表格边框
-
-        table {
-          width: 100%;
-          margin-bottom: 1rem;
-          border-collapse: collapse;
-          border: 1px solid #dee2e6;
-          background-color: var(--chat-content-bg);
-          color: var(--theme-text-color-primary);
-
-          thead {
-            th {
-              border: 1px solid #dee2e6;
-              vertical-align: bottom;
-              border-bottom: 2px solid #dee2e6;
-              padding: 10px;
-            }
-          }
-
-          td {
-            border: 1px solid #dee2e6;
-            padding: 10px;
-          }
-        }
-
-        // 代码快
-
-        blockquote {
-          margin: 0 0 0.8rem 0;
-          background-color: var(--quote-bg-color);
-          padding: 0.8rem 1.5rem;
-          color: var(--quote-text-color);
-          border-left: 0.4rem solid #6b50e1; /* 紫色边框 */
-          font-size: 16px;
-          line-height: 1.6;
-        }
-      }
-    }
-  }
-
-  .chat-line-reply-list {
-    justify-content: center;
-    background-color: var(--chat-content-bg);
-    color: var(--theme-text-color-primary);
-    width: 100%;
-    padding-bottom: 1.5rem;
-    padding-top: 1.5rem;
-    border: 1px solid var(--el-border-color);
-    border-radius: 10px;
-
-    .chat-line-inner {
-      display: flex;
-      width: 100%;
-      max-width: 900px;
-      padding-left: 10px;
-
-      .chat-icon {
-        margin-right: 20px;
-
-        img {
-          width: 36px;
-          height: 36px;
-          border-radius: 50%;
-          padding: 1px;
-        }
-      }
-
-      .chat-item {
-        width: 100%;
-        position: relative;
-        padding: 0;
-        overflow: hidden;
-
-        .content-wrapper {
-          min-height: 20px;
-          word-break: break-word;
-          padding: 0;
-          color: var(--theme-text-color-primary);
-          font-size: var(--content-font-size);
-          border-radius: 5px;
-          overflow: auto;
-        }
-
-        .bar {
-          padding: 10px 10px 10px 0;
-
-          .bar-item {
-            margin-right: 10px;
-            border-radius: 5px;
-            cursor: pointer;
-            display: flex;
-            align-items: center;
-            justify-content: center;
-            height: 26px;
-
-            .voice-icon {
-              width: 20px;
-              height: 20px;
-            }
-
-            .el-icon {
-              position: relative;
-              top: 2px;
-              cursor: pointer;
-            }
-          }
-
-          .el-button {
-            height: 20px;
-            padding: 5px 2px;
-          }
-        }
-      }
-
-      .tool-box {
-        font-size: 16px;
-
-        .el-button {
-          height: 20px;
-          padding: 5px 2px;
-        }
-      }
-    }
-  }
-
   .chat-line-reply-chat {
     justify-content: center;
     padding: 1.5rem;
@@ -584,41 +312,14 @@ const setupCodeBlockEvents = () => {
             background-color: var(--chat-content-bg);
             border-radius: 0 10px 10px 10px;
             width: 100%;
-          }
+
+            p:first-child {
+              margin-top: 0;
             }
 
-        .bar {
+            p:last-child {
-          padding: 10px 10px 10px 0;
+              margin-bottom: 0;
-          display: flex;
-
-          .bar-item {
-            margin-right: 10px;
-            border-radius: 5px;
-            display: flex;
-            align-items: center;
-            justify-content: center;
-            height: 26px;
-
-            .voice-icon {
-              width: 20px;
-              height: 20px;
             }
-
-            .el-icon {
-              position: relative;
-              top: 2px;
-              cursor: pointer;
-            }
-          }
-
-          .bar-item.bg {
-            // background-color var( --gray-btn-bg)
-            cursor: pointer;
-          }
-
-          .el-button {
-            height: 20px;
-            padding: 5px 2px;
           }
         }
       }

web/src/components/ChatReplyLine.vue

@@ -0,0 +1,450 @@
+<template>
+  <div class="chat-reply">
+    <div class="chat-line chat-line-reply-list">
+      <div class="chat-line-inner">
+        <div class="chat-icon">
+          <img :src="data.icon" alt="ChatGPT" />
+        </div>
+
+        <div class="chat-item">
+          <div
+            class="content-wrapper"
+            v-html="md.render(processContent(data.content.text))"
+            v-if="data.content.text"
+          ></div>
+          <div class="content-wrapper flex justify-start items-center" v-else>
+            <span class="mr-2">AI 思考中</span> <Thinking :duration="1.5" />
+          </div>
+          <div
+            class="flex text-gray-500 text-sm py-2 items-center space-x-2"
+            v-if="data.created_at"
+          >
+            <span class="flex items-center">
+              <i class="iconfont icon-clock mr-1"></i> {{ dateFormat(data.created_at) }}</span
+            >
+            <span class="flex items-center">
+              <el-tooltip class="box-item" effect="dark" content="复制回答" placement="top">
+                <i
+                  class="iconfont icon-copy cursor-pointer !text-sm"
+                  :data-clipboard-text="data.content"
+                >
+                </i>
+              </el-tooltip>
+            </span>
+          </div>
+        </div>
+      </div>
+    </div>
+
+    <audio ref="audio" @ended="isPlaying = false" />
+  </div>
+</template>
+
+<script setup>
+import { useSharedStore } from '@/store/sharedata'
+import { httpPost } from '@/utils/http'
+import { dateFormat, processContent } from '@/utils/libs'
+import { ElMessage } from 'element-plus'
+import hl from 'highlight.js'
+import MarkdownIt from 'markdown-it'
+import emoji from 'markdown-it-emoji'
+import mathjaxPlugin from 'markdown-it-mathjax3'
+import { nextTick, onMounted, reactive, ref, watchEffect } from 'vue'
+import Thinking from './Thinking.vue'
+// eslint-disable-next-line no-undef,no-unused-vars
+const props = defineProps({
+  data: {
+    type: Object,
+    default: {
+      type: 'text',
+      icon: '',
+      content: {
+        text: '',
+        files: [],
+      },
+      created_at: '',
+      tokens: 0,
+    },
+  },
+  readOnly: {
+    type: Boolean,
+    default: false,
+  },
+  listStyle: {
+    type: String,
+    default: 'list',
+  },
+})
+
+const audio = ref(null)
+const isPlaying = ref(false)
+const playIcon = ref('/images/voice.gif')
+const store = useSharedStore()
+
+// 添加代码块展开/收起状态管理
+const codeBlockStates = reactive({})
+
+const md = new MarkdownIt({
+  breaks: true,
+  html: true,
+  linkify: true,
+  typographer: true,
+  highlight: function (str, lang) {
+    const codeIndex = parseInt(Date.now()) + Math.floor(Math.random() * 10000000)
+    // 显示复制代码按钮和展开/收起按钮
+    const copyBtn = `<div class="flex">
+      <span class="text-[12px] mr-2 text-[#00e0e0] cursor-pointer expand-btn" data-code-id="${codeIndex}" onclick="window.toggleCodeBlock('${codeIndex}')">收起</span>
+      <span class="copy-code-btn" data-clipboard-action="copy" data-clipboard-target="#copy-target-${codeIndex}">复制</span>
+      </div><textarea style="position: absolute;top: -9999px;left: -9999px;z-index: -9999;" id="copy-target-${codeIndex}">${str.replace(
+      /<\/textarea>/g,
+      '&lt;/textarea>'
+    )}</textarea>`
+    let langHtml = ''
+    let preCode = ''
+    // 处理代码高亮
+    if (lang && hl.getLanguage(lang)) {
+      langHtml = `<span class="lang-name">${lang}</span>`
+      preCode = hl.highlight(str, { language: lang }).value
+    } else {
+      preCode = md.utils.escapeHtml(str)
+    }
+
+    // 将代码包裹在 pre 中，添加展开状态的类（默认展开）
+    return `<pre class="code-container flex flex-col code-expanded" data-code-id="${codeIndex}">
+      <div class="flex justify-between bg-[#50505a] w-full rounded-tl-[10px] rounded-tr-[10px] px-3 py-1">${langHtml}${copyBtn}</div>
+      <code class="language-${lang} hljs">${preCode}</code> 
+      <span class="copy-code-btn absolute right-3 bottom-3" data-clipboard-action="copy" data-clipboard-target="#copy-target-${codeIndex}">复制</span></pre>`
+  },
+})
+md.use(mathjaxPlugin)
+md.use(emoji)
+const emits = defineEmits(['regen'])
+
+if (!props.data.icon) {
+  props.data.icon = 'images/gpt-icon.png'
+}
+
+const synthesis = (text) => {
+  isPlaying.value = true
+  httpPost('/api/chat/tts', { text: text, model_id: store.ttsModel }, { responseType: 'blob' })
+    .then((response) => {
+      // 创建 Blob 对象，明确指定 MIME 类型
+      const blob = new Blob([response], { type: 'audio/mpeg' }) // 假设音频格式为 MP3
+      const audioUrl = URL.createObjectURL(blob)
+      // 播放音频
+      audio.value.src = audioUrl
+      audio.value
+        .play()
+        .then(() => {
+          // 播放完成后释放 URL
+          URL.revokeObjectURL(audioUrl)
+        })
+        .catch(() => {
+          ElMessage.error('音频播放失败，请检查浏览器是否支持该音频格式')
+          isPlaying.value = false
+        })
+    })
+    .catch((e) => {
+      ElMessage.error('语音合成失败：' + e.message)
+      isPlaying.value = false
+    })
+}
+
+const stopSynthesis = () => {
+  isPlaying.value = false
+  audio.value.pause()
+  audio.value.currentTime = 0
+}
+
+// 重新生成
+const reGenerate = (messageId) => {
+  emits('regen', messageId)
+}
+
+// 添加代码块展开/收起功能
+const toggleCodeBlock = (codeId) => {
+  const codeContainer = document.querySelector(`pre[data-code-id="${codeId}"]`)
+  const expandBtn = document.querySelector(`.expand-btn[data-code-id="${codeId}"]`)
+
+  if (codeContainer && expandBtn) {
+    if (codeContainer.classList.contains('code-collapsed')) {
+      codeContainer.classList.remove('code-collapsed')
+      codeContainer.classList.add('code-expanded')
+      expandBtn.textContent = '收起'
+    } else {
+      codeContainer.classList.remove('code-expanded')
+      codeContainer.classList.add('code-collapsed')
+      expandBtn.textContent = '展开'
+    }
+  }
+}
+
+// 将函数暴露到全局作用域
+window.toggleCodeBlock = toggleCodeBlock
+
+// 添加事件监听
+onMounted(() => {
+  nextTick(() => {
+    setupCodeBlockEvents()
+  })
+})
+
+// 监听内容变化，重新绑定事件
+watchEffect(() => {
+  if (props.data.content.text) {
+    nextTick(() => {
+      // 延迟一点时间确保DOM完全渲染
+      setTimeout(() => {
+        setupCodeBlockEvents()
+      }, 100)
+    })
+  }
+})
+
+const setupCodeBlockEvents = () => {
+  // 检查所有代码块并设置展开按钮的显示状态
+  const expandBtns = document.querySelectorAll('.expand-btn')
+
+  expandBtns.forEach((btn) => {
+    const codeId = btn.getAttribute('data-code-id')
+    const codeContainer = document.querySelector(`pre[data-code-id="${codeId}"]`)
+    const codeElement = codeContainer?.querySelector('.hljs')
+
+    if (codeElement) {
+      // 临时移除高度限制来获取真实高度
+      const originalMaxHeight = codeElement.style.maxHeight
+      codeElement.style.maxHeight = 'none'
+      const realHeight = codeElement.scrollHeight
+      codeElement.style.maxHeight = originalMaxHeight
+
+      // 如果代码块高度小于等于200px，隐藏展开按钮
+      if (realHeight <= 200) {
+        btn.style.display = 'none'
+        // 移除收起状态的类，让短代码块完全展示
+        codeContainer.classList.remove('code-collapsed')
+        codeContainer.classList.add('code-expanded')
+      } else {
+        btn.style.display = 'inline'
+        // 确保长代码块默认展开
+        if (
+          !codeContainer.classList.contains('code-expanded') &&
+          !codeContainer.classList.contains('code-collapsed')
+        ) {
+          codeContainer.classList.add('code-expanded')
+        }
+      }
+    }
+  })
+}
+</script>
+
+<style lang="scss">
+@use '@/assets/css/markdown/vue.css' as *;
+
+.chat-page,
+.chat-export {
+  --font-family: Menlo, '微软雅黑', 'Roboto Mono', 'Courier New', Courier, monospace, 'Inter',
+    sans-serif;
+  font-family: var(--font-family);
+
+  .chat-line {
+    .boxed {
+      border: 1px solid var(--el-border-color);
+      border-radius: 5px;
+      padding: 0 5px;
+    }
+    .chat-item {
+      .content-wrapper {
+        img {
+          max-width: 600px;
+          border-radius: 10px;
+        }
+
+        p {
+          line-height: 1.5;
+
+          code {
+            color: var(--theme-text-color-primary);
+            font-weight: 600;
+          }
+        }
+
+        p:last-child {
+          margin-bottom: 0;
+        }
+
+        p:first-child {
+          margin-top: 0;
+        }
+
+        .code-container {
+          background-color: #2b2b2b;
+          border-radius: 10px;
+          position: relative;
+
+          .hljs {
+            border-radius: 10px;
+            width: 100%;
+          }
+
+          .copy-code-btn {
+            cursor: pointer;
+            font-size: 12px;
+            color: #c1c1c1;
+
+            &:hover {
+              color: #20a0ff;
+            }
+          }
+        }
+
+        // 添加代码块展开/收起样式
+        .code-collapsed {
+          .hljs {
+            max-height: 200px;
+            overflow: hidden;
+            position: relative;
+            transition: max-height 0.3s ease;
+
+            &::after {
+              content: '';
+              position: absolute;
+              bottom: 0;
+              left: 0;
+              right: 0;
+              height: 30px;
+              background: linear-gradient(transparent, #2b2b2b);
+              pointer-events: none;
+            }
+          }
+        }
+
+        .code-expanded {
+          .hljs {
+            max-height: none;
+            overflow: auto;
+            transition: max-height 0.3s ease;
+
+            &::after {
+              display: none;
+            }
+          }
+        }
+
+        .expand-btn {
+          transition: color 0.2s ease;
+
+          &:hover {
+            color: #20a0ff !important;
+          }
+        }
+
+        .lang-name {
+          color: #00e0e0;
+        }
+
+        // 设置表格边框
+
+        table {
+          width: 100%;
+          margin-bottom: 1rem;
+          border-collapse: collapse;
+          border: 1px solid #dee2e6;
+          background-color: var(--chat-content-bg);
+          color: var(--theme-text-color-primary);
+
+          thead {
+            th {
+              border: 1px solid #dee2e6;
+              vertical-align: bottom;
+              border-bottom: 2px solid #dee2e6;
+              padding: 10px;
+            }
+          }
+
+          td {
+            border: 1px solid #dee2e6;
+            padding: 10px;
+          }
+        }
+
+        // 代码快
+
+        blockquote {
+          margin: 0 0 0.8rem 0;
+          background-color: var(--quote-bg-color);
+          padding: 0.8rem 1.5rem;
+          color: var(--quote-text-color);
+          border-left: 0.4rem solid #6b50e1; /* 紫色边框 */
+          font-size: 16px;
+          line-height: 1.6;
+        }
+      }
+    }
+  }
+
+  .chat-line-reply-list {
+    justify-content: center;
+    background-color: var(--chat-content-bg);
+    color: var(--theme-text-color-primary);
+    width: 100%;
+    padding-bottom: 0.5rem;
+    padding-top: 1rem;
+    padding-right: 1rem;
+    border: 1px solid var(--el-border-color);
+    border-radius: 10px;
+
+    .chat-line-inner {
+      display: flex;
+      width: 100%;
+      max-width: 900px;
+      padding-left: 10px;
+
+      .chat-icon {
+        margin-right: 20px;
+
+        img {
+          width: 36px;
+          height: 36px;
+          border-radius: 50%;
+          padding: 1px;
+        }
+      }
+
+      .chat-item {
+        width: 100%;
+        position: relative;
+        padding: 0;
+        overflow: hidden;
+
+        .content-wrapper {
+          min-height: 20px;
+          word-break: break-word;
+          padding: 0;
+          color: var(--theme-text-color-primary);
+          font-size: var(--content-font-size);
+          border-radius: 5px;
+          overflow: auto;
+
+          p:first-child {
+            margin-top: 0;
+          }
+
+          p:last-child {
+            margin-bottom: 0;
+          }
+        }
+      }
+
+      .tool-box {
+        font-size: 16px;
+
+        .el-button {
+          height: 20px;
+          padding: 5px 2px;
+        }
+      }
+    }
+  }
+}
+</style>

web/src/components/ChatSetting.vue

@@ -8,21 +8,8 @@
     title="聊天配置"
   >
     <div class="chat-setting">
-      <el-form :model="data" label-width="100px" label-position="left">
+      <el-form :model="data" label-width="100px" label-position="top">
-        <el-form-item label="聊天样式：">
+        <el-form-item label="启用流式输出：">
-          <el-radio-group
-            v-model="data.style"
-            @change="
-              (val) => {
-                store.setChatListStyle(val)
-              }
-            "
-          >
-            <el-radio value="list">列表样式</el-radio>
-            <el-radio value="chat">对话样式</el-radio>
-          </el-radio-group>
-        </el-form-item>
-        <el-form-item label="流式输出：">
           <el-switch
             v-model="data.stream"
             @change="
@@ -32,7 +19,7 @@
             "
           />
         </el-form-item>
-        <el-form-item label="语音音色：">
+        <el-form-item label="朗读语音模型：">
           <el-select v-model="data.ttsModel" placeholder="请选择语音音色" @change="changeTTSModel">
             <el-option v-for="v in models" :value="v.id" :label="v.name" :key="v.id">
               {{ v.name }}
@@ -51,7 +38,6 @@ import { computed, onMounted, ref } from 'vue'
 const store = useSharedStore()
 
 const data = ref({
-  style: store.chatListStyle,
   stream: store.chatStream,
   ttsModel: store.ttsModel,
 })

web/src/components/FileUpload.vue

@@ -0,0 +1,469 @@
+<template>
+  <div class="file__upload-container">
+    <!-- 单文件模式 -->
+    <template v-if="props.maxCount === 1">
+      <div class="single-upload">
+        <div v-if="fileList.length === 0" class="upload-btn upload-btn-single">
+          <el-upload
+            drag
+            :auto-upload="true"
+            :show-file-list="false"
+            :http-request="handleUpload"
+            :multiple="false"
+            :accept="accept"
+            class="uploader"
+          >
+            <div class="upload-placeholder">
+              <el-icon :size="20"><UploadFilled /></el-icon>
+              <span>上传文件</span>
+            </div>
+          </el-upload>
+        </div>
+        <div
+          v-else
+          class="relative inline-flex items-center border border-gray-200 rounded-xl bg-white dark:bg-[#2b2b2b] dark:border-gray-700 p-2 w-full"
+        >
+          <img :src="getFileImage(fileList[0].url)" class="w-10 h-10 mr-2" />
+          <div class="min-w-0 flex flex-col items-center gap-1 text-sm">
+            <a
+              :href="fileList[0].url"
+              target="_blank"
+              class="truncate block text-[var(--theme-text-color-primary,#0d0d0d)] max-w-[220px]"
+            >
+              {{ fileList[0].name }}
+            </a>
+            <div class="text-xs flex w-full justify-start text-gray-500">
+              {{ GetFileType(getFileExt(fileList[0].name)) }} ·
+              {{ FormatFileSize(fileList[0].size || 0) }}
+            </div>
+          </div>
+          <button
+            class="absolute -right-2 -top-2 w-5 h-5 rounded-full bg-rose-600 text-white flex items-center justify-center text-[10px]"
+            @click="removeFile(0)"
+            aria-label="remove"
+          >
+            ×
+          </button>
+        </div>
+      </div>
+    </template>
+
+    <!-- 多文件模式 -->
+    <template v-else>
+      <div class="flex flex-col gap-2 px-2 pt-2 !items-start" v-if="fileList.length > 0">
+        <div
+          v-for="(file, index) in fileList"
+          :key="file.url || index"
+          class="relative inline-flex items-center border border-gray-200 rounded-xl bg-white dark:bg-[#2b2b2b] dark:border-gray-700 p-2 w-full"
+        >
+          <img :src="getFileImage(file.url)" class="w-10 h-10 mr-2" />
+          <div class="min-w-0 flex flex-col items-center gap-1 text-sm">
+            <a :href="file.url" target="_blank" class="truncate block max-w-[180px]">{{
+              file.name
+            }}</a>
+            <div class="text-xs flex w-full justify-start text-gray-500">
+              {{ GetFileType(getFileExt(file.name)) }} · {{ FormatFileSize(file.size || 0) }}
+            </div>
+          </div>
+          <button
+            class="absolute -right-2 -top-2 w-5 h-5 rounded-full bg-rose-600 text-white flex items-center justify-center text-[10px]"
+            @click="removeFile(index)"
+            aria-label="remove"
+          >
+            ×
+          </button>
+        </div>
+        <!-- 上传按钮 -->
+        <div v-if="!multiple || fileList.length < maxCount" class="upload-btn">
+          <el-upload
+            drag
+            :auto-upload="true"
+            :show-file-list="false"
+            :http-request="handleUpload"
+            :multiple="multiple"
+            :accept="accept"
+            class="uploader"
+            :limit="maxCount"
+          >
+            <div class="upload-placeholder">
+              <el-icon :size="20"><UploadFilled /></el-icon>
+              <span>上传文件</span>
+            </div>
+          </el-upload>
+        </div>
+      </div>
+      <!-- 初始上传区域 -->
+      <div v-else class="upload-area">
+        <el-upload
+          drag
+          :auto-upload="true"
+          :show-file-list="false"
+          :http-request="handleUpload"
+          :multiple="multiple"
+          :accept="accept"
+          class="uploader"
+          :limit="maxCount"
+        >
+          <el-icon :size="40" class="el-icon--upload"><UploadFilled /></el-icon>
+          <div class="el-upload__text">拖拽文件到此处，或 <em>点击上传</em></div>
+          <template #tip>
+            <div class="el-upload__tip text-gray-500 text-sm">
+              支持 {{ accept }} 格式，最多上传 {{ maxCount }} 个，单个最大 {{ maxSize }}MB
+            </div>
+          </template>
+        </el-upload>
+      </div>
+    </template>
+
+    <!-- 上传进度 -->
+    <el-progress
+      v-if="uploading"
+      :percentage="uploadProgress"
+      :stroke-width="4"
+      class="upload-progress"
+    />
+  </div>
+</template>
+
+<script setup>
+import { FormatFileSize, GetFileIcon, GetFileType } from '@/store/system'
+import { httpPost } from '@/utils/http'
+import { isImage, replaceImg } from '@/utils/libs'
+import { UploadFilled } from '@element-plus/icons-vue'
+import { ElMessage } from 'element-plus'
+import { computed, ref, watch } from 'vue'
+
+const props = defineProps({
+  modelValue: {
+    type: [String, Array],
+    default: '',
+  },
+  multiple: {
+    type: Boolean,
+    default: false,
+  },
+  maxCount: {
+    type: Number,
+    default: 1,
+  },
+  maxSize: {
+    type: Number,
+    default: 10,
+  },
+  accept: {
+    type: String,
+    default: '.pdf,.doc,.docx,.xls,.xlsx,.ppt,.pptx,.txt,.md,.zip,.rar,.7z',
+  },
+})
+
+const emit = defineEmits(['update:modelValue', 'upload-success', 'remove-file'])
+
+// 上传状态
+const uploading = ref(false)
+const uploadProgress = ref(0)
+const FileInfoList = ref([])
+
+// 文件列表
+const fileList = computed({
+  get() {
+    if (props.multiple || props.maxCount > 1) {
+      return FileInfoList.value
+    } else {
+      return FileInfoList.value && FileInfoList.value.length > 0 ? FileInfoList.value : []
+    }
+  },
+  set(value) {
+    const isMulti = props.multiple || props.maxCount > 1
+    const normalized = Array.isArray(value) ? value : value ? [value] : []
+    FileInfoList.value = normalized
+    if (isMulti) {
+      const urls = normalized.map((v) => v && v.url).filter((u) => !!u)
+      emit('update:modelValue', urls)
+    } else {
+      const url =
+        normalized.length > 0 && normalized[0] && normalized[0].url ? normalized[0].url : ''
+      emit('update:modelValue', url)
+    }
+  },
+})
+
+const uploadCount = ref(1)
+
+// 获取文件扩展名
+const getFileExt = (filename) => {
+  return '.' + filename.split('.').pop().toLowerCase()
+}
+
+const getFileName = (url) => {
+  return url.split('/').pop()
+}
+
+// 获取文件
+const getFileImage = (url) => {
+  return isImage(url) ? url : GetFileIcon(getFileExt(url))
+}
+
+// 将外部 modelValue 同步为内部文件对象列表
+const urlToFileInfo = (url) => ({
+  url,
+  name: getFileName(url),
+  size: 0,
+  ext: getFileExt(url),
+})
+
+// 通过 HEAD 请求尝试获取远程资源大小
+const fetchRemoteFileSize = async (url) => {
+  try {
+    const res = await fetch(url, { method: 'HEAD' })
+    const len = res.headers.get('content-length')
+    return len ? parseInt(len, 10) : 0
+  } catch (e) {
+    return 0
+  }
+}
+
+// 对 size 为空或 0 的项进行补充
+const updateUnknownSizes = async (items) => {
+  const tasks = items.map(async (it) => {
+    if (!it || !it.url) return it
+    if (!it.size || it.size === 0) {
+      const s = await fetchRemoteFileSize(it.url)
+      if (s > 0) {
+        it.size = s
+      }
+    }
+    return it
+  })
+  await Promise.all(tasks)
+}
+
+watch(
+  () => props.modelValue,
+  (newVal) => {
+    const isMulti = props.multiple || props.maxCount > 1
+    if (isMulti) {
+      const urls = Array.isArray(newVal) ? newVal : []
+      FileInfoList.value = urls.map((u) => urlToFileInfo(u))
+      // 异步补齐大小
+      updateUnknownSizes(FileInfoList.value)
+    } else {
+      const url = typeof newVal === 'string' ? newVal : ''
+      FileInfoList.value = url ? [urlToFileInfo(url)] : []
+      // 异步补齐大小
+      updateUnknownSizes(FileInfoList.value)
+    }
+  },
+  { immediate: true }
+)
+
+// 处理上传
+const handleUpload = async (uploadFile) => {
+  const file = uploadFile.file
+  // 检查文件大小
+  if (file.size > props.maxSize * 1024 * 1024) {
+    ElMessage.error(`文件大小不能超过 ${props.maxSize}MB`)
+    return
+  }
+
+  // 检查数量限制
+  if (uploadCount.value > props.maxCount) {
+    ElMessage.error(`最多只能上传 ${props.maxCount} 个文件`)
+    return
+  }
+  uploadCount.value++
+
+  uploading.value = true
+  uploadProgress.value = 0
+
+  try {
+    const formData = new FormData()
+    formData.append('file', file)
+
+    // 模拟上传进度
+    const progressTimer = setInterval(() => {
+      if (uploadProgress.value < 90) {
+        uploadProgress.value += 10
+      }
+    }, 100)
+
+    const response = await httpPost('/api/upload', formData)
+
+    clearInterval(progressTimer)
+    uploadProgress.value = 100
+
+    const fileUrl = replaceImg(response.data.url)
+    const fileInfo = {
+      name: file.name,
+      url: fileUrl,
+      size: file.size,
+      ext: getFileExt(file.name),
+    }
+
+    // 更新文件列表
+    if (props.multiple || props.maxCount > 1) {
+      const newList = [...fileList.value, fileInfo]
+      fileList.value = newList
+    } else {
+      fileList.value = [fileInfo]
+    }
+
+    emit('upload-success', fileInfo)
+    ElMessage.success('上传成功')
+  } catch (error) {
+    ElMessage.error('上传失败: ' + (error.message || '网络错误'))
+  } finally {
+    uploading.value = false
+    uploadProgress.value = 0
+  }
+}
+
+// 移除文件
+const removeFile = (index) => {
+  const file = fileList.value[index]
+  const newList = [...fileList.value]
+  newList.splice(index, 1)
+  fileList.value = newList
+  uploadCount.value--
+  emit('remove-file', file)
+}
+</script>
+
+<style lang="scss">
+.file__upload-container {
+  width: 100%;
+
+  .single-upload {
+    width: 100%;
+    position: relative;
+
+    .upload-btn-single {
+      .uploader {
+        width: 100%;
+
+        .el-upload-dragger {
+          width: 100px;
+          height: 100px;
+          display: flex;
+          align-items: center;
+          justify-content: center;
+        }
+      }
+    }
+  }
+
+  // .single-file-item {
+  //   width: 100%;
+  //   position: relative;
+  //   border-radius: 6px;
+  //   overflow: hidden;
+  //   border: 1px solid #dcdfe6;
+  //   background-color: var(--chat-content-bg, #f5f5f5);
+  //   padding: 8px;
+  // }
+
+  .upload-list {
+    display: flex;
+    flex-wrap: wrap;
+    gap: 10px;
+  }
+
+  .upload-item {
+    position: relative;
+    width: 100%;
+    max-width: 300px;
+    border-radius: 6px;
+    overflow: hidden;
+    border: 1px solid #dcdfe6;
+    background-color: var(--chat-content-bg, #f5f5f5);
+    padding: 8px;
+
+    .file-info {
+      display: flex;
+      flex-direction: row;
+      align-items: center;
+
+      .icon {
+        .el-image {
+          width: 40px;
+          height: 40px;
+        }
+      }
+
+      .body {
+        margin-left: 8px;
+        flex: 1;
+        font-size: 12px;
+
+        .title {
+          color: #0d0d0d;
+          margin-bottom: 4px;
+        }
+
+        .info {
+          color: #b4b4b4;
+
+          span {
+            margin-right: 10px;
+          }
+        }
+      }
+    }
+
+    .upload-overlay {
+      position: absolute;
+      top: -8px;
+      right: -8px;
+      opacity: 1;
+
+      .remove-btn {
+        background: rgba(245, 108, 108, 0.8);
+        border: none;
+        color: white;
+      }
+    }
+  }
+
+  .upload-btn {
+    .uploader {
+      width: 100%;
+
+      .el-upload-dragger {
+        display: flex;
+        align-items: center;
+        justify-content: center;
+      }
+    }
+
+    .upload-placeholder {
+      display: flex;
+      flex-direction: column;
+      align-items: center;
+      gap: 5px;
+      font-size: 12px;
+      color: #8c939d;
+    }
+  }
+
+  .upload-area {
+    .el-upload-dragger {
+      width: 100%;
+    }
+
+    .uploader {
+      width: 100%;
+    }
+  }
+
+  .upload-progress {
+    margin-top: 10px;
+  }
+
+  :deep(.el-upload) {
+    width: 100%;
+    height: 100%;
+    display: flex;
+    align-items: center;
+    justify-content: center;
+  }
+}
+</style>

web/src/components/FooterBar.vue

@@ -2,10 +2,10 @@
   <div class="foot-container">
     <div class="footer text-base">
       <div>
-        <a :href="gitURL" target="_blank">
+        <span>
           {{ title }} -
           {{ version }}
-        </a>
+        </span>
       </div>
       <div class="flex justify-center text-sm">
         <span class="mr-2">{{ copyRight }}</span>

web/src/components/ImageUpload.vue

@@ -1,5 +1,5 @@
 <template>
-  <div class="image-upload">
+  <div class="image__upload-container">
     <!-- 单图模式 -->
     <template v-if="props.maxCount === 1">
       <div class="single-upload">
@@ -59,7 +59,7 @@
             :show-file-list="false"
             :http-request="handleUpload"
             :multiple="multiple"
-            accept="image/*"
+            :accept="accept"
             class="uploader"
             :limit="maxCount"
           >
@@ -78,7 +78,7 @@
           :show-file-list="false"
           :http-request="handleUpload"
           :multiple="multiple"
-          accept="image/*"
+          :accept="accept"
           class="uploader"
           :limit="maxCount"
         >
@@ -86,7 +86,7 @@
           <div class="el-upload__text">拖拽图片到此处，或 <em>点击上传</em></div>
           <template #tip>
             <div class="el-upload__tip text-gray-500 text-sm">
-              支持 JPG、PNG 格式，最多上传 {{ maxCount }} 张，单张最大 5MB
+              支持 {{ accept }} 格式，最多上传 {{ maxCount }} 张，单张最大 {{ maxSize }}MB
             </div>
           </template>
         </el-upload>
@@ -123,6 +123,14 @@ const props = defineProps({
     type: Number,
     default: 1,
   },
+  maxSize: {
+    type: Number,
+    default: 5,
+  },
+  accept: {
+    type: String,
+    default: '.png,.jpg,.jpeg',
+  },
 })
 
 const emit = defineEmits(['update:modelValue', 'upload-success'])
@@ -137,7 +145,7 @@ const imageList = computed({
     if (props.multiple || props.maxCount > 1) {
       return Array.isArray(props.modelValue) ? props.modelValue : []
     } else {
-      return props.modelValue ? [props.modelValue] : []
+      return props.modelValue && props.modelValue.length > 0 ? [props.modelValue] : []
     }
   },
   set(value) {
@@ -149,7 +157,7 @@ const imageList = computed({
   },
 })
 
-const uploadCount = ref(1)
+// 使用已选图片数量进行限制，不再使用全局计数
 // 处理上传
 const handleUpload = async (uploadFile) => {
   const file = uploadFile.file
@@ -161,17 +169,16 @@ const handleUpload = async (uploadFile) => {
   }
 
   // 检查文件大小 (5MB)
-  if (file.size > 5 * 1024 * 1024) {
+  if (file.size > props.maxSize * 1024 * 1024) {
-    ElMessage.error('图片大小不能超过 5MB')
+    ElMessage.error(`图片大小不能超过 ${props.maxSize}MB`)
     return
   }
 
-  // 检查数量限制
+  // 检查数量限制（单图或多图）
-  if (uploadCount.value > props.maxCount) {
+  if ((props.multiple || props.maxCount > 1) && imageList.value.length >= props.maxCount) {
     ElMessage.error(`最多只能上传 ${props.maxCount} 张图片`)
     return
   }
-  uploadCount.value++
 
   uploading.value = true
   uploadProgress.value = 0
@@ -217,14 +224,12 @@ const removeImage = (index) => {
   const newList = [...imageList.value]
   newList.splice(index, 1)
   imageList.value = newList
-  uploadCount.value--
 }
 </script>
 
 <style lang="scss">
-.image-upload {
+.image__upload-container {
   width: 100%;
-}
 
   .single-upload {
     width: 100px;
@@ -324,4 +329,5 @@ const removeImage = (index) => {
     align-items: center;
     justify-content: center;
   }
+}
 </style>

web/src/components/ParamBuilder.vue

@@ -0,0 +1,315 @@
+<template>
+  <div class="param-builder flex flex-col">
+    <ParamEmpty
+      v-if="items.length === 0"
+      :progress="progress"
+      :title="title"
+      :status-text="statusText"
+      :description="description"
+    />
+    <div v-else class="flex flex-col w-full space-y-5">
+      <el-select
+        v-model="selectedModel"
+        placeholder="请选择模型"
+        @change="changeModel"
+        popper-class="model-select"
+        value-key="name"
+      >
+        <template #prefix>
+          <i class="iconfont icon-model"></i>
+        </template>
+
+        <el-option v-for="item in items" :key="item.name" :label="item.name" :value="item">
+          <div class="flex justify-start">
+            <span
+              class="flex items-center justify-center text-white model-version mr-2 w-[40px] h-[40px] rounded-lg"
+              :class="item.icon.size ? item.icon.size : '!text-xl'"
+              >{{ item.icon.text }}</span
+            >
+            <div class="flex !items-start flex-col py-2 space-y-1">
+              <span class="label text-sm">{{ item.name }}</span>
+              <div class="whitespace-pre-line">
+                <span
+                  class="text-xs text-gray-500 break-words line-clamp-1 max-w-[250px]"
+                  :title="item.label"
+                  >{{ item.label }}</span
+                >
+              </div>
+            </div>
+          </div>
+        </el-option>
+      </el-select>
+      <template v-for="param in selectedModel.params">
+        <div class="w-full" :key="param.name" v-if="param.type !== 'hidden'">
+          <div class="w-full flex flex-col !items-start space-y-2" v-if="param.type === 'switch'">
+            <div class="w-full flex justify-between">
+              <label class="label font-bold">{{ param.label }}</label>
+              <el-switch v-model="modelValue[param.name]" size="large" />
+            </div>
+            <p v-if="param.info" class="text-xs text-gray-500 mb-1">{{ param.info }}</p>
+          </div>
+          <div class="w-full flex flex-col !items-start space-y-2" v-else>
+            <label class="label font-bold">
+              {{ param.label }}
+              <span v-if="param.required" class="text-red-500 ml-1">*</span>
+            </label>
+            <p v-if="param.info" class="text-xs text-gray-500 mb-1">{{ param.info }}</p>
+            <div class="flex w-full">
+              <el-input
+                v-if="param.type === 'text'"
+                v-model="modelValue[param.name]"
+                :placeholder="param.placeholder"
+              />
+              <el-input-number
+                v-if="param.type === 'number'"
+                v-model="modelValue[param.name]"
+                class="!w-full"
+                :placeholder="param.placeholder"
+                :min="param.min"
+                :max="param.max"
+                :step="param.step"
+              />
+              <el-slider
+                v-if="param.type === 'slider'"
+                v-model="modelValue[param.name]"
+                :min="param.min"
+                :max="param.max"
+                :step="param.step"
+              />
+              <el-date-picker
+                v-if="param.type === 'date'"
+                v-model="modelValue[param.name]"
+                :placeholder="param.placeholder"
+              />
+              <el-time-picker
+                v-if="param.type === 'time'"
+                v-model="modelValue[param.name]"
+                :placeholder="param.placeholder"
+              />
+              <el-select
+                v-if="param.type === 'select'"
+                v-model="modelValue[param.name]"
+                :placeholder="param.placeholder"
+                :popper-class="param.popperClass"
+                filterable
+              >
+                <template #prefix v-if="param.prefix">
+                  <i class="iconfont !text-lg" :class="param.prefix"></i>
+                </template>
+                <el-option
+                  v-for="option in param.options"
+                  :key="option.value"
+                  :label="option.label"
+                  :value="option.value"
+                >
+                  <div class="flex justify-start" v-if="option.image">
+                    <span class="flex py-3 mr-2">
+                      <img
+                        :src="option.image"
+                        class="rounded-lg"
+                        :style="{ width: param.imgSize, height: param.imgSize }"
+                    /></span>
+                    <div class="flex !items-start flex-col py-2 space-y-1">
+                      <span class="label text-sm">{{ option.label }}</span>
+                      <span
+                        class="text-xs text-gray-500 break-words line-clamp-1 max-w-[200px]"
+                        :title="option.value"
+                        >{{ option.value }}</span
+                      >
+                    </div>
+                  </div>
+                  <div class="flex justify-start items-center h-full" v-else>
+                    <span class="label text-sm">{{ option.label }}</span>
+                  </div>
+                </el-option>
+              </el-select>
+              <el-input
+                type="textarea"
+                v-if="param.type === 'textarea'"
+                v-model="modelValue[param.name]"
+                :autosize="param.autosize"
+                :maxlength="param.maxlength"
+                :show-word-limit="param.showWordLimit"
+                :placeholder="param.placeholder"
+              />
+              <ImageUpload
+                v-if="param.type === 'image'"
+                v-model="modelValue[param.name]"
+                :max-count="param.maxCount"
+                :multiple="param.multiple"
+                :max-size="param.maxSize"
+                :accept="param.accept"
+              />
+              <FileUpload
+                v-if="param.type === 'file'"
+                v-model="modelValue[param.name]"
+                :max-count="param.maxCount"
+                :multiple="param.multiple"
+                :max-size="param.maxSize"
+                :accept="param.accept"
+              />
+            </div>
+          </div>
+        </div>
+      </template>
+    </div>
+  </div>
+</template>
+
+<script setup>
+import FileUpload from './FileUpload.vue'
+import ImageUpload from './ImageUpload.vue'
+import ParamEmpty from './ui/ParamEmpty.vue'
+
+const title = ref('参数构建器')
+const statusText = ref('功能正在开发中')
+const description = ref('我们正在努力完善当前功能，敬请期待！')
+
+const props = defineProps({
+  modelValue: {
+    type: Object,
+    required: true,
+  },
+  requiredKeys: {
+    type: Object,
+    default: {},
+    required: false,
+  },
+  items: {
+    type: Array,
+    required: true,
+  },
+  progress: {
+    type: Number,
+    default: 65,
+    validator: (value) => value >= 0 && value <= 100,
+  },
+})
+
+const selectedModel = ref(props.items[0])
+const requiredKeys = ref(props.requiredKeys)
+
+const emit = defineEmits(['update:modelValue', 'update:requiredKeys'])
+
+// 初始化 modelValue 默认值
+const initModelValue = (model) => {
+  if (props.items.length === 0) {
+    return {}
+  }
+  const defaultValues = {}
+  requiredKeys.value = {}
+  if (model && model.params) {
+    model.params.forEach((param) => {
+      if (param.required) {
+        requiredKeys.value[param.name] = { required: true, label: param.label }
+      }
+      // 根据参数类型设置默认值
+      switch (param.type) {
+        case 'text':
+        case 'textarea':
+          defaultValues[param.name] = param.value || ''
+          break
+        case 'number':
+          defaultValues[param.name] = param.value || 0
+          break
+        case 'slider':
+          defaultValues[param.name] = param.value || param.min || 0
+          break
+        case 'select':
+          // 如果有选项，选择第一个选项作为默认值
+          defaultValues[param.name] =
+            param.value || (param.options && param.options[0] ? param.options[0].value : '')
+          break
+        case 'checkbox':
+        case 'switch':
+          defaultValues[param.name] = param.value || false
+          break
+        case 'date':
+        case 'time':
+          defaultValues[param.name] = param.value || null
+          break
+        case 'image':
+          defaultValues[param.name] = param.value || []
+          break
+        default:
+          defaultValues[param.name] = param.value || ''
+      }
+    })
+  }
+  // 初始化 req_key 和 action
+  defaultValues.req_key = selectedModel.value.key
+  defaultValues.action = selectedModel.value.action
+    ? selectedModel.value.action
+    : 'CVSync2AsyncSubmitTask'
+  return defaultValues
+}
+
+// 初始化默认值
+const modelValue = ref(initModelValue(selectedModel.value))
+
+// 监听 modelValue 变化，通知父组件
+watch(
+  modelValue,
+  (newValue) => {
+    emit('update:modelValue', newValue)
+  },
+  { deep: true }
+)
+
+watch(
+  requiredKeys,
+  (newValue) => {
+    emit('update:requiredKeys', newValue)
+  },
+  { deep: true }
+)
+
+watch(
+  () => props.items,
+  (newValue) => {
+    selectedModel.value = newValue[0]
+    modelValue.value = initModelValue(selectedModel.value)
+  },
+  { deep: true }
+)
+
+// 组件挂载时初始化
+onMounted(() => {
+  // 确保初始值被正确设置
+  if (props.modelValue && Object.keys(props.modelValue).length > 0) {
+    modelValue.value = { ...props.modelValue }
+  } else {
+    modelValue.value = initModelValue(selectedModel.value)
+  }
+})
+
+const changeModel = (item) => {
+  if (item) {
+    selectedModel.value = item
+    // 更新 modelValue 为选中模型的默认值
+    modelValue.value = initModelValue(item)
+  }
+}
+</script>
+
+<style lang="scss">
+.param-builder {
+  .model-version {
+    background: url('@/assets/img/model-version.png') no-repeat center center;
+    background-size: cover;
+  }
+  .el-select__wrapper {
+    min-height: 34px;
+    line-height: 25px;
+  }
+}
+.model-select {
+  .el-select-dropdown__item {
+    height: auto !important;
+  }
+  .model-version {
+    background: url('@/assets/img/model-version.png') no-repeat center center;
+    background-size: cover;
+  }
+}
+</style>

web/src/components/admin/AdminSidebar.vue

@@ -279,7 +279,7 @@ const items = [
       {
         icon: 'xmind',
         index: '/admin/config/markmap',
-        title: '思维导图配置',
+        title: '思维导图',
       },
     ],
   },