#!/bin/bash

# Start Ollama server in the background
# We bind it to port 11434 as expected by HF Spaces
OLLAMA_HOST=0.0.0.0:11434 ollama serve &

# Wait for the server to be ready
until curl -s localhost:11434/api/tags > /dev/null; do
  echo "Waiting for Ollama server..."
  sleep 2
done

# Pull the cloud model using the secret environment variable
# HF automatically injects your secret as an ENV variable
# settting OLLAMA_API_KEY not necessary
# if [ -n "$OLLAMA_API_KEY" ]; then
    echo "Authenticating and pulling cloud model..."
    ollama pull gemma4:31b-cloud
    # ollama pull deepseek-v3.2:cloud
    #ollama pull minimax-m2.7:cloud
    # ollama pull gemini-3-flash-preview:cloud
    ollama pull minimax-m2.5:cloud
    ollama pull gpt-oss:120b-cloud
    ollama pull gpt-oss:20b-cloud
    # ollama pull digitsflow/bonsai-8b
    # ollama pull qwen3.5:0.8b
    ollama pull qwen3-coder-next:cloud

    ollama pull qwen3.5:9b
# else
    # echo "Error: OLLAMA_API_KEY secret not found in Space settings."
# fi

# export ACCESS_TOKEN=${ACCESS_TOKEN}
# export OLLAMA_API_KEY=${OLLAMA_API_KEY}

envsubst '${ACCESS_TOKEN}' < /etc/nginx/nginx.conf.template > /etc/nginx/nginx.conf

echo "Starting Nginx Reverse Proxy..."
nginx -g "daemon on;"

echo "------------------------------------------------------"
echo "READY: Port 7860 is now active"
echo "------------------------------------------------------"

ollama ls
ollama signin

wait -n