feat: stable diffusion image drawing on mobile is ready

This commit is contained in:
RockYang
2024-04-03 18:13:48 +08:00
parent ed286c8894
commit a9adc0c1a2
19 changed files with 1457 additions and 545 deletions

View File

@@ -67,7 +67,7 @@ func (s *Service) Run() {
continue
}
// 如果是 mj-proxy 则自动翻译提示词
// translate prompt
if utils.HasChinese(task.Prompt) {
content, err := utils.OpenAIRequest(s.db, fmt.Sprintf(service.TranslatePromptTemplate, task.Prompt))
if err == nil {
@@ -76,6 +76,15 @@ func (s *Service) Run() {
logger.Warnf("error with translate prompt: %v", err)
}
}
// translate negative prompt
if task.NegPrompt != "" && utils.HasChinese(task.NegPrompt) {
content, err := utils.OpenAIRequest(s.db, fmt.Sprintf(service.TranslatePromptTemplate, task.NegPrompt))
if err == nil {
task.NegPrompt = content
} else {
logger.Warnf("error with translate prompt: %v", err)
}
}
var job model.MidJourneyJob
tx := s.db.Where("id = ?", task.Id).First(&job)