FROM python:3.11-slim

WORKDIR /app

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

# Clone repo (includes env, graphforge, training, pyproject.toml)
RUN git clone --depth 1 https://github.com/nithin062006/scaler.git .

# CPU-only torch (keeps image small; no GPU needed for 0.5B demo)
RUN pip install --no-cache-dir torch --index-url https://download.pytorch.org/whl/cpu

# ML stack + Gradio
RUN pip install --no-cache-dir \
    "transformers>=4.40" \
    "peft>=0.10" \
    "huggingface_hub>=0.22" \
    "gradio>=4.36" \
    "datasets>=2.18" \
    "pydantic>=2.6" \
    "pyyaml>=6.0" \
    gitpython

# Install project
RUN pip install --no-cache-dir -e "." || true

ENV PYTHONUNBUFFERED=1
ENV PYTHONPATH=/app

EXPOSE 7860

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