438 lines
18 KiB
Python
438 lines
18 KiB
Python
|
#!/usr/bin/env python3
|
|||
|
|
"""
|
|||
|
|
AI文案生成服务集成
|
|||
|
|
使用AI大模型为照片生成创意文案
|
|||
|
|
支持多种文案风格和用途
|
|||
|
|
支持DeepSeek和DashScope两种大模型
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
import os
|
|||
|
|
import json
|
|||
|
|
import requests
|
|||
|
|
from dotenv import load_dotenv
|
|||
|
|
|
|||
|
|
# 加载环境变量
|
|||
|
|
load_dotenv()
|
|||
|
|
|
|||
|
|
class AICopywriter:
|
|||
|
|
"""AI文案生成服务类"""
|
|||
|
|
|
|||
|
|
def __init__(self, provider='deepseek'):
|
|||
|
|
"""初始化AI文案生成客户端"""
|
|||
|
|
self.provider = provider
|
|||
|
|
|
|||
|
|
if provider == 'deepseek':
|
|||
|
|
self.api_key = os.getenv('DEEPSEEK_API_KEY')
|
|||
|
|
if not self.api_key:
|
|||
|
|
raise Exception("DeepSeek API密钥未配置，请在.env文件中设置DEEPSEEK_API_KEY")
|
|||
|
|
self.base_url = "https://api.deepseek.com/v1/chat/completions"
|
|||
|
|
elif provider == 'dashscope':
|
|||
|
|
self.api_key = os.getenv('DASHSCOPE_API_KEY')
|
|||
|
|
if not self.api_key:
|
|||
|
|
raise Exception("DashScope API密钥未配置，请在.env文件中设置DASHSCOPE_API_KEY")
|
|||
|
|
else:
|
|||
|
|
raise Exception(f"不支持的AI提供商: {provider}")
|
|||
|
|
|
|||
|
|
def generate_photo_caption(self, image_description, style='creative', length='medium'):
|
|||
|
|
"""为照片生成文案"""
|
|||
|
|
try:
|
|||
|
|
if self.provider == 'deepseek':
|
|||
|
|
return self._generate_with_deepseek(image_description, style, length)
|
|||
|
|
elif self.provider == 'dashscope':
|
|||
|
|
return self._generate_with_dashscope(image_description, style, length)
|
|||
|
|
else:
|
|||
|
|
raise Exception(f"不支持的AI提供商: {self.provider}")
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
raise Exception(f"AI文案生成失败: {str(e)}")
|
|||
|
|
|
|||
|
|
def _generate_with_deepseek(self, image_description, style, length):
|
|||
|
|
"""使用DeepSeek生成文案"""
|
|||
|
|
try:
|
|||
|
|
prompt = self._build_prompt(image_description, style, length)
|
|||
|
|
|
|||
|
|
headers = {
|
|||
|
|
'Authorization': f'Bearer {self.api_key}',
|
|||
|
|
'Content-Type': 'application/json'
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
data = {
|
|||
|
|
'model': 'deepseek-chat',
|
|||
|
|
'messages': [
|
|||
|
|
{
|
|||
|
|
'role': 'system',
|
|||
|
|
'content': '你是一个专业的创意文案创作助手，擅长为照片生成各种风格的创意文案。你具有丰富的文学素养和营销知识，能够根据照片内容创作出富有创意和感染力的文案。'
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
'role': 'user',
|
|||
|
|
'content': prompt
|
|||
|
|
}
|
|||
|
|
],
|
|||
|
|
'max_tokens': 500,
|
|||
|
|
'temperature': 0.8,
|
|||
|
|
'top_p': 0.9
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
response = requests.post(self.base_url, headers=headers, json=data)
|
|||
|
|
result = response.json()
|
|||
|
|
|
|||
|
|
if 'choices' in result and len(result['choices']) > 0:
|
|||
|
|
caption = result['choices'][0]['message']['content'].strip()
|
|||
|
|
# 清理可能的格式标记
|
|||
|
|
caption = caption.replace('"', '').replace('\n', ' ').strip()
|
|||
|
|
return caption
|
|||
|
|
else:
|
|||
|
|
# 如果API调用失败，使用备用文案生成
|
|||
|
|
return self._generate_fallback_caption(image_description, style, length)
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
# API调用失败时使用备用方案
|
|||
|
|
return self._generate_fallback_caption(image_description, style, length)
|
|||
|
|
|
|||
|
|
def _generate_with_dashscope(self, image_description, style, length):
|
|||
|
|
"""使用DashScope生成文案"""
|
|||
|
|
try:
|
|||
|
|
url = "https://dashscope.aliyuncs.com/api/v1/services/aigc/text-generation/generation"
|
|||
|
|
|
|||
|
|
headers = {
|
|||
|
|
'Authorization': f'Bearer {self.api_key}',
|
|||
|
|
'Content-Type': 'application/json'
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
# 根据风格和长度构建提示词
|
|||
|
|
prompt = self._build_prompt(image_description, style, length)
|
|||
|
|
|
|||
|
|
data = {
|
|||
|
|
'model': 'qwen-turbo',
|
|||
|
|
'input': {
|
|||
|
|
'messages': [
|
|||
|
|
{
|
|||
|
|
'role': 'system',
|
|||
|
|
'content': '你是一个专业的文案创作助手，擅长为照片生成各种风格的创意文案。'
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
'role': 'user',
|
|||
|
|
'content': prompt
|
|||
|
|
}
|
|||
|
|
]
|
|||
|
|
},
|
|||
|
|
'parameters': {
|
|||
|
|
'max_tokens': 500,
|
|||
|
|
'temperature': 0.8
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
response = requests.post(url, headers=headers, json=data)
|
|||
|
|
result = response.json()
|
|||
|
|
|
|||
|
|
if 'output' in result and 'text' in result['output']:
|
|||
|
|
return result['output']['text']
|
|||
|
|
else:
|
|||
|
|
# 如果API调用失败，使用备用文案生成
|
|||
|
|
return self._generate_fallback_caption(image_description, style, length)
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
# API调用失败时使用备用方案
|
|||
|
|
return self._generate_fallback_caption(image_description, style, length)
|
|||
|
|
|
|||
|
|
def _build_prompt(self, image_description, style, length):
|
|||
|
|
"""构建AI提示词"""
|
|||
|
|
|
|||
|
|
style_descriptions = {
|
|||
|
|
'creative': '创意文艺风格，富有诗意和想象力',
|
|||
|
|
'professional': '专业正式风格，简洁明了',
|
|||
|
|
'social': '社交媒体风格，活泼有趣，适合朋友圈',
|
|||
|
|
'marketing': '营销推广风格，吸引眼球，促进转化',
|
|||
|
|
'simple': '简单描述风格，直接明了',
|
|||
|
|
'emotional': '情感表达风格，温暖感人'
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
length_descriptions = {
|
|||
|
|
'short': '10-20字，简洁精炼',
|
|||
|
|
'medium': '30-50字，适中长度',
|
|||
|
|
'long': '80-120字，详细描述'
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
prompt = f"""
|
|||
|
|
请为以下照片内容生成{style_descriptions.get(style, '创意')}的文案，要求{length_descriptions.get(length, '适中长度')}。
|
|||
|
|
|
|||
|
|
照片内容描述：{image_description}
|
|||
|
|
|
|||
|
|
文案要求：
|
|||
|
|
1. 符合{style}风格
|
|||
|
|
2. 长度{length}
|
|||
|
|
3. 有创意，吸引人
|
|||
|
|
4. 适合社交媒体分享
|
|||
|
|
|
|||
|
|
请直接输出文案内容，不要添加其他说明。
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
return prompt.strip()
|
|||
|
|
|
|||
|
|
def _generate_fallback_caption(self, image_description, style, length):
|
|||
|
|
"""备用文案生成（当AI服务不可用时）"""
|
|||
|
|
|
|||
|
|
# 基于照片描述的简单文案生成
|
|||
|
|
keywords = image_description.lower().split()
|
|||
|
|
|
|||
|
|
# 提取关键信息
|
|||
|
|
objects = []
|
|||
|
|
scenes = []
|
|||
|
|
|
|||
|
|
# 简单的关键词分类（实际应用中可以使用更复杂的NLP处理）
|
|||
|
|
object_keywords = ['人', '建筑', '天空', '树', '花', '动物', '车', '食物', '水', '山']
|
|||
|
|
scene_keywords = ['户外', '室内', '自然', '城市', '夜景', '日出', '日落', '海滩', '森林']
|
|||
|
|
|
|||
|
|
for word in keywords:
|
|||
|
|
if any(obj in word for obj in object_keywords):
|
|||
|
|
objects.append(word)
|
|||
|
|
if any(scene in word for scene in scene_keywords):
|
|||
|
|
scenes.append(word)
|
|||
|
|
|
|||
|
|
# 根据风格生成文案
|
|||
|
|
if style == 'creative':
|
|||
|
|
if scenes:
|
|||
|
|
caption = f"在{scenes[0]}的怀抱中，时光静静流淌"
|
|||
|
|
elif objects:
|
|||
|
|
caption = f"{objects[0]}的美丽瞬间，定格永恒"
|
|||
|
|
else:
|
|||
|
|
caption = "捕捉生活中的美好，让每一刻都值得珍藏"
|
|||
|
|
|
|||
|
|
elif style == 'social':
|
|||
|
|
if objects:
|
|||
|
|
caption = f"今天遇到的{objects[0]}太可爱了！分享给大家～"
|
|||
|
|
else:
|
|||
|
|
caption = "分享一张美照，希望大家喜欢！"
|
|||
|
|
|
|||
|
|
elif style == 'professional':
|
|||
|
|
if scenes and objects:
|
|||
|
|
caption = f"专业拍摄：{scenes[0]}场景中的{objects[0]}特写"
|
|||
|
|
else:
|
|||
|
|
caption = "专业摄影作品展示"
|
|||
|
|
|
|||
|
|
elif style == 'marketing':
|
|||
|
|
if objects:
|
|||
|
|
caption = f"惊艳！这个{objects[0]}你一定要看看！"
|
|||
|
|
else:
|
|||
|
|
caption = "不容错过的精彩瞬间，点击了解更多！"
|
|||
|
|
|
|||
|
|
else: # simple or emotional
|
|||
|
|
if objects:
|
|||
|
|
caption = f"美丽的{objects[0]}照片"
|
|||
|
|
else:
|
|||
|
|
caption = "一张值得分享的照片"
|
|||
|
|
|
|||
|
|
# 根据长度调整
|
|||
|
|
if length == 'long' and len(caption) < 50:
|
|||
|
|
caption += "。这张照片记录了珍贵的瞬间，展现了生活的美好，值得细细品味和珍藏。"
|
|||
|
|
elif length == 'short' and len(caption) > 20:
|
|||
|
|
# 简化长文案
|
|||
|
|
caption = caption[:20] + "..."
|
|||
|
|
|
|||
|
|
return caption
|
|||
|
|
|
|||
|
|
def generate_multiple_captions(self, image_description, count=3, style='creative'):
|
|||
|
|
"""生成多个文案选项"""
|
|||
|
|
try:
|
|||
|
|
if self.provider == 'deepseek':
|
|||
|
|
return self._generate_multiple_with_deepseek(image_description, count, style)
|
|||
|
|
elif self.provider == 'dashscope':
|
|||
|
|
return self._generate_multiple_with_dashscope(image_description, count, style)
|
|||
|
|
else:
|
|||
|
|
raise Exception(f"不支持的AI提供商: {self.provider}")
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
raise Exception(f"生成多个文案失败: {str(e)}")
|
|||
|
|
|
|||
|
|
def _generate_multiple_with_deepseek(self, image_description, count=3, style='creative'):
|
|||
|
|
"""使用DeepSeek生成多个文案选项"""
|
|||
|
|
try:
|
|||
|
|
captions = []
|
|||
|
|
|
|||
|
|
# 使用不同的提示词变体生成多个文案
|
|||
|
|
prompt_variants = [
|
|||
|
|
f"请为'{image_description}'照片创作一个{style}风格的文案，要求新颖独特",
|
|||
|
|
f"基于照片内容'{image_description}'，写一个{style}风格的创意文案",
|
|||
|
|
f"为这张'{image_description}'的照片设计一个{style}风格的吸引人文案"
|
|||
|
|
]
|
|||
|
|
|
|||
|
|
for i in range(min(count, len(prompt_variants))):
|
|||
|
|
prompt = prompt_variants[i]
|
|||
|
|
|
|||
|
|
headers = {
|
|||
|
|
'Authorization': f'Bearer {self.api_key}',
|
|||
|
|
'Content-Type': 'application/json'
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
data = {
|
|||
|
|
'model': 'deepseek-chat',
|
|||
|
|
'messages': [
|
|||
|
|
{
|
|||
|
|
'role': 'system',
|
|||
|
|
'content': '你是专业的创意文案专家，擅长为照片创作多种风格的文案。'
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
'role': 'user',
|
|||
|
|
'content': prompt
|
|||
|
|
}
|
|||
|
|
],
|
|||
|
|
'max_tokens': 200,
|
|||
|
|
'temperature': 0.9, # 提高温度增加多样性
|
|||
|
|
'top_p': 0.95
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
response = requests.post(self.base_url, headers=headers, json=data)
|
|||
|
|
result = response.json()
|
|||
|
|
|
|||
|
|
if 'choices' in result and len(result['choices']) > 0:
|
|||
|
|
caption = result['choices'][0]['message']['content'].strip()
|
|||
|
|
caption = caption.replace('"', '').replace('\n', ' ').strip()
|
|||
|
|
|
|||
|
|
captions.append({
|
|||
|
|
'option': i + 1,
|
|||
|
|
'caption': caption,
|
|||
|
|
'style': style,
|
|||
|
|
'char_count': len(caption)
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
return captions
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
raise Exception(f"DeepSeek多文案生成失败: {str(e)}")
|
|||
|
|
|
|||
|
|
def _generate_multiple_with_dashscope(self, image_description, count=3, style='creative'):
|
|||
|
|
"""使用DashScope生成多个文案选项"""
|
|||
|
|
try:
|
|||
|
|
captions = []
|
|||
|
|
|
|||
|
|
# 尝试使用不同的长度和微调风格
|
|||
|
|
lengths = ['short', 'medium', 'long']
|
|||
|
|
|
|||
|
|
for i in range(min(count, len(lengths))):
|
|||
|
|
caption = self.generate_photo_caption(image_description, style, lengths[i])
|
|||
|
|
captions.append({
|
|||
|
|
'option': i + 1,
|
|||
|
|
'caption': caption,
|
|||
|
|
'length': lengths[i],
|
|||
|
|
'char_count': len(caption)
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
# 如果数量不足，使用不同风格补充
|
|||
|
|
if len(captions) < count:
|
|||
|
|
additional_styles = ['social', 'professional', 'emotional']
|
|||
|
|
for i, add_style in enumerate(additional_styles):
|
|||
|
|
if len(captions) >= count:
|
|||
|
|
break
|
|||
|
|
caption = self.generate_photo_caption(image_description, add_style, 'medium')
|
|||
|
|
captions.append({
|
|||
|
|
'option': len(captions) + 1,
|
|||
|
|
'caption': caption,
|
|||
|
|
'style': add_style,
|
|||
|
|
'char_count': len(caption)
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
return captions
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
raise Exception(f"DashScope多文案生成失败: {str(e)}")
|
|||
|
|
|
|||
|
|
def analyze_photo_suitability(self, image_description):
|
|||
|
|
"""分析照片适合的文案风格"""
|
|||
|
|
try:
|
|||
|
|
# 简单的风格适合性分析
|
|||
|
|
keywords = image_description.lower()
|
|||
|
|
|
|||
|
|
suitability = {
|
|||
|
|
'creative': 0,
|
|||
|
|
'professional': 0,
|
|||
|
|
'social': 0,
|
|||
|
|
'marketing': 0,
|
|||
|
|
'emotional': 0
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
# 关键词匹配（实际应用中可以使用更复杂的NLP分析）
|
|||
|
|
creative_words = ['美丽', '艺术', '创意', '独特', '梦幻']
|
|||
|
|
professional_words = ['专业', '商业', '产品', '展示', '特写']
|
|||
|
|
social_words = ['朋友', '聚会', '日常', '分享', '生活']
|
|||
|
|
marketing_words = ['促销', '优惠', '新品', '限时', '推荐']
|
|||
|
|
emotional_words = ['情感', '感动', '回忆', '温暖', '幸福']
|
|||
|
|
|
|||
|
|
for word in creative_words:
|
|||
|
|
if word in keywords:
|
|||
|
|
suitability['creative'] += 1
|
|||
|
|
|
|||
|
|
for word in professional_words:
|
|||
|
|
if word in keywords:
|
|||
|
|
suitability['professional'] += 1
|
|||
|
|
|
|||
|
|
for word in social_words:
|
|||
|
|
if word in keywords:
|
|||
|
|
suitability['social'] += 1
|
|||
|
|
|
|||
|
|
for word in marketing_words:
|
|||
|
|
if word in keywords:
|
|||
|
|
suitability['marketing'] += 1
|
|||
|
|
|
|||
|
|
for word in emotional_words:
|
|||
|
|
if word in keywords:
|
|||
|
|
suitability['emotional'] += 1
|
|||
|
|
|
|||
|
|
# 排序并返回推荐
|
|||
|
|
recommended = sorted(suitability.items(), key=lambda x: x[1], reverse=True)
|
|||
|
|
|
|||
|
|
return {
|
|||
|
|
'suitability_scores': suitability,
|
|||
|
|
'recommended_styles': [style for style, score in recommended if score > 0],
|
|||
|
|
'most_suitable': recommended[0][0] if recommended[0][1] > 0 else 'creative'
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
raise Exception(f"照片适合性分析失败: {str(e)}")
|
|||
|
|
|
|||
|
|
def generate_photo_caption(image_description, style='creative', length='medium', provider='dashscope'):
|
|||
|
|
"""为照片生成文案"""
|
|||
|
|
try:
|
|||
|
|
copywriter = AICopywriter(provider)
|
|||
|
|
return copywriter.generate_photo_caption(image_description, style, length)
|
|||
|
|
except Exception as e:
|
|||
|
|
raise Exception(f"照片文案生成失败: {str(e)}")
|
|||
|
|
|
|||
|
|
def generate_multiple_captions(image_description, count=3, style='creative', provider='dashscope'):
|
|||
|
|
"""生成多个文案选项"""
|
|||
|
|
try:
|
|||
|
|
copywriter = AICopywriter(provider)
|
|||
|
|
return copywriter.generate_multiple_captions(image_description, count, style)
|
|||
|
|
except Exception as e:
|
|||
|
|
raise Exception(f"多文案生成失败: {str(e)}")
|
|||
|
|
|
|||
|
|
def analyze_photo_suitability(image_description, provider='dashscope'):
|
|||
|
|
"""分析照片适合的文案风格"""
|
|||
|
|
try:
|
|||
|
|
copywriter = AICopywriter(provider)
|
|||
|
|
return copywriter.analyze_photo_suitability(image_description)
|
|||
|
|
except Exception as e:
|
|||
|
|
raise Exception(f"照片适合性分析失败: {str(e)}")
|
|||
|
|
|
|||
|
|
def check_copywriter_config(provider='deepseek'):
|
|||
|
|
"""检查AI文案生成配置是否完整"""
|
|||
|
|
try:
|
|||
|
|
if provider == 'deepseek':
|
|||
|
|
api_key = os.getenv('DEEPSEEK_API_KEY')
|
|||
|
|
if not api_key:
|
|||
|
|
return False, "DeepSeek API密钥未配置"
|
|||
|
|
|
|||
|
|
# 测试连接
|
|||
|
|
copywriter = AICopywriter(provider)
|
|||
|
|
return True, "AI文案生成配置正确（DeepSeek大模型）"
|
|||
|
|
elif provider == 'dashscope':
|
|||
|
|
api_key = os.getenv('DASHSCOPE_API_KEY')
|
|||
|
|
if not api_key:
|
|||
|
|
return False, "DashScope API密钥未配置"
|
|||
|
|
|
|||
|
|
# 测试连接
|
|||
|
|
copywriter = AICopywriter(provider)
|
|||
|
|
return True, "AI文案生成配置正确（DashScope）"
|
|||
|
|
else:
|
|||
|
|
return False, f"不支持的AI提供商: {provider}"
|
|||
|
|
except Exception as e:
|
|||
|
|
return False, f"AI文案生成配置错误: {str(e)}"
|