| import json | |
| import re | |
| from typing import Dict, List | |
| from ai import llm | |
| from utils.io import print_system | |
| PROMPT = """ | |
| Your goal is to find the best image that can be used as an ad for a website. | |
| The ad must have certain dimensions but the image can be edited or cut. | |
| So you must consider an image that is suited to fit those dimensions after editing. | |
| You should consider the content of the website. The content of the images has been labeled. | |
| You will create a headline for the ad. | |
| Summary of the website: | |
| {summary} | |
| Image urls, with labels and dimensions: | |
| {image_infos} | |
| Dimensions for the ad: {dimensions}. | |
| Use the following format. | |
| ``` | |
| {{ | |
| "reason": Why the image was chosen | |
| "image_url": | |
| "headline"": | |
| "image_dimensions": The dimensions of the original image | |
| }} | |
| ``` | |
| """ | |
| def get_headline_for_image( | |
| summary: str, dimensions: str, image_infos: List[Dict] | |
| ) -> str: | |
| return _get_headline_for_image( | |
| PROMPT, summary=summary, dimensions=dimensions, image_infos=image_infos | |
| ) | |
| def _get_headline_for_image(prompt: str, **kwargs) -> str: | |
| print_system("Generating ad from images...") | |
| instructions = prompt.format(**kwargs) | |
| messages = [{"role": "user", "content": instructions}] | |
| return llm.next(messages) | |