FROM python:3.10

WORKDIR /app

# 安装基础依赖
RUN apt-get update && apt-get install -y \
    build-essential \
    curl \
    git \
    ffmpeg \
    libsm6 \
    libxext6 \
    && rm -rf /var/lib/apt/lists/*

# 创建并设置模型和缓存目录
RUN mkdir -p /app/models /tmp/iopaint/cache \
    && chmod -R 777 /app/models /tmp/iopaint

# 设置环境变量
ENV PYTHONUNBUFFERED=1 \
    PYTHONDONTWRITEBYTECODE=1 \
    PIP_NO_CACHE_DIR=1 \
    IOPAINT_MODEL=cv2 \
    IOPAINT_MODEL_DIR=/app/models \
    IOPAINT_DEVICE=cpu \
    TORCH_HOME=/app/models \
    HUGGINGFACE_HUB_CACHE=/app/models \
    IOPAINT_CACHE_DIR=/tmp/iopaint/cache

# 复制项目文件
COPY . .

# 安装必要的Python依赖
RUN pip install --no-cache-dir -e .

# 安装额外的依赖，确保fastapi、uvicorn和loguru已安装
RUN pip install --no-cache-dir fastapi uvicorn loguru opencv-python pillow numpy torch torchvision python-socketio

# 暴露端口
EXPOSE 7860

# 启动API服务
CMD ["python", "-m", "iopaint.api_only"]