Update Dockerfile

Dockerfile

@@ -1,29 +1,19 @@
-# Use an official lightweight Ubuntu base image
 FROM ubuntu:22.04
 
-# Set environment variables
-ENV DEBIAN_FRONTEND=noninteractive
-ENV OLLAMA_HOME=/home/user/.ollama
-
-# Create a non-root user to avoid permission issues
-RUN useradd -m -s /bin/bash user
-
 # Install dependencies
-RUN apt update && apt install -y \
-    curl git && \
-    rm -rf /var/lib/apt/lists/*
+RUN apt update && apt install -y curl git python3 python3-pip
 
 # Install Ollama
 RUN curl -fsSL https://ollama.com/install.sh | sh
 
-# Ensure the Ollama home directory is writable
-RUN mkdir -p $OLLAMA_HOME && chown -R user:user $OLLAMA_HOME
+# Pull the specified Llama 3.2 model
+RUN ollama pull llama3.2:3b-instruct-fp16
 
-# Switch to non-root user
-USER user
+# Install Python dependencies
+RUN pip install langchain langchain_ollama
 
 # Expose Ollama API port
 EXPOSE 11434
 
-# Start Ollama server
+# Run Ollama on start
 CMD ["ollama", "serve"]