#!/bin/bash
#
# 🚀 QUICK START - Qwen-Soloband на RunPod
# Скрипт быстрой установки с нуля
#

set -e  # Exit on error

echo "╔════════════════════════════════════════════════════════════╗"
echo "║  🚀 QWEN-SOLOBAND QUICK START                             ║"
echo "║  Автоматическая установка на RunPod                       ║"
echo "╚════════════════════════════════════════════════════════════╝"
echo ""

# =================================================================
# ПРОВЕРКА HF_TOKEN
# =================================================================

if [ -z "$HF_TOKEN" ]; then
    echo "❌ ERROR: HF_TOKEN not set!"
    echo ""
    echo "Пожалуйста, установите HF_TOKEN перед запуском:"
    echo "  export HF_TOKEN='your_token_here'"
    echo ""
    echo "Или запустите:"
    echo "  HF_TOKEN='your_token_here' ./quick_start.sh"
    echo ""
    echo "Получить токен: https://huggingface.co/settings/tokens"
    exit 1
fi

echo "✓ HF_TOKEN найден"

# =================================================================
# УСТАНОВКА ЗАВИСИМОСТЕЙ
# =================================================================

echo ""
echo "[1/6] Устанавливаем Hugging Face CLI..."
export HF_HUB_ENABLE_HF_TRANSFER=0
pip install -q huggingface_hub[cli]

echo ""
echo "[2/6] Авторизуемся в Hugging Face..."
hf auth login --token "$HF_TOKEN"

# =================================================================
# СКАЧИВАНИЕ ПРОЕКТА
# =================================================================

echo ""
echo "[3/6] Переходим в /workspace..."
cd /workspace

echo ""
echo "[4/6] Скачиваем проект Qwen-ImageForFlo-Advanced..."
if [ -d "Qwen-ImageForFlo-Advanced" ]; then
    echo "  ⚠️  Папка уже существует, обновляем..."
    cd Qwen-ImageForFlo-Advanced
    hf download Gerchegg/Qwen-ImageForFlo-Advanced --repo-type space --local-dir .
else
    hf download Gerchegg/Qwen-ImageForFlo-Advanced --repo-type space --local-dir Qwen-ImageForFlo-Advanced
    cd Qwen-ImageForFlo-Advanced
fi

# =================================================================
# УСТАНОВКА PYTHON ЗАВИСИМОСТЕЙ
# =================================================================

echo ""
echo "[5/6] Устанавливаем Python зависимости..."
pip install -r requirements.txt

# =================================================================
# СОЗДАНИЕ СТРУКТУРЫ ПАПОК
# =================================================================

echo ""
echo "[6/6] Создаём папки для LoRA..."
mkdir -p /workspace/loras
echo "  ✓ /workspace/loras"

# =================================================================
# ГОТОВО!
# =================================================================

echo ""
echo "╔════════════════════════════════════════════════════════════╗"
echo "║  ✅ УСТАНОВКА ЗАВЕРШЕНА!                                  ║"
echo "╚════════════════════════════════════════════════════════════╝"
echo ""
echo "🎨 Для запуска приложения выполните:"
echo ""
echo "    cd /workspace/Qwen-ImageForFlo-Advanced"
echo "    python app.py"
echo ""
echo "📝 Опциональные переменные окружения:"
echo ""
echo "    MODEL_REPO='username/custom-model' python app.py"
echo "      └─ Использовать кастомную модель"
echo ""
echo "🎭 LoRA: Положите .safetensors файлы в /workspace/loras/"
echo ""
echo "📺 После запуска подключитесь через:"
echo "    RunPod Dashboard → Connect → HTTP Service [Port 7860]"
echo ""