#!/bin/bash
# Run DreamTail on Jetson AGX Orin
set -e

echo "🎨 Starting DreamTail..."

# Configuration
CONTAINER_NAME="dreamtail"
PORT=8765
MODELS_DIR="/mnt/nvme/models"  # Models on NVMe
STORAGE_DIR="/mnt/nvme/dreamtail"  # DreamTail storage on NVMe

# Create storage directory if it doesn't exist
mkdir -p "$STORAGE_DIR"

# Stop existing container if running
if docker ps -a --format '{{.Names}}' | grep -q "^${CONTAINER_NAME}$"; then
    echo "Stopping existing DreamTail container..."
    docker stop "$CONTAINER_NAME" 2>/dev/null || true
    docker rm "$CONTAINER_NAME" 2>/dev/null || true
fi

# Run container
echo "Starting DreamTail container..."
docker run -d \
  --name "$CONTAINER_NAME" \
  --runtime=nvidia \
  --restart unless-stopped \
  -p ${PORT}:8765 \
  -v "${MODELS_DIR}:/app/models" \
  -v "${STORAGE_DIR}:/app/storage" \
  -e DREAMTAIL_STORAGE=/app/storage \
  -e DREAMTAIL_MODELS=/app/models \
  -e LOG_LEVEL=INFO \
  dreamtail:latest

echo "✅ DreamTail started!"
echo ""
echo "API available at: http://bigorin:${PORT}"
echo ""
echo "To check logs:"
echo "  docker logs -f ${CONTAINER_NAME}"
echo ""
echo "To stop:"
echo "  docker stop ${CONTAINER_NAME}"