generator.py

from transformers import pipeline

text_generator = pipeline("text-generation", model="gpt2")

def generate_ad(platform, audience, product, tone):
    prompt = f"Write a {tone} ad for {product} targeting {audience} on {platform}."
    result = text_generator(prompt, max_length=80, num_return_sequences=1)
    return result[0]['generated_text']