# 定义基础镜像
# 使用阿里魔塔镜像
FROM modelscope-registry.cn-hangzhou.cr.aliyuncs.com/modelscope-repo/modelscope:ubuntu22.04-cuda12.1.0-py310-torch2.3.0-tf2.16.1-1.17.0


# 设置工作目录
WORKDIR /app

# 复制要求文件到容器中
COPY . /app

# 安装依赖包 国内走清华源加速依赖包下载
RUN pip install -r requirement/requirements-gpu.txt -i https://pypi.tuna.tsinghua.edu.cn/simple/

# 安装 FFmpeg
# 注意：这里假设 ffmpeg-release-amd64-static.tar.xz 文件已经在当前目录中
COPY  /ffmpeg/ffmpeg-release-amd64-static.tar.xz /tmp/
RUN tar -xJf /tmp/ffmpeg-release-amd64-static.tar.xz -C /tmp && \
    mv /tmp/ffmpeg-*-amd64-static/ffmpeg /usr/local/bin/ && \
    mv /tmp/ffmpeg-*-amd64-static/ffprobe /usr/local/bin/ && \
    chmod +x /usr/local/bin/ffmpeg && \
    chmod +x /usr/local/bin/ffprobe && \
    rm -rf /tmp/ffmpeg-*-amd64-static /tmp/ffmpeg-release-amd64-static.tar.xz

# 设置 FFmpeg 环境变量
ENV PATH="/usr/local/bin:${PATH}"

# 暴露端口（如果需要）
EXPOSE 7860

# 运行命令，启动应用
CMD ["streamlit", "run", "--server.port=7860", "app.py"]