#!/bin/bash
# Local Development Script for Depth Anything 3
# This script sets up and runs the app locally for fast iteration

set -e

SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
cd "$SCRIPT_DIR"

echo "🔧 Setting up local development environment..."

# Check if virtual environment exists
if [ ! -d "venv" ]; then
    echo "📦 Creating virtual environment..."
    python3 -m venv venv
fi

# Activate virtual environment
echo "🐍 Activating virtual environment..."
source venv/bin/activate

# Upgrade pip
echo "⬆️  Upgrading pip..."
pip install --upgrade pip setuptools wheel --quiet

# Install/upgrade dependencies
echo "📚 Installing dependencies..."
pip install -q -r requirements.txt

# Check if model directory exists (optional - will download on first use)
MODEL_DIR="${DA3_MODEL_DIR:-depth-anything/DA3NESTED-GIANT-LARGE}"
if [ ! -d "$MODEL_DIR" ]; then
    echo "⚠️  Model directory not found: $MODEL_DIR"
    echo "   The model will be downloaded automatically on first use."
fi

# Create workspace directories
mkdir -p workspace/gradio
mkdir -p workspace/gallery

echo ""
echo "✅ Setup complete!"
echo ""
echo "🚀 Starting local development server..."
echo "   URL: http://127.0.0.1:7860"
echo "   Press Ctrl+C to stop"
echo ""

# Run the local app
python app_local.py