diff --git a/pkg/service/import/generateTextToImage.go b/pkg/service/import/generateTextToImage.go index 3d146e7..8941922 100644 --- a/pkg/service/import/generateTextToImage.go +++ b/pkg/service/import/generateTextToImage.go @@ -16,7 +16,8 @@ type TextToImageRequest struct { } type TextToImageInput struct { - Prompt string `json:"prompt"` + Prompt string `json:"prompt"` + NegativePrompt string `json:"negative_prompt"` } type TextToImageParameters struct { @@ -54,7 +55,8 @@ func (g *AiGenerator) TextToImage(prompt string, size string, n int) (*TextToIma req := TextToImageRequest{ Model: g.cfg.TextToImageModel, Input: TextToImageInput{ - Prompt: prompt, + Prompt: prompt, + NegativePrompt: "低质量、残缺、人物正脸、多余的手指、乱码字符和文字、比例不良丶场景以国内场景为主", }, Params: TextToImageParameters{ Size: size, diff --git a/pkg/service/import/imageContentProcessor.go b/pkg/service/import/imageContentProcessor.go index 50b4fe5..277b4bd 100644 --- a/pkg/service/import/imageContentProcessor.go +++ b/pkg/service/import/imageContentProcessor.go @@ -585,7 +585,6 @@ func downloadAndUploadToBucket(imageURL string) (string, error) { return "", fmt.Errorf("读取本地文件失败: %v", err) } - // 上传到桶 BOSClient, err := objstorage.NewOSS( os.Getenv(config.ConfigData.Oss.AccessKeyId), os.Getenv(config.ConfigData.Oss.AccessKeySecret), @@ -599,7 +598,7 @@ func downloadAndUploadToBucket(imageURL string) (string, error) { return "", fmt.Errorf("上传文件失败: %v", err) } url := fmt.Sprintf("%s/%s", os.Getenv(config.ConfigData.Oss.CdnHost), fileName) - // 上传到桶 + //上传到桶 //BOSClient, err := objstorage.NewOSS( // config.ConfigData.Oss.AccessKeyId, // config.ConfigData.Oss.AccessKeySecret, @@ -649,10 +648,10 @@ func (p *BatchProcessor) generateImage(req *excelData) (string, error) { if req.Content != "" { prompt += fmt.Sprintf("2:内容:%s\n", req.Content) // 关联内容 } - //if req.Desc != "" { - // prompt += fmt.Sprintf("3:艺人简介:%s", req.Desc) - //} - prompt += "\n请基于标题和内容生成相关图片,要求多张图片不能相似。" + if req.Desc != "" { + prompt += fmt.Sprintf("3:艺人简介:%s,艺人简介的优先级要低,只依据艺人简介的风格", req.Desc) + } + prompt += "\n请基于标题和内容生成单张图片,高质量,高分辨率。" result, err := NewAiGenerator().TextToImage( prompt,