#!/bin/bash

echo "🚀 Setup Base LLM Environment"
echo "=============================="

# Check if Python 3.8+ is available
python_version=$(python3 --version 2>&1 | grep -oP '\d+\.\d+' | head -1)
if [[ -z "$python_version" ]]; then
    echo "❌ Python 3 tidak ditemukan!"
    echo "Silakan install Python 3.8+ terlebih dahulu"
    exit 1
fi

echo "✅ Python version: $python_version"

# Create virtual environment
echo "📦 Creating virtual environment..."
python3 -m venv venv

# Activate virtual environment
echo "🔧 Activating virtual environment..."
source venv/bin/activate

# Upgrade pip
echo "⬆️  Upgrading pip..."
pip install --upgrade pip

# Install requirements
echo "📚 Installing requirements..."
pip install -r requirements.txt

# Install additional tools
echo "🛠️  Installing additional tools..."
pip install huggingface-cli

echo ""
echo "✅ Setup selesai!"
echo ""
echo "📋 Langkah selanjutnya:"
echo "1. Aktifkan virtual environment: source venv/bin/activate"
echo "2. Set HuggingFace token: export HUGGINGFACE_TOKEN='your_token'"
echo "3. Jalankan: python scripts/download_model.py"
echo "4. Jalankan: python scripts/finetune_lora.py"
echo ""
echo "💡 Tips:"
echo "- Selalu aktifkan venv sebelum menjalankan script"
echo "- Gunakan 'deactivate' untuk keluar dari venv"
echo "- Pastikan GPU tersedia untuk training"