xinyin025/geekai
1
0
Fork 0
mirror of https://github.com/yangjian102621/geekai.git synced 2025-11-06 17:23:42 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

some page ui optimization, add release v3.1.3

Browse Source
This commit is contained in:
RockYang
2023-09-20 11:39:11 +08:00
parent 59f316b341
commit 79ded6018b
17 changed files with 680 additions and 162 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
api/handler/mj_handler.go
View File

@@ -177,9 +177,11 @@ func (h *MidJourneyHandler) notifyHandler(c *gin.Context, data notifyData) (erro
var jobVo vo.MidJourneyJob
err := utils.CopyObject(job, &jobVo)
if err == nil {
image, err := utils.DownloadImage(jobVo.ImgURL, h.App.Config.ProxyURL)
if err == nil {
jobVo.ImgURL = "data:image/png;base64," + base64.StdEncoding.EncodeToString(image)
if data.Progress < 100 {
image, err := utils.DownloadImage(jobVo.ImgURL, h.App.Config.ProxyURL)
if err == nil {
jobVo.ImgURL = "data:image/png;base64," + base64.StdEncoding.EncodeToString(image)
}
}
// 推送任务到前端
@@ -265,7 +267,8 @@ func (h *MidJourneyHandler) notifyHandler(c *gin.Context, data notifyData) (erro
}
// 更新用户剩余绘图次数
if data.Status == Finished && task.Type != service.Upscale {
// TODO: 放大图片是否需要消耗绘图次数？
if data.Status == Finished {
h.db.Model(&model.User{}).Where("id = ?", task.UserId).UpdateColumn("img_calls", gorm.Expr("img_calls - ?", 1))
}
@@ -306,7 +309,7 @@ func (h *MidJourneyHandler) Image(c *gin.Context) {
resp.ERROR(c, types.InvalidArgs)
return
}
if h.checkLimits(c) {
if !h.checkLimits(c) {
return
}
@@ -391,7 +394,7 @@ func (h *MidJourneyHandler) Upscale(c *gin.Context) {
return
}
if h.checkLimits(c) {
if !h.checkLimits(c) {
return
}
@@ -460,7 +463,7 @@ func (h *MidJourneyHandler) Variation(c *gin.Context) {
return
}
if h.checkLimits(c) {
if !h.checkLimits(c) {
return
}