@echo off
chcp 65001 >nul
title ZuperAI - установщик
echo ============================================
echo   ZuperAI-14B — локальная установка
echo ============================================
echo.

REM --- 1. Проверка Python ---
where python >nul 2>nul
if errorlevel 1 (
    echo [!] Python не найден. Скачай с https://www.python.org/downloads/
    echo     При установке отметь галочку "Add Python to PATH".
    pause
    exit /b 1
)

REM --- 2. Виртуальное окружение + зависимости ---
if not exist venv (
    echo [*] Создаю виртуальное окружение...
    python -m venv venv
)
call venv\Scripts\activate.bat
echo [*] Ставлю зависимости...
pip install -q --upgrade pip >nul 2>nul
pip install -q fastapi uvicorn transformers peft torch accelerate datasets >nul 2>nul

REM --- 3. Скачивание модели (если ещё нет) ---
if not exist ZuperAI-14B-Q4_K_M.gguf (
    echo [*] Скачиваю модель ZuperAI-14B (Q4_K_M, ~9 ГБ). Это займёт время...
    curl -L --progress-bar -o ZuperAI-14B-Q4_K_M.gguf ^
        https://huggingface.co/Tyesee/ZuperAI-14B/resolve/main/ZuperAI-14B-Q4_K_M.gguf
) else (
    echo [*] Модель уже скачана.
)

REM --- 4. Копия в LM Studio (если установлен) ---
if exist "%USERPROFILE%\.lmstudio\models" (
    mkdir "%USERPROFILE%\.lmstudio\models\ZuperAI" 2>nul
    copy /Y ZuperAI-14B-Q4_K_M.gguf "%USERPROFILE%\.lmstudio\models\ZuperAI\" >nul
    echo [*] Модель добавлена в LM Studio (открой LM Studio, чтобы она появилась).
)

REM --- 5. Копия в Ollama (если установлен) ---
where ollama >nul 2>nul
if not errorlevel 1 (
    if exist publish\Modelfile (
        echo [*] Создаю модель в Ollama...
        ollama create ZuperAI-14B -f publish\Modelfile
    )
)

REM --- 6. Запуск веб-морды ---
echo.
echo [*] Запускаю веб-морду: http://localhost:8001
echo     (закрыть - Ctrl+C)
echo.
python webui.py
pause