#!/bin/bash
# BeastBullet One-Click Installer
# Sonnet-level AI on Potato Hardware
# By Shrikant Bhosale (Thalaivar)

set -e

echo "════════════════════════════════════════════════════════════════════"
echo "🥔 BeastBullet v1.0 - Premise-Locked Edition"
echo "════════════════════════════════════════════════════════════════════"
echo "Sonnet-level AI reasoning on potato hardware"
echo "91% quality | 96% confidence | Zero hallucinations"
echo ""

# Check prerequisites
echo "🔍 Checking prerequisites..."

if ! command -v python3 &> /dev/null; then
    echo "❌ Python 3 not found! Please install Python 3.8+"
    exit 1
fi

if ! command -v git &> /dev/null; then
    echo "❌ Git not found! Please install git"
    exit 1
fi

if ! command -v ollama &> /dev/null; then
    echo "❌ Ollama not found!"
    echo "Install from: https://ollama.com"
    echo "Then run this script again"
    exit 1
fi

echo "✅ All prerequisites found"

# Pull TinyLlama
echo ""
echo "📥 Pulling TinyLlama synthesizer..."
ollama pull tinyllama || {
    echo "⚠️  TinyLlama pull failed, but continuing..."
}

# Ask about upgrade
echo ""
read -p "🚀 Use stronger synthesizer? (Qwen2.5-7B recommended for best results) [y/N]: " upgrade
if [[ $upgrade =~ ^[Yy]$ ]]; then
    echo "📥 Pulling Qwen2.5-7B (this may take a few minutes)..."
    ollama pull qwen2.5:7b
    SYNTH_MODEL="qwen2.5:7b"
else
    SYNTH_MODEL="tinyllama"
fi

# Clone repository
echo ""
echo "📦 Cloning BeastBullet repository..."
INSTALL_DIR="$HOME/beastbullet"

if [ -d "$INSTALL_DIR" ]; then
    echo "⚠️  Directory $INSTALL_DIR already exists"
    read -p "Remove and reinstall? [y/N]: " reinstall
    if [[ $reinstall =~ ^[Yy]$ ]]; then
        rm -rf "$INSTALL_DIR"
    else
        echo "Installation cancelled"
        exit 0
    fi
fi

git clone https://huggingface.co/SetMD/beastbullet-experts "$INSTALL_DIR"
cd "$INSTALL_DIR"

# Create config
echo ""
echo "⚙️  Configuring BeastBullet..."
cat > config.json <<EOF
{
    "synthesizer_model": "$SYNTH_MODEL",
    "experts_dir": "experts/",
    "isl_threshold": 0.80,
    "premise_lock_enabled": true,
    "confidence_penalty_per_violation": 0.20
}
EOF

# Create launcher script
cat > beastbullet.sh <<'EOF'
#!/bin/bash
cd "$HOME/beastbullet"
python3 main.py "$@"
EOF
chmod +x beastbullet.sh

# Add to PATH
if [[ ":$PATH:" != *":$HOME/beastbullet:"* ]]; then
    echo 'export PATH="$HOME/beastbullet:$PATH"' >> ~/.bashrc
    echo 'alias beastbullet="$HOME/beastbullet/beastbullet.sh"' >> ~/.bashrc
fi

echo ""
echo "════════════════════════════════════════════════════════════════════"
echo "🏆 BEASTBULLET INSTALLED SUCCESSFULLY!"
echo "════════════════════════════════════════════════════════════════════"
echo ""
echo "📍 Installation directory: $INSTALL_DIR"
echo "🤖 Synthesizer model: $SYNTH_MODEL"
echo ""
echo "🚀 Quick Start:"
echo "   cd $INSTALL_DIR"
echo "   python3 main.py"
echo ""
echo "   Or (after restart):"
echo "   beastbullet"
echo ""
echo "📚 Documentation:"
echo "   README.md - Full documentation"
echo "   BEASTBULLET_V1_SPEC.md - Technical specification"
echo "   VICTORY_RUN_*.md - Test results"
echo ""
echo "🔥 Features:"
echo "   • 18 specialized expert models"
echo "   • Premise-lock validator (zero logic drift)"
echo "   • 91% quality score, 96% confidence"
echo "   • Runs on potato hardware (4GB RAM)"
echo ""
echo "Mind it! 🎯"
echo "════════════════════════════════════════════════════════════════════"