# ベースイメージ: CUDA対応のPyTorch公式イメージ（HF SpacesのT4/GPUと互換）
FROM pytorch/pytorch:2.1.0-cuda11.8-cudnn8-runtime

# 作業ディレクトリ
WORKDIR /app

# Python依存ファイルをコピー
COPY requirements.txt ./

# 依存ライブラリのインストール（Gradio、FastAPI、HuggingFace関連など）
RUN apt-get update && apt-get install -y \
    git wget libgl1-mesa-glx libglib2.0-0 \
    && pip install --upgrade pip \
    && pip install --no-cache-dir -r requirements.txt

# アプリケーションのファイル一式をコピー
COPY . .

# Hugging Face CLI認証用（必要に応じて）
ENV HF_HUB_DISABLE_SYMLINKS_WARNING=1
ENV TRANSFORMERS_CACHE=/app/cache
ENV HF_HOME=/app/cache

# ポート（Spaces UI用: 7860 / FastAPI用: 8000 どちらも対応）
EXPOSE 7860
EXPOSE 8000

# 実行コマンド（Gradio + FastAPI対応のapp.py）
CMD ["python", "app.py"]