# ベースイメージとしてCUDA対応のUbuntuを使用
FROM nvidia/cuda:12.2.2-base-ubuntu20.04

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

# 必要なパッケージをインストール
RUN apt-get update && apt-get install -y --no-install-recommends \
    python3-pip \
    python3-dev \
    && apt-get clean \
    && rm -rf /var/lib/apt/lists/*

# Pythonのシンボリックリンクを作成
RUN ln -s /usr/bin/python3 /usr/bin/python

# 必要なPythonライブラリをインストールするための依存ファイルをコピー
COPY requirements.txt /app/requirements.txt

# 必要なPythonパッケージをインストール
RUN pip install --no-cache-dir -r requirements.txt

# アプリケーションコードをコンテナにコピー
COPY . /app

# ポート設定（Gradioのデフォルトポート7860）
EXPOSE 7860

# アプリケーションを起動
CMD ["python", "app.py"]