# 使用与原始模板相同的基础镜像
FROM nvidia/cuda:12.5.1-cudnn-devel-ubuntu20.04

# 设置环境变量，避免交互式提示
ENV DEBIAN_FRONTEND=noninteractive \
	TZ=Europe/Paris

# 安装所有必要的工具
RUN rm -f /etc/apt/sources.list.d/*.list && \
    apt-get update && apt-get install -y --no-install-recommends \
    curl ca-certificates git wget procps git-lfs zip unzip htop vim nano bzip2 \
    libx11-6 build-essential libsndfile-dev software-properties-common dos2unix mount file \
 && rm -rf /var/lib/apt/lists/*

# 安装 nvtop 和 nodejs
RUN add-apt-repository ppa:flexiondotorg/nvtop && \
    apt-get upgrade -y && \
    apt-get install -y --no-install-recommends nvtop
RUN curl -sL https://deb.nodesource.com/setup_21.x  | bash - && \
    apt-get install -y nodejs && \
    npm install -g configurable-http-proxy

# 设置 Conda 环境
ENV CONDA_AUTO_UPDATE_CONDA=false \
    PATH=/opt/conda/bin:$PATH
RUN curl -sLo ~/miniconda.sh https://repo.continuum.io/miniconda/Miniconda3-py39_4.10.3-Linux-x86_64.sh \
 && chmod +x ~/miniconda.sh \
 && ~/miniconda.sh -b -p /opt/conda \
 && rm ~/miniconda.sh \
 && conda clean -ya

# 设置 HOME 环境变量和工作目录
ENV HOME=/root
WORKDIR $HOME/app

# 保留您原有的、完整的root执行部分
RUN --mount=target=/root/packages.txt,source=packages.txt \
    apt-get update && \
    xargs -r -a /root/packages.txt apt-get install -y --no-install-recommends \
    && rm -rf /var/lib/apt/lists/*
RUN --mount=target=/root/on_startup.sh,source=on_startup.sh,readwrite \
	bash /root/on_startup.sh
RUN --mount=target=requirements.txt,source=requirements.txt \
    pip install --no-cache-dir --upgrade -r requirements.txt

# --- 终极核心解决方案：重定向所有配置和数据目录 ---

# 1. 在 /data 目录下创建 jupyter 和 ipython 需要的所有子目录
RUN mkdir -p /data/jupyter/config /data/jupyter/data /data/jupyter/runtime /data/ipython

# 2. 通过环境变量，强制 jupyter 和 ipython 使用 /data 下的目录
ENV JUPYTER_CONFIG_DIR=/data/jupyter/config
ENV JUPYTER_DATA_DIR=/data/jupyter/data
ENV JUPYTER_RUNTIME_DIR=/data/jupyter/runtime
ENV IPYTHONDIR=/data/ipython

# --- 解决方案结束 ---

# 复制所有项目文件
COPY . $HOME/app

# 确保脚本可执行
RUN chmod +x start_server.sh

# 设置其他环境变量
ENV PYTHONUNBUFFERED=1 \
	GRADIO_ALLOW_FLAGGING=never \
	GRADIO_NUM_PORTS=1 \
	GRADIO_SERVER_NAME=0.0.0.0 \
	GRADIO_THEME=huggingface \
	SYSTEM=spaces \
	SHELL=/bin/bash

# 使用最稳健的方式启动
CMD ["/bin/bash", "./start_server.sh"]