set -e

echo "🔴 NETRAX AI - Vision System Setup"

echo "===================================="

echo ""

RED='\033[0;31m'

GREEN='\033[0;32m'

YELLOW='\033[1;33m'

NC='\033[0m' # No Color

OS="unknown"

if [[ "$OSTYPE" == "linux-gnu"* ]]; then

OS="linux"

elif [[ "$OSTYPE" == "darwin"* ]]; then

OS="macos"

elif [[ "$OSTYPE" == "msys" ]] || [[ "$OSTYPE" == "cygwin" ]]; then

OS="windows"

echo -e "${GREEN}Detected OS: $OS${NC}"

echo ""

echo "Checking Python version..."

if ! command -v python3 &> /dev/null; then

echo -e "${RED}Python 3 not found. Please install Python 3.10+${NC}"

exit 1

PYTHON_VERSION=$(python3 -c 'import sys; print(".".join(map(str, sys.version_info[:2])))')

echo -e "${GREEN}Python version: $PYTHON_VERSION${NC}"

if (( $(echo "$PYTHON_VERSION < 3.10" | bc -l) )); then

echo -e "${RED}Python 3.10+ required. Current: $PYTHON_VERSION${NC}"

exit 1

echo ""

echo "Creating directory structure..."

mkdir -p vision_engine

mkdir -p models

mkdir -p logs

touch vision_engine/__init__.py

echo -e "${GREEN}✓ Directories created${NC}"

echo ""

echo "Setting up virtual environment..."

if [ ! -d "venv" ]; then

python3 -m venv venv

echo -e "${GREEN}✓ Virtual environment created${NC}"

echo -e "${YELLOW}Virtual environment already exists${NC}"

source venv/bin/activate

echo "Upgrading pip..."

pip install --upgrade pip setuptools wheel

echo -e "${GREEN}✓ Pip upgraded${NC}"

echo ""

echo "Installing dependencies..."

pip install -r requirements.txt

echo -e "${GREEN}✓ Dependencies installed${NC}"

echo ""

if [ ! -f ".env" ]; then

echo "Creating .env file..."

cp .env.example .env

echo -e "${GREEN}✓ Configuration file created${NC}"

echo -e "${YELLOW}⚠ Please edit .env with your settings${NC}"

echo -e "${YELLOW}.env file already exists${NC}"

echo ""

echo "Checking camera access..."

if [[ "$OS" == "linux" ]]; then

if [ -e "/dev/video0" ]; then

echo -e "${GREEN}✓ Camera found at /dev/video0${NC}"

else

echo -e "${YELLOW}⚠ No camera found at /dev/video0${NC}"

echo "Available video devices:"

ls -la /dev/video* 2>/dev/null || echo "None found"

fi

elif [[ "$OS" == "macos" ]]; then

echo -e "${YELLOW}⚠ Camera access: Grant permissions in System Preferences${NC}"

echo ""

echo "Testing MediaPipe installation..."

python3 -c "import mediapipe as mp; print('MediaPipe version:', mp.__version__)" && \

echo -e "${GREEN}✓ MediaPipe working${NC}" || \

echo -e "${RED}✗ MediaPipe installation failed${NC}"

echo ""

echo "Testing OpenCV installation..."

python3 -c "import cv2; print('OpenCV version:', cv2.__version__)" && \

echo -e "${GREEN}✓ OpenCV working${NC}" || \

echo -e "${RED}✗ OpenCV installation failed${NC}"

echo ""

read -p "Install object detection (YOLOv8)? This is heavy (~500MB). [y/N] " -n 1 -r

if [[ $REPLY =~ ^[Yy]$ ]]; then

echo "Installing ultralytics..."

pip install ultralytics

echo -e "${GREEN}✓ Object detection installed${NC}"

echo -e "${YELLOW}Skipping object detection${NC}"

echo ""

echo "Checking GPU availability..."

python3 -c "

" 2>/dev/null || echo -e "${YELLOW}Using CPU mode${NC}"

echo ""

if [[ "$OS" == "linux" ]]; then

read -p "Create systemd service for auto-start? [y/N] " -n 1 -r

echo

if [[ $REPLY =~ ^[Yy]$ ]]; then

SERVICE_FILE="/etc/systemd/system/netrax-vision.service"

WORK_DIR=$(pwd)

sudo tee $SERVICE_FILE > /dev/null <<EOF

sudo systemctl daemon-reload

echo -e "${GREEN}✓ Service created${NC}"

echo "Start with: sudo systemctl start netrax-vision"

echo "Enable auto-start: sudo systemctl enable netrax-vision"

fi

echo ""

if command -v docker &> /dev/null; then

echo -e "${GREEN}✓ Docker detected${NC}"

read -p "Build Docker image? [y/N] " -n 1 -r

echo

if [[ $REPLY =~ ^[Yy]$ ]]; then

echo "Building Docker image..."

docker-compose build

echo -e "${GREEN}✓ Docker image built${NC}"

echo "Run with: docker-compose up -d"

fi

echo -e "${YELLOW}Docker not found - skipping Docker setup${NC}"

echo ""

echo "===================================="

echo -e "${GREEN}🔴 Setup Complete!${NC}"

echo "===================================="

echo ""

echo "Next steps:"

echo "1. Edit .env with your configuration"

echo "2. Run the server:"

echo " python main.py"

echo " OR"

echo " docker-compose up -d"

echo ""

echo "3. Open frontend:"

echo " Open index.html in your browser"

echo ""

echo "4. Access API:"

echo " http://localhost:8000"

echo " ws://localhost:8000/ws"

echo ""

echo -e "${YELLOW}⚠ Remember to grant camera permissions!${NC}"

echo ""

echo "Documentation: README.md"

echo "Support: Check troubleshooting section"

echo ""

echo -e "${GREEN}NETRAX IS READY TO WATCH 👁️${NC}"