FROM python:3.10-slim

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

# Install Python packages
RUN pip install --no-cache-dir torch transformers flask kernels accelerate huggingface_hub

# Create cache dir for Hugging Face models
RUN mkdir -p /app/model_cache
ENV HF_HOME=/app/model_cache

# Download GPT-OSS-20B at build time
RUN python -c "from huggingface_hub import snapshot_download; snapshot_download('openai/gpt-oss-20b', cache_dir='/app/model_cache', resume_download=True)"

# Copy app code
WORKDIR /app
COPY app.py /app
RUN chmod -R 777 /app

# Expose port and start app
EXPOSE 7860
CMD ["python", "app.py"]