Update Dockerfile

Dockerfile

@@ -9,10 +9,11 @@ ENV USER=coder
 RUN useradd -m -d $HOME -s /bin/bash $USER && \
     echo "$USER ALL=(ALL) NOPASSWD:ALL" >> /etc/sudoers
 
-# Install dependencies and code-server
+# Install dependencies, code-server, and ollama
 RUN apt-get update && \
-    apt-get install -y curl wget gpg apt-transport-https git nodejs npm python3 python3-pip && \
+    apt-get install -y curl wget gpg apt-transport-https sudo git nodejs npm python3 python3-pip && \
     curl -fsSL https://code-server.dev/install.sh | sh && \
+    curl -fsSL https://ollama.com/install.sh | sh && \
     apt-get clean && \
     rm -rf /var/lib/apt/lists/*
 
@@ -34,8 +35,22 @@ RUN code-server --install-extension ms-python.python && \
 # Create a workspace directory
 RUN mkdir -p $HOME/workspace
 
-# Expose the port
-EXPOSE 7860
-
-# Set command to start code-server
-CMD ["code-server", "--disable-telemetry", "--bind-addr", "0.0.0.0:7860", "/home/coder/workspace"]
+# Create a startup script to run both services
+USER root
+RUN echo '#!/bin/bash\n\
+# Start ollama in the background\n\
+ollama serve &\n\
+\n\
+# Give ollama a moment to start\n\
+sleep 2\n\
+\n\
+# Start code-server in the foreground\n\
+exec sudo -u $USER code-server --disable-telemetry --bind-addr 0.0.0.0:7860 /home/coder/workspace\n\
+' > /start.sh && \
+    chmod +x /start.sh
+
+# Expose ports for both services
+EXPOSE 7860 11434
+
+# Start both services
+CMD ["/start.sh"]