import llm
import time
from pathlib import Path

MODELS = [
    "gpt-4o",
    "claude-3-5-sonnet-latest",
    "gemini-2.0-flash-exp",
]

SUBJECT = "an extremely detailed, beautiful sunset"

PROMPT = f"Generate pure HTML/CSS art of {SUBJECT}. No talk or code fences. Code only."


now = str(int(time.time()))
out_dir = Path("out") / now
out_dir.mkdir(parents=True, exist_ok=True)
prompt_file = out_dir / "prompt.txt"
prompt_file.write_text(PROMPT)

for m in MODELS:
    model = llm.get_model(m)
    response = model.prompt(PROMPT)
    out_file = out_dir / f"{m}.html"
    out_file.write_text(response.text())