ARG USERNAME
ARG PASSWORD
ARG OPENAI_API_KEY


FROM python:3.12

WORKDIR /app

# 设置argostranslate专用缓存路径
ENV ARGOS_TRANSLATE_PACKAGES_DIR=/app/.cache/argos-translate
ENV ARGOS_TRANSLATE_DATA_DIR=/app/.cache/argos-translate
ENV PYTHONUNBUFFERED=1

RUN apt-get update && apt-get install -y libgl1 git \
    && rm -rf /var/lib/apt/lists/*

# 创建缓存目录并设置权限
RUN mkdir -p ${ARGOS_TRANSLATE_DATA_DIR} && \
    chmod 777 ${ARGOS_TRANSLATE_DATA_DIR}

# 添加非root用户
RUN adduser --disabled-password --gecos "" appuser && \
    chown -R appuser:appuser /app

USER appuser
ENV PATH="/home/appuser/.local/bin:${PATH}"

WORKDIR /github
# RUN git clone -b fix-gradio-ui https://github.com/tylzh97/PDFMathTranslate.git PDFMathTranslate && \
#     cd PDFMathTranslate && \
#     pip install --user . && \ 
#     sleep 0

RUN git clone https://github.com/Byaidu/PDFMathTranslate.git PDFMathTranslate && \
    cd PDFMathTranslate && \
    git checkout v1.9.9 && \
    pip install --user . && \ 
    sleep 0

WORKDIR /app
COPY . .

USER root
RUN mkdir -p /home/appuser/.cache/babeldoc/models/ && \
    chmod -R 777 /home/appuser/.cache/ && \
    mv models/doclayout_yolo_docstructbench_imgsz1024.onnx /home/appuser/.cache/babeldoc/models/

# ENV HF_ENDPOINT=https://hf-mirror.com
# ENV HF_HUB_OFFLINE=1

USER appuser
EXPOSE 7860

CMD ["/bin/bash", "start.sh"]