refactor: use waterflow component in mj, sd and dall image drawing page

api/service/sd/pool.go

@@ -25,8 +25,8 @@ type ServicePool struct {
 	notifyQueue *store.RedisQueue
 	db          *gorm.DB
 	Clients     *types.LMap[uint, *types.WsClient] // UserId => Client
-	uploader *oss.UploaderManager
-	levelDB  *store.LevelDB
+	uploader    *oss.UploaderManager
+	levelDB     *store.LevelDB
 }
 
 func NewServicePool(db *gorm.DB, redisCli *redis.Client, manager *oss.UploaderManager, levelDB *store.LevelDB) *ServicePool {
@@ -40,8 +40,8 @@ func NewServicePool(db *gorm.DB, redisCli *redis.Client, manager *oss.UploaderMa
 		services:    services,
 		db:          db,
 		Clients:     types.NewLMap[uint, *types.WsClient](),
-		uploader: manager,
-		levelDB:  levelDB,
+		uploader:    manager,
+		levelDB:     levelDB,
 	}
 }
 
@@ -50,6 +50,7 @@ func (p *ServicePool) InitServices(configs []types.StableDiffusionConfig) {
 	for _, s := range p.services {
 		s.Stop()
 	}
+	p.services = make([]*Service, 0)
 
 	for k, config := range configs {
 		if config.Enabled == false {

api/service/sd/service.go

@@ -60,7 +60,7 @@ func (s *Service) Run() {
 			logger.Errorf("taking task with error: %v", err)
 			continue
 		}
-		logger.Infof("%s handle a new Stable-Diffusion task: %+v", s.name, task)
+
 		// translate prompt
 		if utils.HasChinese(task.Params.Prompt) {
 			content, err := utils.OpenAIRequest(s.db, fmt.Sprintf(service.RewritePromptTemplate, task.Params.Prompt))