FROM python:3.9-slim

# Install curl and other dependencies
RUN apt-get update && apt-get install -y curl && apt-get clean && rm -rf /var/lib/apt/lists/*

# Install Ollama
RUN curl -fsSL https://ollama.ai/install.sh | sh

# Set up user and working directory
RUN useradd -m -u 1000 user
USER user
ENV HOME=/home/user PATH="/home/user/.local/bin:$PATH"
WORKDIR $HOME/app

# Copy and install Python dependencies
COPY --chown=user requirements.txt .
RUN pip install --no-cache-dir --upgrade -r requirements.txt

# Copy the app code
COPY --chown=user . .

# Expose Streamlit port (8501 is default for HF Spaces Streamlit)
EXPOSE 8501

# Command to run Ollama and Streamlit
CMD ["sh", "-c", "ollama serve & sleep 5 && ollama pull llama3.2 && streamlit run app.py --server.port 8501 --server.address 0.0.0.0"]