348 lines
14 KiB
Java
348 lines
14 KiB
Java
package cn.van.business.service;
|
||
|
||
import cn.hutool.core.util.StrUtil;
|
||
import cn.van.business.util.ds.SocialMediaLlmClient;
|
||
import lombok.extern.slf4j.Slf4j;
|
||
import org.springframework.beans.factory.annotation.Autowired;
|
||
import org.springframework.data.redis.core.StringRedisTemplate;
|
||
import org.springframework.stereotype.Service;
|
||
|
||
import java.util.ArrayList;
|
||
import java.util.HashMap;
|
||
import java.util.List;
|
||
import java.util.Map;
|
||
|
||
/**
|
||
* 小红书/抖音内容生成服务
|
||
* 提供关键词提取、文案生成等功能
|
||
*
|
||
* @author System
|
||
*/
|
||
@Slf4j
|
||
@Service
|
||
public class SocialMediaService {
|
||
|
||
@Autowired
|
||
private SocialMediaLlmClient socialMediaLlmClient;
|
||
|
||
@Autowired
|
||
private MarketingImageService marketingImageService;
|
||
|
||
@Autowired(required = false)
|
||
private StringRedisTemplate redisTemplate;
|
||
|
||
// Redis Key 前缀
|
||
private static final String REDIS_KEY_PREFIX = "social_media:prompt:";
|
||
|
||
// 默认提示词模板
|
||
private static final String DEFAULT_KEYWORDS_PROMPT =
|
||
"请从以下商品标题中提取3-5个最核心的关键词，这些关键词要能突出商品的核心卖点和特色。\n" +
|
||
"要求：\n" +
|
||
"1. 每个关键词2-4个字\n" +
|
||
"2. 关键词要能吸引小红书/抖音用户\n" +
|
||
"3. 用逗号分隔，只返回关键词，不要其他内容\n" +
|
||
"商品标题：%s";
|
||
|
||
private static final String DEFAULT_CONTENT_PROMPT_XHS =
|
||
"请为小红书平台生成一篇商品推广文案，要求：\n" +
|
||
"1. 风格：真实、种草、有温度\n" +
|
||
"2. 开头：用emoji或感叹句吸引注意\n" +
|
||
"3. 内容：突出商品亮点、使用场景、性价比\n" +
|
||
"4. 结尾：引导行动（如：快冲、闭眼入等）\n" +
|
||
"5. 长度：150-300字\n" +
|
||
"6. 适当使用emoji和换行\n" +
|
||
"\n商品信息：\n" +
|
||
"商品名称：%s\n" +
|
||
"%s" + // 价格信息
|
||
"%s" + // 关键词
|
||
"\n请直接生成文案内容，不要添加其他说明：";
|
||
|
||
private static final String DEFAULT_CONTENT_PROMPT_DOUYIN =
|
||
"请为抖音平台生成一篇商品推广文案，要求：\n" +
|
||
"1. 风格：直接、有冲击力、吸引眼球\n" +
|
||
"2. 开头：用疑问句或对比句抓住注意力\n" +
|
||
"3. 内容：强调价格优势、限时优惠、稀缺性\n" +
|
||
"4. 结尾：制造紧迫感，引导立即行动\n" +
|
||
"5. 长度：100-200字\n" +
|
||
"6. 使用短句，节奏感强\n" +
|
||
"\n商品信息：\n" +
|
||
"商品名称：%s\n" +
|
||
"%s" + // 价格信息
|
||
"%s" + // 关键词
|
||
"\n请直接生成文案内容，不要添加其他说明：";
|
||
|
||
private static final String DEFAULT_CONTENT_PROMPT_BOTH =
|
||
"请生成一篇适合小红书和抖音平台的商品推广文案，要求：\n" +
|
||
"1. 风格：真实、有吸引力\n" +
|
||
"2. 突出商品亮点和价格优势\n" +
|
||
"3. 长度：150-250字\n" +
|
||
"\n商品信息：\n" +
|
||
"商品名称：%s\n" +
|
||
"%s" + // 价格信息
|
||
"%s" + // 关键词
|
||
"\n请直接生成文案内容，不要添加其他说明：";
|
||
|
||
/**
|
||
* 提取商品标题关键词
|
||
*
|
||
* @param productName 商品名称
|
||
* @return 关键词列表
|
||
*/
|
||
public Map<String, Object> extractKeywords(String productName) {
|
||
Map<String, Object> result = new HashMap<>();
|
||
|
||
if (StrUtil.isBlank(productName)) {
|
||
result.put("success", false);
|
||
result.put("error", "商品名称不能为空");
|
||
return result;
|
||
}
|
||
|
||
try {
|
||
// 从 Redis 读取提示词模板，如果没有则使用默认模板
|
||
String promptTemplate = getPromptTemplate("keywords", DEFAULT_KEYWORDS_PROMPT);
|
||
String prompt = String.format(promptTemplate, productName);
|
||
|
||
String response = socialMediaLlmClient.getResponse(prompt);
|
||
|
||
if (StrUtil.isNotBlank(response)) {
|
||
// 解析关键词
|
||
String[] keywords = response.trim()
|
||
.replaceAll("[，,]", ",")
|
||
.split(",");
|
||
|
||
List<String> keywordList = new ArrayList<>();
|
||
for (String keyword : keywords) {
|
||
String cleaned = keyword.trim();
|
||
if (StrUtil.isNotBlank(cleaned) && cleaned.length() <= 6) {
|
||
keywordList.add(cleaned);
|
||
}
|
||
}
|
||
|
||
// 限制数量
|
||
if (keywordList.size() > 5) {
|
||
keywordList = keywordList.subList(0, 5);
|
||
}
|
||
|
||
result.put("success", true);
|
||
result.put("keywords", keywordList);
|
||
result.put("keywordsText", String.join("、", keywordList));
|
||
log.info("提取关键词成功: {} -> {}", productName, keywordList);
|
||
} else {
|
||
throw new Exception("AI返回结果为空");
|
||
}
|
||
|
||
} catch (Exception e) {
|
||
log.error("提取关键词失败", e);
|
||
result.put("success", false);
|
||
result.put("error", "提取关键词失败: " + e.getMessage());
|
||
// 降级方案：简单提取
|
||
result.put("keywords", simpleExtractKeywords(productName));
|
||
result.put("keywordsText", String.join("、", simpleExtractKeywords(productName)));
|
||
}
|
||
|
||
return result;
|
||
}
|
||
|
||
/**
|
||
* 生成小红书/抖音文案
|
||
*
|
||
* @param productName 商品名称
|
||
* @param originalPrice 原价
|
||
* @param finalPrice 到手价
|
||
* @param keywords 关键词（可选）
|
||
* @param style 文案风格：xhs（小红书）、douyin（抖音）、both（通用）
|
||
* @return 生成的文案
|
||
*/
|
||
public Map<String, Object> generateContent(String productName, Double originalPrice,
|
||
Double finalPrice, String keywords, String style) {
|
||
Map<String, Object> result = new HashMap<>();
|
||
|
||
if (StrUtil.isBlank(productName)) {
|
||
result.put("success", false);
|
||
result.put("error", "商品名称不能为空");
|
||
return result;
|
||
}
|
||
|
||
try {
|
||
// 构建价格信息
|
||
StringBuilder priceInfo = new StringBuilder();
|
||
if (originalPrice != null && originalPrice > 0) {
|
||
priceInfo.append("原价：¥").append(String.format("%.0f", originalPrice)).append("\n");
|
||
}
|
||
if (finalPrice != null && finalPrice > 0) {
|
||
priceInfo.append("到手价：¥").append(String.format("%.0f", finalPrice)).append("\n");
|
||
}
|
||
|
||
// 构建关键词信息
|
||
String keywordsInfo = "";
|
||
if (StrUtil.isNotBlank(keywords)) {
|
||
keywordsInfo = "关键词：" + keywords + "\n";
|
||
}
|
||
|
||
// 从 Redis 读取提示词模板，如果没有则使用默认模板
|
||
String promptTemplate;
|
||
if ("xhs".equals(style)) {
|
||
promptTemplate = getPromptTemplate("content:xhs", DEFAULT_CONTENT_PROMPT_XHS);
|
||
} else if ("douyin".equals(style)) {
|
||
promptTemplate = getPromptTemplate("content:douyin", DEFAULT_CONTENT_PROMPT_DOUYIN);
|
||
} else {
|
||
promptTemplate = getPromptTemplate("content:both", DEFAULT_CONTENT_PROMPT_BOTH);
|
||
}
|
||
|
||
String prompt = String.format(promptTemplate, productName, priceInfo.toString(), keywordsInfo);
|
||
|
||
String content = socialMediaLlmClient.getResponse(prompt.toString());
|
||
|
||
if (StrUtil.isNotBlank(content)) {
|
||
result.put("success", true);
|
||
result.put("content", content.trim());
|
||
log.info("生成文案成功: {}", productName);
|
||
} else {
|
||
throw new Exception("AI返回结果为空");
|
||
}
|
||
|
||
} catch (Exception e) {
|
||
log.error("生成文案失败", e);
|
||
result.put("success", false);
|
||
result.put("error", "生成文案失败: " + e.getMessage());
|
||
// 降级方案：生成简单文案
|
||
result.put("content", generateSimpleContent(productName, originalPrice, finalPrice));
|
||
}
|
||
|
||
return result;
|
||
}
|
||
|
||
/**
|
||
* 一键生成完整内容（关键词 + 文案 + 图片）
|
||
*
|
||
* @param productImageUrl 商品主图URL
|
||
* @param productName 商品名称
|
||
* @param originalPrice 原价
|
||
* @param finalPrice 到手价
|
||
* @param style 文案风格
|
||
* @return 完整内容
|
||
*/
|
||
public Map<String, Object> generateCompleteContent(String productImageUrl, String productName,
|
||
Double originalPrice, Double finalPrice, String style) {
|
||
Map<String, Object> result = new HashMap<>();
|
||
|
||
try {
|
||
// 1. 提取关键词
|
||
Map<String, Object> keywordResult = extractKeywords(productName);
|
||
List<String> keywords = (List<String>) keywordResult.get("keywords");
|
||
String keywordsText = (String) keywordResult.get("keywordsText");
|
||
|
||
// 2. 生成文案
|
||
Map<String, Object> contentResult = generateContent(
|
||
productName, originalPrice, finalPrice, keywordsText, style
|
||
);
|
||
String content = (String) contentResult.get("content");
|
||
|
||
// 3. 生成营销图片
|
||
String imageBase64 = null;
|
||
if (StrUtil.isNotBlank(productImageUrl)) {
|
||
try {
|
||
// 使用提取的关键词作为商品名称显示
|
||
String displayName = keywords != null && !keywords.isEmpty()
|
||
? keywords.get(0)
|
||
: productName;
|
||
imageBase64 = marketingImageService.generateMarketingImage(
|
||
productImageUrl, originalPrice, finalPrice, displayName
|
||
);
|
||
} catch (Exception e) {
|
||
log.warn("生成营销图片失败", e);
|
||
}
|
||
}
|
||
|
||
result.put("success", true);
|
||
result.put("keywords", keywords);
|
||
result.put("keywordsText", keywordsText);
|
||
result.put("content", content);
|
||
result.put("imageBase64", imageBase64);
|
||
|
||
} catch (Exception e) {
|
||
log.error("生成完整内容失败", e);
|
||
result.put("success", false);
|
||
result.put("error", "生成完整内容失败: " + e.getMessage());
|
||
}
|
||
|
||
return result;
|
||
}
|
||
|
||
/**
|
||
* 简单提取关键词（降级方案）
|
||
*/
|
||
private List<String> simpleExtractKeywords(String productName) {
|
||
List<String> keywords = new ArrayList<>();
|
||
|
||
// 移除常见规格信息
|
||
String cleaned = productName
|
||
.replaceAll("\\s*XL|L|M|S|XXL\\s*", "")
|
||
.replaceAll("\\s*\\d+/\\d+[A-Z]?\\s*", "")
|
||
.replaceAll("\\s*【.*?】\\s*", "")
|
||
.replaceAll("\\s*\\(.*?\\)\\s*", "");
|
||
|
||
// 提取前几个词
|
||
String[] words = cleaned.split("\\s+");
|
||
for (String word : words) {
|
||
if (word.length() >= 2 && word.length() <= 6 && keywords.size() < 5) {
|
||
keywords.add(word);
|
||
}
|
||
}
|
||
|
||
return keywords;
|
||
}
|
||
|
||
/**
|
||
* 从 Redis 获取提示词模板，如果没有则返回默认模板
|
||
*
|
||
* @param templateKey 模板键名（如：keywords, content:xhs, content:douyin, content:both）
|
||
* @param defaultTemplate 默认模板
|
||
* @return 提示词模板
|
||
*/
|
||
private String getPromptTemplate(String templateKey, String defaultTemplate) {
|
||
if (redisTemplate == null) {
|
||
log.debug("Redis未配置，使用默认模板: {}", templateKey);
|
||
return defaultTemplate;
|
||
}
|
||
|
||
try {
|
||
String redisKey = REDIS_KEY_PREFIX + templateKey;
|
||
String template = redisTemplate.opsForValue().get(redisKey);
|
||
if (StrUtil.isNotBlank(template)) {
|
||
log.debug("从Redis读取模板: {}", templateKey);
|
||
return template;
|
||
} else {
|
||
log.debug("Redis中未找到模板，使用默认模板: {}", templateKey);
|
||
return defaultTemplate;
|
||
}
|
||
} catch (Exception e) {
|
||
log.warn("读取Redis模板失败，使用默认模板: {}", templateKey, e);
|
||
return defaultTemplate;
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 生成简单文案（降级方案）
|
||
*/
|
||
private String generateSimpleContent(String productName, Double originalPrice, Double finalPrice) {
|
||
StringBuilder content = new StringBuilder();
|
||
|
||
content.append("🔥 ").append(productName).append("\n\n");
|
||
|
||
if (originalPrice != null && finalPrice != null && originalPrice > finalPrice) {
|
||
content.append("💰 原价：¥").append(String.format("%.0f", originalPrice)).append("\n");
|
||
content.append("💸 到手价：¥").append(String.format("%.0f", finalPrice)).append("\n");
|
||
double discount = ((originalPrice - finalPrice) / originalPrice) * 100;
|
||
content.append("✨ 立省：¥").append(String.format("%.0f", originalPrice - finalPrice))
|
||
.append("（").append(String.format("%.0f", discount)).append("%）\n\n");
|
||
}
|
||
|
||
content.append("💡 超值好物，不容错过！\n");
|
||
content.append("🎁 限时优惠，先到先得！");
|
||
|
||
return content.toString();
|
||
}
|
||
}
|
||
|