FROM python:3.11-slim
# v6 force rebuild - 简化依赖安装

WORKDIR /app

# 安装依赖 (v6 - 分步安装避免超时)
RUN pip install --no-cache-dir fastapi==0.104.1 uvicorn==0.24.0 numpy==1.24.4
RUN pip install --no-cache-dir torch --index-url https://download.pytorch.org/whl/cpu
RUN pip install --no-cache-dir huggingface_hub==0.20.1

# 复制源代码
COPY src/ ./src/
COPY data/ ./data/
COPY models/ ./models/
COPY train_200m.py .
COPY app.py .

# 创建目录
RUN mkdir -p /app/checkpoints /app/models /app/tools

ENV PYTHONPATH=/app
ENV PYTHONUNBUFFERED=1
ENV SWARM_PRODUCT_ROOT=/app
ENV HF_MODEL_REPO=lk080424/swarm-weights
ENV ENABLE_TRAINING=1

EXPOSE 7860

CMD ["python", "app.py"]