#!/bin/bash
# Complete BeastBullet + TinyLlama Setup Script
# Following instructions from https://potatobullet.com/beastbullet-project/

echo "======================================================================"
echo "🧠 BeastBullet + TinyLlama Setup"
echo "======================================================================"

# Check if venv exists, create if not
if [ ! -d "venv" ]; then
    echo "📦 Creating virtual environment..."
    python3 -m venv venv
fi

# Activate venv and install dependencies
echo "📥 Installing dependencies..."
./venv/bin/pip install torch ollama --quiet

echo ""
echo "======================================================================"
echo "Step 3: Load Sonnet-Quality Configuration"
echo "======================================================================"

./venv/bin/python3 << 'EOF'
import sys
sys.path.insert(0, 'code')

from beastbullet_config import BeastBulletConfig, create_high_quality_system
import json

# Load Sonnet-quality settings
with open("configs/config_sonnet_quality.json") as f:
    config = json.load(f)

# Create system with high-quality config
grok, config_obj = create_high_quality_system()

print("✅ BeastBullet loaded with Sonnet-quality settings")
print()
print("Key Settings:")
print(f"  • Confidence threshold: {config_obj.ISL_CONFIDENCE_THRESHOLD}")
print(f"  • Chain-of-thought: Enabled")
print(f"  • Validator: Strict mode (0% false positives)")
print(f"  • Evidence sources: Minimum {config_obj.GROK_MIN_EVIDENCE_SOURCES}")
EOF

echo ""
echo "======================================================================"
echo "Step 4: Integrate with TinyLlama"
echo "======================================================================"

./venv/bin/python3 step4_integrate.py

echo ""
echo "======================================================================"
echo "✅ Setup Complete!"
echo "======================================================================"
echo ""
echo "Next steps:"
echo "  1. Run a query: ./venv/bin/python3 -c 'from step4_integrate import *; ...'"
echo "  2. See examples in code/integration_example.py"
echo ""