FROM python:3.11-slim

WORKDIR /app

# 依存関係のインストール
COPY requirements.txt .
RUN pip install --no-cache-dir -r requirements.txt

# モデルの事前ダウンロード（ビルド時にキャッシュ）
RUN python -c "from transformers import GPT2LMHeadModel, GPT2Tokenizer; \
    GPT2LMHeadModel.from_pretrained('gpt2'); \
    GPT2Tokenizer.from_pretrained('gpt2')"

# アプリケーションコードをコピー
COPY app.py .
COPY src/ ./src/

EXPOSE 8501

CMD ["streamlit", "run", "app.py", "--server.headless", "true", "--server.address", "0.0.0.0"]