# Utiliser l'image officielle d'Ollama comme base
FROM ollama/ollama

# Installer Python et pip
RUN apt-get update && apt-get install -y python3 python3-pip

# Copier les fichiers de l'application
COPY app.py /app/app.py
COPY requirements.txt /app/requirements.txt

# Installer les dépendances Python
RUN pip3 install -r /app/requirements.txt

# Exposer le port 7860 pour FastAPI
EXPOSE 7860

# Définir le volume pour les données d'Ollama
VOLUME /root/.ollama
# Installer socat pour rediriger le port
RUN apt-get update && apt-get install -y socat

# Script pour lancer Ollama, télécharger le modèle et démarrer l'application FastAPI
RUN echo '#!/bin/sh\n\
socat TCP-LISTEN:7860,fork TCP:127.0.0.1:11434\n\
ollama serve &\n\
sleep 10\n\
ollama pull llama3\n\
python3 /app/app.py\n\
' > /run-ollama-fastapi.sh && chmod +x /run-ollama-fastapi.sh

# Définir le point d'entrée
ENTRYPOINT ["/bin/sh", "/run-ollama-fastapi.sh"]