feat: add funcitons manger page

api/service/mj/service.go

@@ -54,7 +54,6 @@ func (s *Service) Run() {
 		err := s.taskQueue.LPop(&task)
 		if err != nil {
 			logger.Errorf("taking task with error: %v", err)
-			s.db.Model(&model.MidJourneyJob{Id: uint(task.Id)}).UpdateColumn("progress", -1)
 			continue
 		}
 
@@ -83,6 +82,8 @@ func (s *Service) Run() {
 			logger.Error("绘画任务执行失败：", err)
 			// update the task progress
 			s.db.Model(&model.MidJourneyJob{Id: uint(task.Id)}).UpdateColumn("progress", -1)
+			// restore img_call quota
+			s.db.Model(&model.User{}).Where("id = ?", task.UserId).UpdateColumn("img_calls", gorm.Expr("img_calls + ?", 1))
 			continue
 		}
 
@@ -157,8 +158,6 @@ func (s *Service) Notify(data CBReq) {
 	}
 
 	if data.Status == Finished {
-		// update user's img calls
-		s.db.Model(&model.User{}).Where("id = ?", job.UserId).UpdateColumn("img_calls", gorm.Expr("img_calls - ?", 1))
 		// release lock task
 		atomic.AddInt32(&s.handledTaskNum, -1)
 	}

api/service/sd/pool.go

@@ -26,7 +26,7 @@ func NewServicePool(db *gorm.DB, redisCli *redis.Client, manager *oss.UploaderMa
 
 		// create sd service
 		name := fmt.Sprintf("StableDifffusion Service-%d", k)
-		service := NewService(name, 4, 600, config, queue, db, manager)
+		service := NewService(name, 1, 300, config, queue, db, manager)
 		// run sd service
 		go func() {
 			service.Run()

api/service/sd/service.go

@@ -69,6 +69,8 @@ func (s *Service) Run() {
 			logger.Error("绘画任务执行失败：", err)
 			// update the task progress
 			s.db.Model(&model.SdJob{Id: uint(task.Id)}).UpdateColumn("progress", -1)
+			// restore img_call quota
+			s.db.Model(&model.User{}).Where("id = ?", task.UserId).UpdateColumn("img_calls", gorm.Expr("img_calls + ?", 1))
 			// release task num
 			atomic.AddInt32(&s.handledTaskNum, -1)
 			continue

api/service/sd/types.go

@@ -32,14 +32,14 @@ var ParamKeys = map[string]int{
 	"negative_prompt": 2,
 	"steps":           4,
 	"sampler":         5,
-	"face_fix":        6, // 面部修复
-	"cfg_scale":       10,
-	"seed":            11,
-	"height":          17,
-	"width":           18,
-	"hd_fix":          19,
-	"hd_redraw_rate":  20, //高清修复重绘幅度
-	"hd_scale":        21, // 高清修复放大倍数
-	"hd_scale_alg":    22, // 高清修复放大算法
-	"hd_sample_num":   23, // 高清修复采样次数
+	"face_fix":        7, // 面部修复
+	"cfg_scale":       8,
+	"seed":            27,
+	"height":          10,
+	"width":           9,
+	"hd_fix":          11,
+	"hd_redraw_rate":  12, //高清修复重绘幅度
+	"hd_scale":        13, // 高清修复放大倍数
+	"hd_scale_alg":    14, // 高清修复放大算法
+	"hd_sample_num":   15, // 高清修复采样次数
 }