Update Dockerfile

Dockerfile

@@ -1,37 +1,55 @@
 # Use an official Python runtime as a parent image
 FROM python:3.10-slim
 
+# Set environment variables
+ENV PYTHONUNBUFFERED=1
+ENV HOME=/home/appuser
+ENV APP_HOME=/home/appuser/app
+ENV PATH="/home/appuser/.local/bin:${PATH}"
+
 # Install system dependencies
 RUN apt-get update && apt-get install -y \
     build-essential \
+    cmake \
+    git \
+    wget \
     curl \
     && rm -rf /var/lib/apt/lists/*
 
-# Install Ollama
-RUN curl -fsSL https://ollama.ai/install.sh | sh
+# Create a non-root user
+RUN useradd -m -u 1000 appuser && \
+    mkdir -p $APP_HOME && \
+    chown -R appuser:appuser /home/appuser
+
+# Switch to the non-root user
+USER appuser
+WORKDIR $APP_HOME
 
-# Set working directory
-WORKDIR /app
+# Copy the application files
+COPY --chown=appuser:appuser . $APP_HOME/
 
-# Copy the current directory contents into the container
-COPY . .
+# Download the model
+RUN mkdir -p $APP_HOME/models && \
+    wget -O $APP_HOME/models/llama-model.gguf https://huggingface.co/bartowski/Meta-Llama-3.1-8B-Instruct-GGUF/resolve/main/Meta-Llama-3.1-8B-Instruct-IQ2_M.gguf
 
 # Install Python dependencies
-RUN pip install --no-cache-dir \
+RUN pip install --user --no-cache-dir \
     chainlit \
     langchain \
-    langchain_community \
-    langchain_core \
     chromadb \
-    pydantic \
-    typing-extensions
+    sentence-transformers \
+    llama-cpp-python \
+    --extra-index-url https://download.pytorch.org/whl/cpu
 
-# Make port 8000 available to the world outside this container
-EXPOSE 8000
+# Create necessary directories with correct permissions
+RUN mkdir -p $APP_HOME/.chainlit/files && \
+    mkdir -p $APP_HOME/mydb
 
-# Create a script to start both Ollama and the Chainlit app
-RUN echo '#!/bin/bash\nollama serve &\nsleep 5\nollama pull llama3.2\nollama pull llama3.2\nchainlit run app.py --host 0.0.0.0 --port 8000' > start.sh
-RUN chmod +x start.sh
+# Create a chainlit.md file
+RUN echo "# Welcome to RAG Chainlit Application! 👋\n\nThis is a Retrieval-Augmented Generation application using Llama.cpp." > $APP_HOME/chainlit.md
+
+# Expose port 8000
+EXPOSE 8000
 
-# Run the script when the container launches
-CMD ["./start.sh"]
+# Start the Chainlit app
+CMD ["chainlit", "run", "app.py", "--host", "0.0.0.0", "--port", "8000"]