# -*- coding: utf-8 -*-
"""CLI-чат с ZuperAI (№6). Терминальный диалог, как `ollama run`, но с нашим стеком.

Запуск:
  venv/Scripts/python.exe chat.py [deepseek|lmstudio|local]

Выход: 'exit' / 'quit' / Ctrl+C.
"""
import sys

from brain import pick, generate

SYSTEM = ("Ты — ZuperAI, опытный программист. Отвечай по делу, на русском. "
          "Пиши рабочий код, когда просят. Комментарии и докстринги — на русском.")


def main():
    backend = pick(sys.argv[1] if len(sys.argv) > 1 else None)
    print(f"[*] мозг: {backend}. Пиши вопрос (или exit/выход).")
    history = [{"role": "system", "content": SYSTEM}]
    while True:
        try:
            line = input(">>> ")
        except (EOFError, KeyboardInterrupt):
            print()
            break
        line = line.strip()
        if not line:
            continue
        if line.lower() in ("exit", "quit", "выход"):
            break
        history.append({"role": "user", "content": line})
        ans = generate(backend, history, 1024)
        history.append({"role": "assistant", "content": ans})
        print(ans)
        print()


if __name__ == "__main__":
    main()