# 使用官方 Python 环境
FROM python:3.9-slim

# 设置工作目录
WORKDIR /code

# 先复制依赖文件并安装，利用缓存层加速构建
COPY ./requirements.txt /code/requirements.txt
RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt

# 创建缓存目录，防止模型每次重启都重新下载 (HF Space 特性)
# 设置 transformer 缓存路径
ENV TRANSFORMERS_CACHE=/code/.cache
RUN mkdir -p /code/.cache && chmod 777 /code/.cache

# 复制其余代码
COPY . .

# 启动命令 (Host 0.0.0.0, Port 7860)
CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]