Update Dockerfile

Dockerfile

@@ -1,30 +1,36 @@
-# Utiliser l'image officielle Ollama
+# Utiliser l'image officielle Ollama pour le build
 FROM ollama/ollama:latest
 
 # Configuration des variables d'environnement
-# Hugging Face Spaces tourne sur l'utilisateur 1000
 ENV OLLAMA_HOST=0.0.0.0
 ENV OLLAMA_MODELS=/data/models
 ENV HOME=/app
 
 WORKDIR /app
 
-# Création du répertoire de données avec les bonnes permissions
+# Création du répertoire de données avec les bonnes permissions pour HF (User 1000)
 RUN mkdir -p /data/models && chmod -R 777 /data/models
 
-# On réinitialise l'ENTRYPOINT pour éviter que Docker ne passe /start.sh à ollama
+# --- PHASE DE PRE-CHARGEMENT ---
+# On lance temporairement ollama pour télécharger le modèle durant le build
+# Cela évite le timeout de 30min au lancement du Space
+RUN ollama serve & \
+    sleep 5 && \
+    ollama pull mistral-nemo
+
+# Nettoyage des caches inutiles pour réduire la taille de l'image si besoin
+# (Optionnel sur HF car ils ont de l'espace disque de build)
+
+# On réinitialise l'ENTRYPOINT
 ENTRYPOINT []
 
-# Création du script de démarrage directement dans le Dockerfile
+# Script de démarrage simplifié car le modèle est déjà là
 RUN echo '#!/bin/bash \n\
-ollama serve & \n\
-sleep 10 \n\
-echo "📥 Téléchargement du modèle Mistral-Nemo..." \n\
-ollama run mistral-nemo \n\
-wait' > /app/start.sh && chmod +x /app/start.sh
+echo "🚀 Démarrage de Pangea AI Engine..." \n\
+ollama serve' > /app/start.sh && chmod +x /app/start.sh
 
-# Exposer le port par défaut (Hugging Face mappe automatiquement 7860)
+# Port standard HF
 EXPOSE 7860
 
-# Lancer le script via le shell
+# Lancer via bash
 CMD ["/bin/bash", "/app/start.sh"]