File size: 1,285 Bytes
1bc7525
 
5760606
1bc7525
 
 
5760606
1bc7525
 
bbf59eb
 
5760606
0eb1950
 
1bc7525
 
 
 
 
 
 
 
 
0eb1950
1bc7525
 
 
0eb1950
bbf59eb
1bc7525
 
bbf59eb
1bc7525
 
 
4fb1dcc
1bc7525
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
import json

from ai import image

# from ai_tasks.best_headlines import get_headlines
from ai_tasks.headlines_for_images import get_headlines
from ai_tasks.image_prompt import generate_prompt
from ai_tasks.text_summary import summarize_text
from code_tasks.custom import get_info_for_images
from code_tasks.images_in_url import get_images_from_url
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",
]


def run():
    # url = user_input("URL: ")
    url = "https://www.beachterracemc.com/"
    print_system("Getting URL data...")
    text = get_text_from_url(url)
    images = get_images_from_url(url)
    image_info = get_info_for_images(images)
    print_system(json.dumps(image_info, indent=2))

    summary = summarize_text(text)
    print_assistant(summary)
    headlines = get_headlines(summary, DIMENSIONS, image_info)
    print_assistant(headlines)

    # headlines = get_headlines(text)
    # print_assistant(headlines)
    # prompt = generate_prompt(text)
    # print_assistant(prompt)
    # print_system("Generating images...")
    # image_urls = image.urls(prompt, n=4)
    # print_assistant("\n\n".join(image_urls))