ai_tasks/best_headlines.py

from ai import llm
from utils.io import print_system


PROMPT = """
Here is the text from a website.
Extract or create the 10 best headlines for an ad about the website.

{text}
"""


def get_headlines(text) -> str:
    print_system("Generating headlines...")
    instructions = PROMPT.format(text=text)
    messages = [{"role": "user", "content": instructions}]
    return llm.next(messages, temperature=0)