import json, time, urllib.request, sys
URL = "http://localhost:11434/api/chat"
MODEL = sys.argv[1] if len(sys.argv) > 1 else "hymt2-1.8b"
FRASES = [
    "Bom dia a todos, vamos começar a live de hoje.",
    "A gente vai falar sobre inteligência artificial e soberania de dados.",
    "Se vocês estiverem me ouvindo, mandem um oi no chat.",
    "O projeto roda no meu computador, sem depender de nenhuma empresa grande.",
    "Isso não funcionou da outra vez, então eu resolvi criar o meu próprio sistema.",
    "Pessoal, o link da pesquisa está na descrição, não esqueçam de responder.",
]
def traduz(t):
    body = {"model": MODEL, "stream": False, "keep_alive": "30m",
            "messages": [{"role": "user", "content":
              f"Translate the following segment into Spanish, without additional explanation.\n\n{t}"}]}
    r = urllib.request.Request(URL, json.dumps(body).encode(), {"Content-Type": "application/json"})
    t0 = time.perf_counter()
    out = json.load(urllib.request.urlopen(r, timeout=120))["message"]["content"].strip()
    return out, time.perf_counter() - t0
for i, f in enumerate(FRASES):
    out, dt = traduz(f)
    print(f"[{dt:5.2f}s]{' (frio)' if i == 0 else ''} {f}\n         -> {out}")