modules/creatives.py

from typing import List, Dict

def generate_meta_ads(product: str, audience: str, hook_type: str = "Benefit") -> str:
    """Generates primary text, headline, and description for Meta."""
    hooks = {
        "Benefit": f"Stop struggling with [Pain Point]. {product} is designed for {audience} to achieve [Result] in half the time.",
        "Fear": f"Are you losing money by ignoring [Problem]? {product} helps {audience} secure their future.",
        "Social Proof": f"Join 10,000+ {audience} who have already transformed their workflow with {product}."
    }
    
    primary_text = hooks.get(hook_type, hooks["Benefit"])
    
    return f"""
# Meta Ad Creative
**Primary Text:** {primary_text}
**Headline:** {product} | The #1 Choice for {audience}
**Description:** Get started today and save 20% on your first month.
**CTA:** Learn More
"""

def generate_google_search_ads(product: str, keywords: List[str]) -> str:
    """Generates RSA style headlines and descriptions."""
    headlines = [
        f"Buy {product} Online",
        f"Best {product} for {keywords[0] if keywords else 'Business'}",
        "Official Site | Shop Now",
        "Save 20% with Code START",
        f"High Quality {product}",
        "Free Shipping & Returns",
        f"Trusted by {keywords[0] if keywords else 'Experts'}",
        "Limited Time Offer",
        "Top Rated Solution",
        "Affordable Pricing"
    ]
    
    descriptions = [
        f"Discover why {product} is the preferred choice for professionals. High quality, great value.",
        f"Solve [Pain Point] today with {product}. Rated 5 stars by thousands of customers.",
        "Fast shipping and easy returns on all orders. Shop the latest collection now.",
        "Customizable options available to fit your specific needs. Get a quote today."
    ]
    
    return f"""
# Google Search Ad (RSA)
## Headlines (Top 10)
{chr(10).join([f"- {h}" for h in headlines])}

## Descriptions (Top 4)
{chr(10).join([f"- {d}" for d in descriptions])}
"""

def generate_tiktok_script(product: str, trend_style: str = "Voiceover") -> str:
    """Provides a 15-30s video script."""
    return f"""
# TikTok Script: {product} ({trend_style})

00:00 - 00:03 [Visual: Fast cut of problem] Hook: "POV: You're still doing [Pain Point] the old way."
00:03 - 00:10 [Visual: Transition to {product}] "Then I found {product}. It literally changed everything."
00:10 - 00:20 [Visual: Screen recording/Demo] "Check this out. I just click here, and [Core Benefit] happens automatically."
00:20 - 00:30 [Visual: Happy face/Result] "Stop wasting time. Link in bio to try it for free!"
"""

def generate_twitter_x_ads(product: str, tone: str = "Punchy") -> str:
    """Creates short-form ad copy."""
    return f"""
# Twitter/X Ad Copy
**Option 1:** {product} is the cheat code for {product}. Stop overthinking it. [Link]
**Option 2:** {product} > Everything else. 
**Thread Hook:** Why most {product} fail (and how to be in the top 1% using {product}): 🧵
"""

def generate_creative_brief(product: str, objective: str) -> str:
    """Creates a formal brief."""
    return f"""
# Creative Brief: {product}
**Objective:** {objective}
**Target Audience:** [Detailed Persona]
**Key Message:** [One core takeaway]
**Visual Style:** [Clean/Vibrant/Corporate]
**Deliverables:** 3x Meta Squares, 1x TikTok 9:16 Video, 5x Search RSA
**Success Metric:** [e.g., Click-Through Rate > 1.5%]
"""