Update ads from images

ai_tasks/headlines_for_images.py

@@ -6,11 +6,11 @@ from utils.io import print_system
 
 
 PROMPT = """
-Your goal is to find a set of the best images that can be used as ads for a website.
-The ads have to be in certain dimensions. The images can be edited or cut.
-So you must consider which images would be better suited to fit those dimensions after editing.
+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 each ad.
+You will create a headline for the ad.
 
 Summary of the website:
 {summary}
@@ -18,26 +18,25 @@ Summary of the website:
 Image urls, with labels and dimensions:
 {images}
 
-Dimensions for the ads: {dimensions}
+Dimensions for the ad: {dimensions}.
 
-Use the folliowing format. Each target dimension must be represented.
+Use the folliowing format.
 
 Why the image was chosen:
 Url:
 Headline:
-Image dimensions:
-Target dimensions: It can be more than one
+Original dimensions:
 """
 
 
 def get_headlines_for_images(
-    summary: str, dimensions: List[str], image_labels: List[Dict]
+    summary: str, dimensions: str, image_labels: List[Dict]
 ) -> str:
-    print_system("Generating headlines for images...")
+    print_system("Generating ad from images...")
     instructions = PROMPT.format(
         summary=summary,
         images=json.dumps(image_labels, indent=2),
         dimensions=dimensions,
     )
     messages = [{"role": "user", "content": instructions}]
-    return llm.next(messages, temperature=0)
+    return llm.next(messages)

ai_tasks/text_summary.py

@@ -14,4 +14,4 @@ def summarize_text(text: str) -> str:
     print_system("Summarizing text...")
     instructions = PROMPT.format(text=text)
     messages = [{"role": "user", "content": instructions}]
-    return llm.next(messages, temperature=0.4)
+    return llm.next(messages)

control_flow/main.py

@@ -1,4 +1,5 @@
 from typing import Dict
+import json
 
 from ai import image
 from ai_tasks.headlines_for_images import get_headlines_for_images
@@ -10,33 +11,35 @@ from code_tasks.text_in_url import get_text_from_url
 from utils.io import print_assistant, print_system, user_input
 
 
-DIMENSIONS = [
-    "300x50",
-    "300x250",
-    "300x600",
-    "728x90",
-    "160x600",
-]
+"""
+"300x50",
+"300x250",
+"300x600",
+"728x90",
+"160x600",
+"""
 
 
 def run():
     url = user_input("URL: ")
+    dimensions = user_input("Dimensions: ")
 
     # Code tasks (most code was generated by ChatGPT-4)
     print_system("Getting URL data...")
     text = get_text_from_url(url)
     images = get_images_from_url(url)
-    image_info = run_parallel_jobs(get_image_info, images, max=10)
+    image_info = run_parallel_jobs(get_image_info, images, max=20)
+    print_system(json.dumps(image_info, indent=2))
 
     # AI tasks
     summary = summarize_text(text)
     print_assistant(summary)
-    headlines = get_headlines_for_images(summary, DIMENSIONS, image_info)
+    headlines = get_headlines_for_images(summary, dimensions, image_info)
     print_assistant(headlines)
 
-    headlines_prompts = generate_headlines(summary, DIMENSIONS)
-    print_system("Generating AI images...")
-    run_parallel_jobs(gen_image, headlines_prompts)
+    # headlines_prompts = generate_headlines(summary, DIMENSIONS)
+    # print_system("Generating AI images...")
+    # run_parallel_jobs(gen_image, headlines_prompts)
 
 
 def gen_image(input: Dict[str, str]) -> None: