# HuggingFace Space Dockerfile for FBMC Chronos-2 Zero-Shot Forecasting
# GPU-enabled JupyterLab environment

FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-runtime

# Set working directory
WORKDIR /app

# Install system dependencies
RUN apt-get update && apt-get install -y \
    git \
    curl \
    && rm -rf /var/lib/apt/lists/*

# Copy requirements and install Python dependencies
COPY requirements.txt .
RUN pip install --no-cache-dir -r requirements.txt

# Copy source code and notebooks
COPY src/ ./src/
COPY inference_smoke_test.ipynb .
COPY inference_full_14day.ipynb .
COPY evaluation.ipynb .

# Expose JupyterLab port
EXPOSE 7860

# Set environment variables
ENV JUPYTER_ENABLE_LAB=yes

# Start JupyterLab (HF Spaces expects port 7860)
CMD ["jupyter", "lab", "--ip=0.0.0.0", "--port=7860", "--no-browser", "--allow-root", "--NotebookApp.token=''", "--NotebookApp.password=''"]