FROM continuumio/miniconda3

RUN apt-get update -y 
RUN apt-get install nano unzip curl -y

# Création de l'utilisateur pour Hugging Face
RUN useradd -m -u 1000 user
USER user

# Configuration des variables d'environnement
ENV HOME=/home/user \
    PATH=/home/user/.local/bin:$PATH

# Définition du répertoire de travail
WORKDIR $HOME/app

# Installation des dépendances de base
RUN pip install --no-cache-dir \
    fastapi>=0.68.0 \
    uvicorn>=0.15.0 \
    numpy>=1.21.0 \
    scikit-learn>=0.24.2 \
    xgboost>=1.4.2 \
    joblib>=1.0.1 \
    pydantic>=1.8.2 \
    python-multipart>=0.0.5 \
    pandas>=1.3.0 \
    ta>=0.7.0

# Création des répertoires nécessaires
RUN mkdir -p $HOME/app/models

# Copie des fichiers avec les bonnes permissions
COPY --chown=user requirements.txt $HOME/app/
RUN pip install --no-cache-dir -r requirements.txt

# Copie des fichiers Python
COPY --chown=user ./train_strategy_models.py $HOME/app/
COPY --chown=user ./app.py $HOME/app/

# Création du répertoire models s'il n'existe pas déjà
RUN mkdir -p $HOME/app/models

# Variable d'environnement pour le port
ENV PORT=7860

# Commande pour démarrer l'application
CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860", "--proxy-headers"]