setup_demo.sh

#!/bin/bash
# setup_demo.sh - Quick setup script for demo

echo "╔══════════════════════════════════════════════════════════╗"
echo "║  Microsoft Graph Red Team Framework - Demo Setup        ║"
echo "╚══════════════════════════════════════════════════════════╝"
echo ""

# Check if virtual environment exists
if [ ! -d "venv311" ]; then
    echo "❌ Virtual environment not found (venv311)"
    echo "Creating new virtual environment..."
    python3.11 -m venv venv311
fi

# Activate virtual environment
echo "✓ Activating virtual environment..."
source venv311/bin/activate

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

# Check for .env file
if [ ! -f ".env" ]; then
    echo "⚠️  No .env file found"
    echo "Creating .env template..."
    echo "OPENAI_API_KEY=your-openai-key-here" > .env
    echo "⚠️  Please edit .env and add your OpenAI API key"
fi

# Check for token.txt
if [ ! -f "token.txt" ]; then
    echo "⚠️  No token.txt file found"
    touch token.txt
    echo "⚠️  Please add your Microsoft Graph access token to token.txt"
fi

echo ""
echo "╔══════════════════════════════════════════════════════════╗"
echo "║  Setup Status                                            ║"
echo "╚══════════════════════════════════════════════════════════╝"

# Check OpenAI key
if grep -q "your-openai-key-here" .env 2>/dev/null || ! grep -q "OPENAI_API_KEY" .env 2>/dev/null; then
    echo "  [ ] OpenAI API Key - NOT CONFIGURED"
    echo "      → Edit .env and add your key"
else
    echo "  [✓] OpenAI API Key - Configured"
fi

# Check token
if [ ! -s "token.txt" ]; then
    echo "  [ ] Graph Token - NOT CONFIGURED"
    echo "      → Add your token to token.txt"
else
    echo "  [✓] Graph Token - Present"
fi

echo "  [✓] Dependencies - Installed"
echo ""

# Test OpenAI connection
echo "Testing OpenAI API connection..."
python test_openai.py

echo ""
echo "╔══════════════════════════════════════════════════════════╗"
echo "║  Ready to Demo!                                          ║"
echo "╚══════════════════════════════════════════════════════════╝"
echo ""
echo "Run the demo with:"
echo "  source venv311/bin/activate && python main_demo.py"
echo ""
echo "Or standard mode:"
echo "  source venv311/bin/activate && python main.py"
echo ""