#!/bin/bash
set -e

datetime_prefix() {
  date "+[%Y-%m-%d %H:%M:%S]"
}

echo "$(datetime_prefix) Starting Whisper Transcriptor in MCP mode..."

# Check if image exists
if ! docker image inspect transcriptor-apimcp:latest &> /dev/null; then
    echo "$(datetime_prefix) Image not found. Building first..."
    ./docker-build.sh
fi

# Stop and remove existing container if running
if docker ps -a --format '{{.Names}}' | grep -q '^transcriptor-mcp$'; then
    echo "$(datetime_prefix) Stopping existing container..."
    docker stop transcriptor-mcp || true
    docker rm transcriptor-mcp || true
fi

# Run the container in MCP mode (interactive stdio)
echo "$(datetime_prefix) Starting MCP server in stdio mode..."
echo "$(datetime_prefix) Press Ctrl+C to stop"
echo ""

docker run -it --rm \
    --name transcriptor-mcp \
    --gpus all \
    -e SERVER_MODE=mcp \
    -e CUDA_VISIBLE_DEVICES=0 \
    -e TRANSCRIPTION_MODEL=large-v3 \
    -e TRANSCRIPTION_DEVICE=auto \
    -e TRANSCRIPTION_COMPUTE_TYPE=auto \
    -e JOB_QUEUE_MAX_SIZE=100 \
    -v "$(pwd)/models:/models" \
    -v "$(pwd)/outputs:/outputs" \
    -v "$(pwd)/logs:/logs" \
    transcriptor-apimcp:latest