FROM python:3.10-slim

# Crée le dossier de travail
WORKDIR /app

# Définir des dossiers de cache Hugging Face accessibles
ENV HF_HOME=/app/.cache/huggingface
ENV TRANSFORMERS_CACHE=/app/.cache/transformers
ENV DIFFUSERS_CACHE=/app/.cache/diffusers

# Créer les dossiers de cache
RUN mkdir -p /app/.cache/huggingface \
    && mkdir -p /app/.cache/transformers \
    && mkdir -p /app/.cache/diffusers

# Installer uniquement les dépendances nécessaires
RUN apt-get update && apt-get install -y \
    build-essential \
    git \
    curl \
    && rm -rf /var/lib/apt/lists/*

# Copier requirements.txt et installer les dépendances Python
COPY requirements.txt .

RUN pip install --upgrade pip \
    && pip install --no-cache-dir -r requirements.txt

# Copier le code source dans l'image
COPY src/ ./src/

# Lancer l'app Streamlit
CMD ["streamlit", "run", "src/streamlit_app.py", "--server.port=8501", "--server.address=0.0.0.0"]