setup_env.sh

#!/bin/bash

###########################################
# System Updates and Package Installation #
###########################################

# Update system
sudo apt update -y

# Install core dependencies
sudo apt install -y \
    python3-pip \
    nano \
    libgl1 \
    npm \
    ffmpeg \
    unzip

# Install build dependencies
sudo apt install -y \
    build-essential \
    cmake \
    libopenblas-dev \
    liblapack-dev \
    libx11-dev \
    libgtk-3-dev

# Install process manager
sudo npm install -g pm2@latest

############################
# Python Package Installation
############################

pip install git+https://github.com/huggingface/diffusers.git@6a51427b6a226591ccc40249721c486855f53e1c
pip install -e .

############################
# Environment Files Setup  #
############################

# Create miner.env if it doesn't exist
if [ -f "miner.env" ]; then
    echo "File 'miner.env' already exists. Skipping creation."
else
    cat > miner.env << 'EOL'
# Default options
#--------------------

# Detector Configuration
IMAGE_DETECTOR=CAMO                   # Options: CAMO, UCF, NPR, None
IMAGE_DETECTOR_CONFIG=camo.yaml       # Configs in base_miner/deepfake_detectors/configs
VIDEO_DETECTOR=TALL                   # Options: TALL, None
VIDEO_DETECTOR_CONFIG=tall.yaml       # Configs in base_miner/deepfake_detectors/configs

# Device Settings
IMAGE_DETECTOR_DEVICE=cpu             # Options: cpu, cuda
VIDEO_DETECTOR_DEVICE=cpu

# Subtensor Network Configuration
NETUID=34                            # Network User ID options: 34, 168
SUBTENSOR_NETWORK=finney             # Networks: finney, test, local
SUBTENSOR_CHAIN_ENDPOINT=wss://entrypoint-finney.opentensor.ai:443
                                     # Endpoints:
                                     # - wss://entrypoint-finney.opentensor.ai:443
                                     # - wss://test.finney.opentensor.ai:443/

# Wallet Configuration
WALLET_NAME=default
WALLET_HOTKEY=default

# Miner Settings
MINER_AXON_PORT=8091
BLACKLIST_FORCE_VALIDATOR_PERMIT=True # Force validator permit for blacklisting
EOL
    echo "File 'miner.env' created."
fi

# Create validator.env if it doesn't exist
if [ -f "validator.env" ]; then
    echo "File 'validator.env' already exists. Skipping creation."
else
    cat > validator.env << 'EOL'
# Default options
#--------------------

# Subtensor Network Configuration
NETUID=34                            # Network User ID options: 34, 168
SUBTENSOR_NETWORK=finney             # Networks: finney, test, local
SUBTENSOR_CHAIN_ENDPOINT=wss://entrypoint-finney.opentensor.ai:443
                                     # Endpoints:
                                     # - wss://entrypoint-finney.opentensor.ai:443
                                     # - wss://test.finney.opentensor.ai:443/

# Wallet Configuration
WALLET_NAME=default
WALLET_HOTKEY=default

# Validator Settings
VALIDATOR_AXON_PORT=8092             # If using RunPod, must be >= 70000 for symmetric mapping
VALIDATOR_PROXY_PORT=10913
DEVICE=cuda

# API Keys
WANDB_API_KEY=your_wandb_api_key_here
HUGGING_FACE_TOKEN=your_hugging_face_token_here
EOL
    echo "File 'validator.env' created."
fi

echo "Environment setup completed successfully."