# ✅ Image officielle PyTorch avec Python 3.10, CUDA 11.8 (ou CPU-only)
FROM pytorch/pytorch:2.2.2-cuda11.8-cudnn8-runtime

# Installation des dépendances système supplémentaires
RUN apt-get update && apt-get install -y \
    libgl1-mesa-glx \
    libglib2.0-0 \
    && rm -rf /var/lib/apt/lists/*

# Création de l'utilisateur user (requis par Hugging Face)
RUN useradd -m -u 1000 user
ENV HOME=/home/user
ENV PATH=/home/user/.local/bin:$PATH

# Définir le répertoire de travail
WORKDIR $HOME/app

# Copier les fichiers de requirements avec les bonnes permissions
COPY --chown=user:user requirements.txt .

# Passer à l'utilisateur user
USER user

# Installer les dépendances Python
RUN pip install --no-cache-dir --upgrade pip && \
    pip install --no-cache-dir -r requirements.txt

# Copier le reste de l'application
COPY --chown=user:user . .

# Port requis par Hugging Face Spaces
EXPOSE 7860

# Variables d'environnement pour Hugging Face Spaces
ENV PORT=7860
ENV HOST=0.0.0.0

# Commande de démarrage compatible HF Spaces
CMD ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "7860", "--log-level", "info"]