pyr0ball
f17aae3bd2
- manage.sh: dev command starts uvicorn --reload on :8503 and Vite dev server (auto-port from 5173); kills API on EXIT/INT/TERM trap - manage.sh: ENV_UI defaults to 'cf' env (overridable via AVOCET_ENV) - vite.config.ts: add server.proxy to forward /api to :8503 so Vite dev server can reach the backend without CORS issues
260 lines
9.1 KiB
Bash
Executable file
260 lines
9.1 KiB
Bash
Executable file
#!/usr/bin/env bash
|
|
# manage.sh — Avocet label tool manager
|
|
# Usage: ./manage.sh <command> [args]
|
|
set -euo pipefail
|
|
|
|
RED='\033[0;31m'; GREEN='\033[0;32m'; YELLOW='\033[1;33m'; BLUE='\033[0;34m'; NC='\033[0m'
|
|
info() { echo -e "${BLUE}[avocet]${NC} $*"; }
|
|
success() { echo -e "${GREEN}[avocet]${NC} $*"; }
|
|
warn() { echo -e "${YELLOW}[avocet]${NC} $*"; }
|
|
error() { echo -e "${RED}[avocet]${NC} $*" >&2; exit 1; }
|
|
|
|
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
|
cd "$SCRIPT_DIR"
|
|
|
|
PID_FILE=".avocet.pid"
|
|
PORT_FILE=".avocet.port"
|
|
LOG_DIR="log"
|
|
LOG_FILE="${LOG_DIR}/label_tool.log"
|
|
DEFAULT_PORT=8503
|
|
|
|
CONDA_BASE="${CONDA_BASE:-/devl/miniconda3}"
|
|
ENV_UI="${AVOCET_ENV:-cf}"
|
|
ENV_BM="job-seeker-classifiers"
|
|
PYTHON_BM="${CONDA_BASE}/envs/${ENV_BM}/bin/python"
|
|
PYTHON_UI="${CONDA_BASE}/envs/${ENV_UI}/bin/python"
|
|
|
|
# ── Port helpers ──────────────────────────────────────────────────────────────
|
|
|
|
_port_in_use() {
|
|
local port=$1
|
|
# Try lsof first (macOS + most Linux); fall back to ss (systemd Linux)
|
|
if command -v lsof &>/dev/null; then
|
|
lsof -iTCP:"$port" -sTCP:LISTEN -t &>/dev/null
|
|
elif command -v ss &>/dev/null; then
|
|
ss -tlnH 2>/dev/null | awk '{print $4}' | grep -q ":${port}$"
|
|
else
|
|
# Last resort: attempt a connection
|
|
(echo "" >/dev/tcp/127.0.0.1/"$port") 2>/dev/null
|
|
fi
|
|
}
|
|
|
|
_find_free_port() {
|
|
local port=${1:-$DEFAULT_PORT}
|
|
while _port_in_use "$port"; do
|
|
warn "Port ${port} is in use — trying $((port + 1))…"
|
|
((port++))
|
|
done
|
|
echo "$port"
|
|
}
|
|
|
|
# ── PID helpers ───────────────────────────────────────────────────────────────
|
|
|
|
_running_pid() {
|
|
# Returns the PID if a live avocet process is running, empty string otherwise
|
|
if [[ -f "$PID_FILE" ]]; then
|
|
local pid
|
|
pid=$(<"$PID_FILE")
|
|
if kill -0 "$pid" 2>/dev/null; then
|
|
echo "$pid"
|
|
return 0
|
|
else
|
|
rm -f "$PID_FILE" "$PORT_FILE"
|
|
fi
|
|
fi
|
|
echo ""
|
|
}
|
|
|
|
_running_port() {
|
|
[[ -f "$PORT_FILE" ]] && cat "$PORT_FILE" || echo "$DEFAULT_PORT"
|
|
}
|
|
|
|
# ── Usage ─────────────────────────────────────────────────────────────────────
|
|
|
|
usage() {
|
|
echo ""
|
|
echo -e " ${BLUE}Avocet — Email Classifier Training Tool${NC}"
|
|
echo -e " ${YELLOW}Scrape → Store → Process${NC}"
|
|
echo ""
|
|
echo " Usage: ./manage.sh <command> [args]"
|
|
echo ""
|
|
echo " Vue UI + FastAPI:"
|
|
echo -e " ${GREEN}start${NC} Build Vue SPA + start FastAPI on port 8503"
|
|
echo -e " ${GREEN}stop${NC} Stop FastAPI server"
|
|
echo -e " ${GREEN}restart${NC} Stop + rebuild + restart FastAPI server"
|
|
echo -e " ${GREEN}open${NC} Open Vue UI in browser (http://localhost:8503)"
|
|
echo ""
|
|
echo " Benchmark:"
|
|
echo -e " ${GREEN}benchmark [args]${NC} Run benchmark_classifier.py (args passed through)"
|
|
echo -e " ${GREEN}list-models${NC} Shortcut: --list-models"
|
|
echo -e " ${GREEN}score [args]${NC} Shortcut: --score [args]"
|
|
echo -e " ${GREEN}compare [args]${NC} Shortcut: --compare [args]"
|
|
echo ""
|
|
echo " Dev:"
|
|
echo -e " ${GREEN}dev${NC} Hot-reload: uvicorn --reload (:8503) + Vite HMR (:5173)"
|
|
echo -e " ${GREEN}test${NC} Run pytest suite"
|
|
echo ""
|
|
echo " Port defaults to ${DEFAULT_PORT}; auto-increments if occupied."
|
|
echo " Conda envs: UI=${ENV_UI} Benchmark=${ENV_BM}"
|
|
echo ""
|
|
echo " Examples:"
|
|
echo " ./manage.sh start"
|
|
echo " ./manage.sh benchmark --list-models"
|
|
echo " ./manage.sh score --include-slow"
|
|
echo " ./manage.sh compare --limit 30"
|
|
echo ""
|
|
}
|
|
|
|
# ── Commands ──────────────────────────────────────────────────────────────────
|
|
|
|
CMD="${1:-help}"
|
|
shift || true
|
|
|
|
case "$CMD" in
|
|
|
|
start)
|
|
API_PID_FILE=".avocet-api.pid"
|
|
API_PORT=8503
|
|
if [[ -f "$API_PID_FILE" ]] && kill -0 "$(<"$API_PID_FILE")" 2>/dev/null; then
|
|
warn "API already running (PID $(<"$API_PID_FILE")) → http://localhost:${API_PORT}"
|
|
exit 0
|
|
fi
|
|
mkdir -p "$LOG_DIR"
|
|
API_LOG="${LOG_DIR}/api.log"
|
|
info "Building Vue SPA…"
|
|
(cd web && npm run build) >> "$API_LOG" 2>&1
|
|
info "Starting FastAPI on port ${API_PORT}…"
|
|
nohup "$PYTHON_UI" -m uvicorn app.api:app \
|
|
--host 0.0.0.0 --port "$API_PORT" \
|
|
>> "$API_LOG" 2>&1 &
|
|
echo $! > "$API_PID_FILE"
|
|
# Poll until port is actually bound (up to 10 s), not just process alive
|
|
for _i in $(seq 1 20); do
|
|
sleep 0.5
|
|
if (echo "" >/dev/tcp/127.0.0.1/"$API_PORT") 2>/dev/null; then
|
|
success "Avocet started → http://localhost:${API_PORT} (PID $(<"$API_PID_FILE"))"
|
|
break
|
|
fi
|
|
if ! kill -0 "$(<"$API_PID_FILE")" 2>/dev/null; then
|
|
rm -f "$API_PID_FILE"
|
|
error "Server died during startup. Check ${API_LOG}"
|
|
fi
|
|
done
|
|
if ! (echo "" >/dev/tcp/127.0.0.1/"$API_PORT") 2>/dev/null; then
|
|
error "Server did not bind to port ${API_PORT} within 10 s. Check ${API_LOG}"
|
|
fi
|
|
;;
|
|
|
|
stop)
|
|
API_PID_FILE=".avocet-api.pid"
|
|
if [[ ! -f "$API_PID_FILE" ]]; then
|
|
warn "Not running."
|
|
exit 0
|
|
fi
|
|
PID="$(<"$API_PID_FILE")"
|
|
if kill -0 "$PID" 2>/dev/null; then
|
|
kill "$PID" && rm -f "$API_PID_FILE"
|
|
success "Stopped (PID ${PID})."
|
|
else
|
|
warn "Stale PID file (process ${PID} not running). Cleaning up."
|
|
rm -f "$API_PID_FILE"
|
|
fi
|
|
;;
|
|
|
|
restart)
|
|
bash "$0" stop
|
|
exec bash "$0" start
|
|
;;
|
|
|
|
dev)
|
|
API_PORT=8503
|
|
VITE_PORT=5173
|
|
DEV_API_PID_FILE=".avocet-dev-api.pid"
|
|
mkdir -p "$LOG_DIR"
|
|
DEV_API_LOG="${LOG_DIR}/dev-api.log"
|
|
|
|
if [[ -f "$DEV_API_PID_FILE" ]] && kill -0 "$(<"$DEV_API_PID_FILE")" 2>/dev/null; then
|
|
warn "Dev API already running (PID $(<"$DEV_API_PID_FILE"))"
|
|
else
|
|
info "Starting uvicorn with --reload on port ${API_PORT}…"
|
|
nohup "$PYTHON_UI" -m uvicorn app.api:app \
|
|
--host 0.0.0.0 --port "$API_PORT" --reload \
|
|
>> "$DEV_API_LOG" 2>&1 &
|
|
echo $! > "$DEV_API_PID_FILE"
|
|
# Wait for API to bind
|
|
for _i in $(seq 1 20); do
|
|
sleep 0.5
|
|
(echo "" >/dev/tcp/127.0.0.1/"$API_PORT") 2>/dev/null && break
|
|
if ! kill -0 "$(<"$DEV_API_PID_FILE")" 2>/dev/null; then
|
|
rm -f "$DEV_API_PID_FILE"
|
|
error "Dev API died during startup. Check ${DEV_API_LOG}"
|
|
fi
|
|
done
|
|
success "API (hot-reload) → http://localhost:${API_PORT}"
|
|
fi
|
|
|
|
# Kill API on exit (Ctrl+C or Vite exits)
|
|
_cleanup_dev() {
|
|
local pid
|
|
pid=$(<"$DEV_API_PID_FILE" 2>/dev/null || true)
|
|
[[ -n "$pid" ]] && kill "$pid" 2>/dev/null && rm -f "$DEV_API_PID_FILE"
|
|
info "Dev servers stopped."
|
|
}
|
|
trap _cleanup_dev EXIT INT TERM
|
|
|
|
info "Starting Vite HMR on port ${VITE_PORT} (proxy /api → :${API_PORT})…"
|
|
success "Frontend (HMR) → http://localhost:${VITE_PORT}"
|
|
(cd web && npm run dev -- --host 0.0.0.0 --port "$VITE_PORT")
|
|
;;
|
|
|
|
open)
|
|
URL="http://localhost:8503"
|
|
info "Opening ${URL}"
|
|
if command -v xdg-open &>/dev/null; then
|
|
xdg-open "$URL"
|
|
elif command -v open &>/dev/null; then
|
|
open "$URL"
|
|
else
|
|
echo "$URL"
|
|
fi
|
|
;;
|
|
|
|
test)
|
|
info "Running test suite…"
|
|
PYTEST="${CONDA_BASE}/envs/${ENV_UI}/bin/pytest"
|
|
if [[ ! -x "$PYTEST" ]]; then
|
|
error "pytest not found in ${ENV_UI} env at ${PYTEST}"
|
|
fi
|
|
"$PYTEST" tests/ -v "$@"
|
|
;;
|
|
|
|
benchmark)
|
|
info "Running benchmark (${ENV_BM})…"
|
|
if [[ ! -x "$PYTHON_BM" ]]; then
|
|
error "Python not found in ${ENV_BM} env at ${PYTHON_BM}\n" \
|
|
"Create it with: conda env create -f environment.yml"
|
|
fi
|
|
"$PYTHON_BM" scripts/benchmark_classifier.py "$@"
|
|
;;
|
|
|
|
list-models)
|
|
exec "$0" benchmark --list-models
|
|
;;
|
|
|
|
score)
|
|
exec "$0" benchmark --score "$@"
|
|
;;
|
|
|
|
compare)
|
|
exec "$0" benchmark --compare "$@"
|
|
;;
|
|
|
|
help|--help|-h)
|
|
usage
|
|
;;
|
|
|
|
*)
|
|
error "Unknown command: ${CMD}. Run './manage.sh help' for usage."
|
|
;;
|
|
|
|
esac
|