FROM ollama/ollama:latest

# Instala curl e outros pacotes necessários
RUN apt-get update && apt-get install -y curl

# Configura variáveis de ambiente
ENV OLLAMA_HOST=0.0.0.0 \
    OLLAMA_ORIGINS=https://*.hf.space

# Define o diretório de trabalho
WORKDIR /app

# Copia o Modelfile para o diretório de trabalho
COPY Modelfile /app/

# Exponha a porta usada pelo serviço Ollama
EXPOSE 11434

# Cria um script de inicialização
RUN echo '#!/bin/sh' > start.sh && \
    echo 'ollama serve &' >> start.sh && \
    echo 'sleep 10' >> start.sh && \
    echo 'ollama pull llama3.1' >> start.sh && \
    echo 'ollama create mymodel -f Modelfile' >> start.sh && \
    echo 'tail -f /dev/null' >> start.sh && \
    chmod +x start.sh

# Define o comando de inicialização
CMD ["./start.sh"]