#!/bin/bash

# Скачать Miniconda
echo "Downloading Miniconda..."
wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -O ~/Miniconda3-latest-Linux-x86_64.sh

# Установить Miniconda
echo "Installing Miniconda..."
bash ~/Miniconda3-latest-Linux-x86_64.sh -b -p $HOME/miniconda

# Добавить Miniconda в PATH
export PATH="$HOME/miniconda/bin:$PATH"
echo 'export PATH="$HOME/miniconda/bin:$PATH"' >> ~/.bashrc

# Инициализация conda
echo "Initializing Conda..."
conda init bash

# Создать и активировать окружение
echo "Creating Conda environment..."
conda create -n garmentcode python=3.9 -y
conda activate garmentcode

# Установить зависимости через pip
echo "Installing dependencies..."
pip install scipy pyaml>=6.0 svgwrite psutil gradio bpy matplotlib svgpathtools cairosvg nicegui trimesh libigl pyrender cgal numpy pygarment pyliblzfse tyro viser[examples]

# Настроить CUDA и другие пути
echo "Adding paths to .bashrc..."
echo 'export CUDA_HOME=/usr/local/cuda' >> ~/.bashrc
echo 'export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64' >> ~/.bashrc
echo 'export PATH=$PATH:$CUDA_HOME/bin' >> ~/.bashrc
echo 'export PYTHONPATH="$PYTHONPATH:/home/user/app/GarmentCode/"' >> ~/.bashrc

# Автоматическая активация окружения при запуске
echo "Adding automatic Conda activation to .bashrc..."
echo 'conda activate garmentcode' >> ~/.bashrc

# Установка warp для GarmentCode (замените <build and install warp> на реальную команду)
echo "Building and installing Warp for GarmentCode..."
# <build and install warp for GarmentCode>
chmod -R 777 /home/user/app/NvidiaWarp-GarmentCode/tools/
source ~/.bashrc
python /home/user/app/NvidiaWarp-GarmentCode/build_lib.py
pip install viser[examples] tyro
# Сообщение об успешной настройке
echo "Setup complete! Please restart your terminal or run 'source ~/.bashrc' to apply changes."