#!/bin/bash

# VAMGUARD TITAN Setup Script

set -e

echo "🚀 VAMGUARD TITAN Setup"
echo "======================="
echo ""

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

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

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

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

# Create .env from example if it doesn't exist
if [ ! -f .env ]; then
    echo ""
    echo "Creating .env file from template..."
    cp .env.example .env
    echo "⚠️  Please edit .env with your API keys and configuration"
else
    echo ""
    echo ".env file already exists, skipping..."
fi

# Test imports
echo ""
echo "Testing imports..."
python - <<EOF
try:
    import streamlit
    import anthropic
    from huggingface_hub import HfApi
    import yaml
    print("✅ All required packages imported successfully")
except Exception as e:
    print(f"❌ Import error: {e}")
    exit(1)
EOF

echo ""
echo "✅ Setup complete!"
echo ""
echo "Next steps:"
echo "1. Edit .env with your API keys:"
echo "   - HF_TOKEN (HuggingFace)"
echo "   - HF_USERNAME (HuggingFace)"
echo "   - ANTHROPIC_API_KEY (Claude AI)"
echo ""
echo "2. Update config.yaml with your space names"
echo ""
echo "3. Run the Streamlit app:"
echo "   streamlit run app.py"
echo ""
echo "4. Or run automation:"
echo "   python automation.py --help"
echo ""