FROM python:3.10-slim

RUN pip install --no-cache-dir \
    torch==2.5.1 torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121 &&         pip install --no-cache-dir \
    transformers==4.57.6 peft==0.18.1 datasets==3.6.0         huggingface_hub==0.36.2 accelerate==1.10.1 scipy numpy==1.26.4

WORKDIR /app
COPY socratic_routing_eval.py .

# Model repos  (set via Space secrets / env vars)
ENV HF_TOKEN=""
ENV MATH_REPO="dexmac/progressive-cognitive-dream-lora-en"
ENV LOGIC_REPO="dexmac/progressive-cognitive-logic-specialist-en"
ENV MONO_REPO="dexmac/progressive-cognitive-logic-dream-lora-en"
ENV ROUTER_REPO="dexmac/progressive-cognitive-router-en"
ENV RESULTS_REPO="dexmac/progressive-cognitive-results"
ENV BASE_MODEL_7B="Qwen/Qwen2.5-7B"
ENV INCLUDE_7B="0"
ENV SPACE_ID="dexmac/progressive-cognitive-socratic-eval"

EXPOSE 7860
CMD ["sh", "-lc", "python -m http.server 7860 & server_pid=$!; REFERENCE_BASE_MODEL='Qwen/Qwen2.5-3B' REFERENCE_BASE_NAME='qwen_3b_base' python socratic_routing_eval.py --seeds 42 --n 20 --only-configs qwen_3b_base --run-tag 3b_focus; wait $server_pid"]