#!/bin/bash
set -e

MODEL="gpt-oss:20b"

echo "Starting Ollama..."
ollama serve &

echo "Waiting for Ollama..."
RETRIES=60
until curl -sf http://localhost:11434/api/tags >/dev/null 2>&1; do
    RETRIES=$((RETRIES - 1))
    if [ "$RETRIES" -le 0 ]; then
        echo "ERROR: Ollama did not start."
        exit 1
    fi
    sleep 2
done
echo "Ollama ready."

echo "Pulling model: $MODEL"
ollama pull "$MODEL"
echo "Model ready."

cat > /home/user/.bashrc << 'BASHEOF'
export ANTHROPIC_AUTH_TOKEN=ollama
export ANTHROPIC_API_KEY=ollama
export ANTHROPIC_BASE_URL=http://localhost:11434
export PS1="[claude] $ "
echo ""
echo "  Claude Code CLI -- model: gpt-oss:20b"
echo "  Run: claude --model gpt-oss:20b"
echo ""
claude --model gpt-oss:20b
BASHEOF

echo "Launching ttyd on :7860..."
exec ttyd -p 7860 -W -t fontSize=15 /bin/bash --login