fine-tune the new UI theme

api/core/types/config.go

@@ -143,13 +143,14 @@ type SystemConfig struct {
 	OrderPayTimeout int    `json:"order_pay_timeout,omitempty"` //订单支付超时时间
 	VipInfoText     string `json:"vip_info_text,omitempty"`     // 会员页面充值说明
 
-	MjPower       int `json:"mj_power,omitempty"`        // MJ 绘画消耗算力
-	MjActionPower int `json:"mj_action_power,omitempty"` // MJ 操作（放大，变换）消耗算力
-	SdPower       int `json:"sd_power,omitempty"`        // SD 绘画消耗算力
-	DallPower     int `json:"dall_power,omitempty"`      // DALL-E-3 绘图消耗算力
-	SunoPower     int `json:"suno_power,omitempty"`      // Suno 生成歌曲消耗算力
-	LumaPower     int `json:"luma_power,omitempty"`      // Luma 生成视频消耗算力
+	MjPower           int `json:"mj_power,omitempty"`            // MJ 绘画消耗算力
+	MjActionPower     int `json:"mj_action_power,omitempty"`     // MJ 操作（放大，变换）消耗算力
+	SdPower           int `json:"sd_power,omitempty"`            // SD 绘画消耗算力
+	DallPower         int `json:"dall_power,omitempty"`          // DALL-E-3 绘图消耗算力
+	SunoPower         int `json:"suno_power,omitempty"`          // Suno 生成歌曲消耗算力
+	LumaPower         int `json:"luma_power,omitempty"`          // Luma 生成视频消耗算力
 	AdvanceVoicePower int `json:"advance_voice_power,omitempty"` // 高级语音对话消耗算力
+	PromptPower       int `json:"prompt_power,omitempty"`        // 生成提示词消耗算力
 
 	WechatCardURL string `json:"wechat_card_url,omitempty"` // 微信客服地址
 

api/handler/prompt_handler.go

@@ -12,13 +12,13 @@ import (
 	"geekai/core"
 	"geekai/core/types"
 	"geekai/service"
-	"geekai/service/oss"
-	"geekai/service/suno"
+	"geekai/store/model"
 	"geekai/utils"
 	"geekai/utils/resp"
+	"strings"
+
 	"github.com/gin-gonic/gin"
 	"gorm.io/gorm"
-	"strings"
 )
 
 // 提示词生成 handler
@@ -26,8 +26,6 @@ import (
 
 type PromptHandler struct {
 	BaseHandler
-	sunoService *suno.Service
-	uploader    *oss.UploaderManager
 	userService *service.UserService
 }
 
@@ -56,6 +54,15 @@ func (h *PromptHandler) Lyric(c *gin.Context) {
 		return
 	}
 
+	if h.App.SysConfig.PromptPower > 0 {
+		userId := h.GetLoginUserId(c)
+		h.userService.DecreasePower(int(userId), h.App.SysConfig.PromptPower, model.PowerLog{
+			Type:   types.PowerConsume,
+			Model:  h.getPromptModel(),
+			Remark: "生成歌词",
+		})
+	}
+
 	resp.SUCCESS(c, content)
 }
 
@@ -73,7 +80,14 @@ func (h *PromptHandler) Image(c *gin.Context) {
 		resp.ERROR(c, err.Error())
 		return
 	}
-
+	if h.App.SysConfig.PromptPower > 0 {
+		userId := h.GetLoginUserId(c)
+		h.userService.DecreasePower(int(userId), h.App.SysConfig.PromptPower, model.PowerLog{
+			Type:   types.PowerConsume,
+			Model:  h.getPromptModel(),
+			Remark: "生成绘画提示词",
+		})
+	}
 	resp.SUCCESS(c, strings.Trim(content, `"`))
 }
 
@@ -92,6 +106,15 @@ func (h *PromptHandler) Video(c *gin.Context) {
 		return
 	}
 
+	if h.App.SysConfig.PromptPower > 0 {
+		userId := h.GetLoginUserId(c)
+		h.userService.DecreasePower(int(userId), h.App.SysConfig.PromptPower, model.PowerLog{
+			Type:   types.PowerConsume,
+			Model:  h.getPromptModel(),
+			Remark: "生成视频脚本",
+		})
+	}
+
 	resp.SUCCESS(c, strings.Trim(content, `"`))
 }
 
@@ -121,3 +144,12 @@ func (h *PromptHandler) MetaPrompt(c *gin.Context) {
 
 	resp.SUCCESS(c, strings.Trim(content, `"`))
 }
+
+func (h *PromptHandler) getPromptModel() string {
+	if h.App.SysConfig.TranslateModelId > 0 {
+		var chatModel model.ChatModel
+		h.DB.Where("id", h.App.SysConfig.TranslateModelId).First(&chatModel)
+		return chatModel.Value
+	}
+	return "gpt-4o"
+}