#!/bin/bash

echo "Setting up Wall Color Visualizer Backend..."

# Create virtual environment
python3 -m venv venv
source venv/bin/activate

# Upgrade pip
pip install --upgrade pip

# Install dependencies
pip install -r requirements.txt

# Download SAM model checkpoint (vit_h - largest and most accurate)
echo "Downloading SAM model checkpoint..."
wget https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth

# Alternative: Download smaller models if needed
# wget https://dl.fbaipublicfiles.com/segment_anything/sam_vit_l_0b3195.pth
# wget https://dl.fbaipublicfiles.com/segment_anything/sam_vit_b_01ec64.pth

echo "Setup complete!"
echo "To start the server, run: uvicorn main:app --reload --host 0.0.0.0 --port 8000"