#!/bin/bash
# ARC Engine v3.0 Quick Start
# ============================

echo "🤖 ARC Engine v3.0 - Quick Start"
echo "================================="

# Check Python version
PYTHON_VERSION=$(python3 --version 2>&1 | cut -d' ' -f2 | cut -d'.' -f1,2)
echo "Python version: $PYTHON_VERSION"

# Check if requirements are installed
echo ""
echo "Checking dependencies..."

if ! python3 -c "import torch" 2>/dev/null; then
    echo "⚠️  PyTorch not found. Installing core dependencies..."
    pip install torch transformers accelerate peft bitsandbytes datasets trl safetensors tqdm matplotlib requests
fi

if python3 -c "import torch" 2>/dev/null; then
    echo "✓ PyTorch installed"
else
    echo "❌ PyTorch installation failed"
    exit 1
fi

if python3 -c "import transformers" 2>/dev/null; then
    echo "✓ Transformers installed"
else
    echo "❌ Transformers not found"
    exit 1
fi

# Check GPU
echo ""
echo "Checking GPU..."
python3 -c "import torch; print(f'CUDA available: {torch.cuda.is_available()}'); print(f'Device: {torch.cuda.get_device_name(0) if torch.cuda.is_available() else \"CPU\"}') if torch.cuda.is_available() else None"

# Run the engine
echo ""
echo "================================="
echo "Starting ARC Engine v3.0..."
echo "================================="
echo ""
echo "First run will download the model (~16GB)"
echo "Type 'help' for commands, 'help <topic>' for specific help"
echo ""

python3 arc_engine_v30_fixed.py