FROM pytorch/pytorch:2.4.0-cuda12.1-cudnn9-runtime

WORKDIR /app

# Install dependencies
COPY requirements.txt .
RUN pip install --no-cache-dir -r requirements.txt --extra-index-url https://download.pytorch.org/whl/cu121

# Copy app files
COPY app.py .
COPY 3d_scene.html .
COPY three.min.js .

ENV PORT=7860
EXPOSE 7860

CMD ["python", "app.py"]