FROM python:3.10-slim

ENV TZ=Asia/Shanghai \
    OUTPUT_DIR=/opt/data/output \
    IMAGES_DIR=/opt/data/images \
    MODELS_PATH=/opt/data/models \
    DEEPFACE_HOME=/opt/data/models \
    FAISS_INDEX_DIR=/opt/data/faiss \
    CELEBRITY_SOURCE_DIR=/opt/data/chinese_celeb_dataset \
    GENDER_CONFIDENCE=1 \
    UPSCALE_SIZE=2 \
    AGE_CONFIDENCE=0.1 \
    DRAW_SCORE=true \
    FACE_CONFIDENCE=0.7 \
    ENABLE_DDCOLOR=true \
    ENABLE_GFPGAN=true \
    ENABLE_REALESRGAN=true \
    ENABLE_ANIME_STYLE=true \
    ENABLE_RVM=true \
    ENABLE_REMBG=true \
    ENABLE_CLIP=false \
    CLEANUP_INTERVAL_HOURS=1 \
    CLEANUP_AGE_HOURS=1 \
    BEAUTY_ADJUST_GAMMA=0.8 \
    BEAUTY_ADJUST_MIN=1.0 \
    BEAUTY_ADJUST_MAX=9.0 \
    ENABLE_ANIME_PRELOAD=false \
    ENABLE_LOGGING=true \
    BEAUTY_ADJUST_ENABLED=true \
    RVM_LOCAL_REPO=/opt/data/models/RobustVideoMatting \
    RVM_WEIGHTS_PATH=/opt/data/models/torch/hub/checkpoints/rvm_resnet50.pth \
    RVM_MODEL=resnet50 \
    AUTO_INIT_GFPGAN=false \
    AUTO_INIT_DDCOLOR=false \
    AUTO_INIT_REALESRGAN=false \
    AUTO_INIT_ANIME_STYLE=false \
    AUTO_INIT_CLIP=false \
    AUTO_INIT_RVM=false \
    AUTO_INIT_REMBG=false \
    ENABLE_WARMUP=true \
    REALESRGAN_MODEL=realesr-general-x4v3 \
    CELEBRITY_FIND_THRESHOLD=0.87 \
    FEMALE_AGE_ADJUSTMENT=4 \
    HOSTNAME=HG

RUN mkdir -p  /opt/data/chinese_celeb_dataset /opt/data/faiss /opt/data/models /opt/data/images /opt/data/output
WORKDIR /app
COPY requirements.txt .
COPY *.py /app/
COPY facelist-web /app/facelist-web

# 安装必要的系统工具和依赖
RUN apt-get update && apt-get install -y --no-install-recommends \
    build-essential \
    cmake \
    git \
    wget \
    curl \
    ca-certificates \
    libopenblas-dev \
    liblapack-dev \
    libx11-dev \
    libgtk-3-dev \
    libboost-python-dev \
    libglib2.0-0 \
    libsm6 \
    libxext6 \
    libxrender-dev \
    libgomp1 \
    && rm -rf /var/lib/apt/lists/*

RUN pip install --upgrade pip
# 安装所有依赖 - 现在可以一次性完成
RUN pip install --no-cache-dir -r requirements.txt
EXPOSE 7860
CMD ["uvicorn", "app:app", "--workers", "1", "--loop", "asyncio", "--http", "httptools", "--host", "0.0.0.0", "--port", "7860", "--timeout-keep-alive", "600"]