boladeE
93 lines
4.0 KiB
Python
93 lines
4.0 KiB
Python
from typing import List, Dict, Any
|
|
import requests
|
|
import json
|
|
from config import settings
|
|
from brand_style import BrandStyleManager
|
|
from embeddings import CohereEmbeddings
|
|
from vector_store import VectorStore
|
|
|
|
class MarketingCopywriter:
|
|
def __init__(self, brand_style_manager: BrandStyleManager):
|
|
self.settings = settings
|
|
self.api_key = self.settings.DEEPSEEK_API_KEY
|
|
self.api_url = "https://api.deepseek.com/v1/chat/completions"
|
|
self.brand_style_manager = brand_style_manager
|
|
|
|
def _build_prompt(self, prompt: str, context: List[Dict], tone: str,
|
|
brand_voice: Dict[str, Any], sample_campaigns: List[Dict[str, Any]]) -> str:
|
|
"""Build a prompt for the LLM using context and parameters."""
|
|
# Format context from book excerpts
|
|
context_text = "\n".join([f"Reference {i+1}: {ctx['text']}" for i, ctx in enumerate(context)])
|
|
|
|
# Format brand voice guidelines
|
|
brand_voice_text = json.dumps(brand_voice, indent=2)
|
|
|
|
# Format sample campaigns
|
|
sample_campaigns_text = ""
|
|
for i, campaign in enumerate(sample_campaigns):
|
|
sample_campaigns_text += f"\nExample Campaign {i+1}:\n"
|
|
sample_campaigns_text += f"Title: {campaign.get('title', '')}\n"
|
|
sample_campaigns_text += f"Subject: {campaign.get('subject', '')}\n"
|
|
sample_campaigns_text += f"Content:\n{campaign.get('content', '')}\n"
|
|
|
|
return f"""You are a professional marketing copywriter for {self.settings.BRAND_VOICE}.
|
|
Your task is to create content that matches the following request: {prompt}
|
|
|
|
BRAND VOICE GUIDELINES:
|
|
{brand_voice_text}
|
|
|
|
SAMPLE CAMPAIGNS:
|
|
{sample_campaigns_text}
|
|
|
|
RELEVANT BOOK EXCERPTS:
|
|
{context_text}
|
|
|
|
Guidelines:
|
|
1. Maintain a {tone} tone throughout
|
|
2. Follow {self.settings.BRAND_VOICE}'s brand voice guidelines
|
|
3. Be persuasive and engaging
|
|
4. Include a clear call-to-action
|
|
5. Keep the content concise and impactful
|
|
6. Ensure there is no special formatting in the output just plain text.
|
|
7. Make no reference to Adriana James.
|
|
|
|
Generate the marketing copy:
|
|
"""
|
|
|
|
def generate_copy(self, prompt: str, context: List[Dict], tone: str,
|
|
brand_voice: Dict[str, Any], sample_campaigns) -> str:
|
|
"""Generate marketing copy using DeepSeek."""
|
|
full_prompt = self._build_prompt(prompt, context, tone, brand_voice, sample_campaigns)
|
|
|
|
headers = {
|
|
"Authorization": f"Bearer {self.api_key}",
|
|
"Content-Type": "application/json"
|
|
}
|
|
|
|
data = {
|
|
"model": self.settings.DEFAULT_MODEL,
|
|
"messages": [
|
|
{"role": "system", "content": "You are a professional marketing copywriter."},
|
|
{"role": "user", "content": full_prompt}
|
|
],
|
|
"temperature": 0.7,
|
|
"max_tokens": 1000
|
|
}
|
|
|
|
response = requests.post(self.api_url, headers=headers, json=data)
|
|
response.raise_for_status()
|
|
|
|
return response.json()["choices"][0]["message"]["content"].strip()
|
|
|
|
def generate_marketing_copy(prompt: str) -> str:
|
|
"""Helper function to generate marketing copy."""
|
|
embeddings = CohereEmbeddings()
|
|
vector_store = VectorStore()
|
|
brand_style_manager = BrandStyleManager(embeddings, vector_store)
|
|
copywriter = MarketingCopywriter(brand_style_manager)
|
|
context = brand_style_manager.get_relevant_context(prompt)
|
|
tone = "professional and empathetic"
|
|
brand_voice = brand_style_manager.get_brand_voice()
|
|
sample_campaigns = brand_style_manager.get_sample_campaigns()
|
|
print(sample_campaigns)
|
|
return copywriter.generate_copy(prompt, context, tone, brand_voice, sample_campaigns) |