#!/bin/bash

# LogSAD Environment Setup Script
# This script creates and configures the complete environment for LogSAD
# Run this script to set up all dependencies and requirements

set -e  # Exit on any error

echo "LogSAD Environment Setup"
echo "======================="
echo "Setting up conda environment and dependencies..."

# Create conda environment
echo "Creating conda environment 'logsad' with Python 3.10..."
conda create -n logsad python=3.10 -y

# Activate environment
echo "Activating environment..."
conda activate logsad

# Install PyTorch with CUDA support
echo "Installing PyTorch with CUDA 12.1 support..."
pip install torch==2.1.2+cu121 torchvision -f https://mirrors.aliyun.com/pytorch-wheels/cu121/

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

# Install specific numpy version for compatibility
echo "Installing compatible numpy version..."
pip install numpy==1.23.1

echo ""
echo "✓ Environment setup completed successfully!"
echo ""
echo "To activate the environment, run:"
echo "  conda activate logsad"
echo ""
echo "To run evaluations, use:"
echo "  bash scripts/run_few_shot.sh      # For few-shot protocol"
echo "  bash scripts/run_full_data.sh     # For full-data protocol"