setup.sh

#!/bin/bash

set -e  # Exit immediately if a command exits with a non-zero status

# Colors for output
GREEN="\e[32m"
YELLOW="\e[33m"
RED="\e[31m"
RESET="\e[0m"

# Function to print status messages
echo_status() {
    echo -e "${GREEN}[✔] $1${RESET}"
}

echo_warning() {
    echo -e "${YELLOW}[!] $1${RESET}"
}

echo_error() {
    echo -e "${RED}[✘] $1${RESET}" >&2
}

# Install Oh My Zsh if not already installed
echo_status "Checking for Oh My Zsh..."
if [ ! -d "$HOME/.oh-my-zsh" ]; then
    echo_status "Installing Oh My Zsh..."
    sudo apt update && sudo apt install -y zsh
    sh -c "$(curl -fsSL https://raw.githubusercontent.com/ohmyzsh/ohmyzsh/master/tools/install.sh)" || {
        echo_error "Failed to install Oh My Zsh."
        exit 1
    }
else
    echo_warning "Oh My Zsh is already installed. Skipping..."
fi

# Create virtual environment
echo_status "Creating Python virtual environment..."
python3 -m venv venv --system-site-packages || {
    echo_error "Failed to create virtual environment."
    exit 1
}

echo_status "Virtual environment created successfully."

# Activate the virtual environment
echo_status "Activating virtual environment..."
source venv/bin/activate || {
    echo_error "Failed to activate virtual environment."
    exit 1
}

echo_status "Virtual environment activated."

# Upgrade pip
echo_status "Upgrading pip..."
pip install --upgrade pip || {
    echo_error "Failed to upgrade pip."
    exit 1
}

echo_status "Pip upgraded successfully."

# Install required dependencies
echo_status "Installing dependencies: requirements.txt"
venv/bin/pip install -r requirements.txt || {
    echo_error "Failed to install dependencies."
    exit 1
}

pip install --no-cache-dir "git+https://github.com/Tencent/ncnn.git"

echo_status "All dependencies installed successfully."

echo -e "${GREEN}Setup complete. Use 'source venv/bin/activate' to activate the environment.${RESET}"