setup.sh

#!/bin/bash
# setup.sh - Complete setup script for WildGuard AI

set -e

echo "🦁 WildGuard AI - Setup Script"
echo "================================"

# Check Python version
echo "✓ Checking Python..."
PYTHON_VERSION=$(python3 --version 2>&1 | awk '{print $2}')
echo "  Python version: $PYTHON_VERSION"

# Create virtual environment
echo "✓ Creating virtual environment..."
python3 -m venv venv
source venv/bin/activate

# Install dependencies
echo "✓ Installing Python dependencies..."
pip install --upgrade pip
pip install -r requirements.txt

# Copy environment file
echo "✓ Setting up environment..."
if [ ! -f .env ]; then
    cp .env.example .env
    echo "  Created .env file (edit it for custom config)"
fi

# Initialize database
echo "✓ Initializing database..."
python3 -c "
from business_logic import init_db
db = init_db()
print('  Database initialized successfully')
"

# Setup frontend
echo "✓ Setting up frontend..."
cd frontend
npm install
cd ..

# Verify model files
echo "✓ Verifying model files..."
for file in animal_classifier.h5 class_indices.json risk_dictionary.json; do
    if [ -f "$file" ]; then
        echo "  ✓ $file"
    else
        echo "  ✗ $file NOT FOUND"
    fi
done

# Print next steps
echo ""
echo "🎉 Setup complete!"
echo ""
echo "Next steps:"
echo "1. Start backend:   python api.py"
echo "2. Start frontend:  cd frontend && npm run dev"
echo "3. Open browser:    http://localhost:5173"
echo ""
echo "For more info, see README.md or DEPLOYMENT.md"