#!/usr/bin/env python3
# -*- coding: utf-8 -*-
import base64, json, os, sys, urllib.request

KEY = open(os.path.expanduser("~/.openrouter_key")).read().strip()
MODEL = "google/gemini-3.1-flash-lite-image"

def read_design(path):
    b64 = base64.b64encode(open(path, "rb").read()).decode()
    data_url = f"data:image/jpeg;base64,{b64}"
    body = json.dumps({
        "model": MODEL,
        "messages": [{"role": "user", "content": [
            {"type": "text", "text": (
                "Analisa poster ini. Sebutkan singkat & jelas: judul utama, "
                "sub heading, poin isi, CTA, dan tema. Bahasa Indonesia."
            )},
            {"type": "image_url", "image_url": {"url": data_url}}
        ]}],
        "max_tokens": 200
    }).encode()
    req = urllib.request.Request(
        "https://openrouter.ai/api/v1/chat/completions",
        data=body,
        headers={"Authorization": f"Bearer {KEY}", "Content-Type": "application/json"}
    )
    r = urllib.request.urlopen(req, timeout=90)
    d = json.load(r)
    return d["choices"][0]["message"]["content"].strip()

if __name__ == "__main__":
    path = sys.argv[1]
    print(read_design(path))