FROM pytorch/pytorch:2.1.0-cuda12.1-cudnn8-runtime

ENV TZ=Asia/Shanghai

WORKDIR /app

COPY . /app

RUN pip install --no-cache-dir -r requirements.txt
RUN pip install --no-cache-dir opencv-python-headless -i https://pypi.org/simple

# Create required directories and config file with write permissions
RUN mkdir -p /app/presets /app/logs /app/repositories/Fooocus/models/sam /app/repositories/Fooocus/outputs && \
    touch /app/config.txt && \
    chmod -R 775 /app/presets /app/logs /app/repositories/Fooocus/models/sam /app/repositories/Fooocus/outputs /app/config.txt

EXPOSE 8888

CMD ["python", "main.py", "--host", "0.0.0.0", "--port", "8888", "--skip-pip"]