FROM python:3.11

RUN apt update 
RUN apt install -y git fontconfig
RUN git clone https://github.com/yangjianchuan/gpt_academic.git
WORKDIR gpt_academic
RUN pip install -r requirements.txt
RUN pip install torch
RUN pip install --upgrade nougat-ocr tiktoken
RUN pip install zh_langchain
ENV AUTO_CLEAR_TXT=True
ENV ADD_WAIFU=True
ENV LLM_MODEL="gpt-3.5-turbo"
ENV AVAIL_LLM_MODELS='["gpt-3.5-turbo"]'
ENV DARK_MODE=False
ENV WEB_PORT=7860
ENV MPLCONFIGDIR=/tmp/matplotlib
ENV TRANSFORMERS_CACHE=/tmp/transformers
ENV FONTCONFIG_PATH=/etc/fonts

# 创建这些目录并给予写权限
RUN mkdir -p $MPLCONFIGDIR $TRANSFORMERS_CACHE $FONTCONFIG_PATH/fonts/cache gpt_log private_upload __pycache__ && \
    chmod -R a+rw $MPLCONFIGDIR $TRANSFORMERS_CACHE $FONTCONFIG_PATH/fonts/cache gpt_log private_upload __pycache__

EXPOSE 7860
CMD ["python", "main.py", "--host", "0.0.0.0:7860"]