xinyin025/geekai
1
0
Fork 0
mirror of https://github.com/yangjian102621/geekai.git synced 2025-12-25 01:25:59 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

merge luma page code for v4.1.3

Browse Source
This commit is contained in:
RockYang
2024-09-09 18:07:10 +08:00
parent d4e00b960c
commit 5eb4ed157e
22 changed files with 1285 additions and 367 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
api/handler/sd_handler.go
View File

@@ -295,7 +295,7 @@ func (h *SdJobHandler) Remove(c *gin.Context) {
// 如果任务未完成，或者任务失败，则恢复用户算力
if job.Progress != 100 {
err := h.userService.DecreasePower(job.UserId, job.Power, model.PowerLog{
err := h.userService.IncreasePower(job.UserId, job.Power, model.PowerLog{
Type: types.PowerRefund,
Model: "stable-diffusion",
Remark: fmt.Sprintf("任务失败，退回算力。任务ID：%s， Err: %s", job.TaskId, job.ErrMsg),

2
api/handler/suno_handler.go
View File

@@ -235,7 +235,7 @@ func (h *SunoHandler) Remove(c *gin.Context) {
// 如果任务未完成，或者任务失败，则恢复用户算力
if job.Progress != 100 {
err := h.userService.DecreasePower(job.UserId, job.Power, model.PowerLog{
err := h.userService.IncreasePower(job.UserId, job.Power, model.PowerLog{
Type: types.PowerRefund,
Model: job.ModelName,
Remark: fmt.Sprintf("Suno 任务失败，退回算力。任务ID：%s，Err:%s", job.TaskId, job.ErrMsg),

2
api/handler/upload_handler.go
View File

@@ -142,7 +142,7 @@ func (h *NetHandler) Download(c *gin.Context) {
return
}
// 将下载的文件内容写入响应
c.Status(http.StatusOK)
// 将下载的文件内容写入响应
_, _ = io.Copy(c.Writer, r.Body)
}

2
api/handler/video_handler.go
View File

@@ -194,7 +194,7 @@ func (h *VideoHandler) Remove(c *gin.Context) {
// 如果任务未完成，或者任务失败，则恢复用户算力
if job.Progress != 100 {
err = h.userService.DecreasePower(job.UserId, job.Power, model.PowerLog{
err = h.userService.IncreasePower(job.UserId, job.Power, model.PowerLog{
Type: types.PowerRefund,
Model: "luma",
Remark: fmt.Sprintf("Luma 任务失败，退回算力。任务ID：%s，Err:%s", job.TaskId, job.ErrMsg),