#!/usr/bin/env bash
set -euo pipefail
ROOT="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
cd "$ROOT"

export P5_MEMORY_QUILT_FALLBACK_MODEL_PATH=$(ls -d /root/docker-data/hf_cache/hub/models--openbmb--MiniCPM5-1B/snapshots/* | head -n 1)
export P5_MEMORY_QUILT_PRIMARY_MODEL_PATH=$(ls -d /root/docker-data/hf_cache/hub/models--JetBrains--Mellum2-12B-A2.5B-Instruct/snapshots/* 2>/dev/null | head -n 1 || echo "$P5_MEMORY_QUILT_FALLBACK_MODEL_PATH")
export P5_FLUX_MODEL_DIR=$(ls -d /root/docker-data/hf_cache/hub/models--black-forest-labs--FLUX.2-klein-base-4B/snapshots/* | head -n 1)

if [ ! -d "/tmp/p5_venv" ]; then
    python3 -m venv /tmp/p5_venv
    /tmp/p5_venv/bin/pip install -r "$ROOT/requirements.txt"
fi

exec /tmp/p5_venv/bin/python "$ROOT/app.py" "$@"