File size: 2,152 Bytes
79ea999 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 |
#!/bin/bash
# Production startup script for HonestAI
# This script validates environment and starts the application with Gunicorn
set -e # Exit on error
echo "============================================================"
echo "HonestAI Production Startup Script"
echo "============================================================"
# Validate HF_TOKEN
if [ -z "$HF_TOKEN" ]; then
echo "ERROR: HF_TOKEN environment variable is not set"
echo "Please set HF_TOKEN in Space Settings → Repository secrets"
exit 1
fi
echo "✓ HF_TOKEN is set"
# Validate OMP_NUM_THREADS
if [ -z "$OMP_NUM_THREADS" ]; then
echo "WARNING: OMP_NUM_THREADS not set, defaulting to 4"
export OMP_NUM_THREADS=4
elif ! [[ "$OMP_NUM_THREADS" =~ ^[0-9]+$ ]] || [ "$OMP_NUM_THREADS" -le 0 ]; then
echo "WARNING: Invalid OMP_NUM_THREADS='$OMP_NUM_THREADS', setting to 4"
export OMP_NUM_THREADS=4
fi
export MKL_NUM_THREADS=$OMP_NUM_THREADS
echo "✓ OMP_NUM_THREADS set to $OMP_NUM_THREADS"
# Validate MKL_NUM_THREADS
if [ -z "$MKL_NUM_THREADS" ]; then
export MKL_NUM_THREADS=$OMP_NUM_THREADS
fi
echo "✓ MKL_NUM_THREADS set to $MKL_NUM_THREADS"
# Set secure log directory
LOG_DIR=${LOG_DIR:-/tmp/logs}
mkdir -p "$LOG_DIR"
chmod 700 "$LOG_DIR" 2>/dev/null || echo "Warning: Could not set log directory permissions"
echo "✓ Log directory: $LOG_DIR"
# Set default port if not specified
PORT=${PORT:-7860}
echo "✓ Port: $PORT"
# Set default workers (adjust based on CPU cores)
WORKERS=${GUNICORN_WORKERS:-4}
echo "✓ Gunicorn workers: $WORKERS"
# Set rate limiting
RATE_LIMIT_ENABLED=${RATE_LIMIT_ENABLED:-true}
echo "✓ Rate limiting: $RATE_LIMIT_ENABLED"
echo "============================================================"
echo "Starting Gunicorn production server..."
echo "============================================================"
# Start Gunicorn with proper configuration
exec gunicorn \
--bind "0.0.0.0:$PORT" \
--workers "$WORKERS" \
--threads 2 \
--timeout 120 \
--keep-alive 5 \
--access-logfile - \
--error-logfile - \
--log-level info \
--capture-output \
flask_api_standalone:app
|