Update Dockerfile

Dockerfile

@@ -2,55 +2,49 @@ FROM ubuntu:22.04
 
 # Set environment variables
 ENV DEBIAN_FRONTEND=noninteractive
-ENV HOME=/home/coder
-ENV USER=coder
 
-# Create a non-root user
-RUN useradd -m -d $HOME -s /bin/bash $USER && \
-    echo "$USER ALL=(ALL) NOPASSWD:ALL" >> /etc/sudoers
-
-# Install dependencies, code-server, and ollama
+# Install dependencies, code-server, and ollama without sudo in the initial list
 RUN apt-get update && \
-    apt-get install -y curl wget gpg apt-transport-https sudo git nodejs npm python3 python3-pip && \
+    apt-get install -y curl wget gpg apt-transport-https 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/*
 
-# Switch to the non-root user
-USER $USER
-WORKDIR $HOME
+# Create a directory for the workspace
+RUN mkdir -p /workspace
 
 # Create configuration directory
-RUN mkdir -p $HOME/.config/code-server
+RUN mkdir -p /root/.config/code-server
 
 # Configure code-server to run on port 7860 (Hugging Face Spaces default)
-RUN echo "bind-addr: 0.0.0.0:7860\nauth: none\ncert: false" > $HOME/.config/code-server/config.yaml
+RUN echo "bind-addr: 0.0.0.0:7860\nauth: none\ncert: false" > /root/.config/code-server/config.yaml
+
+# Install ollama after code-server is set up
+RUN curl -fsSL https://ollama.com/install.sh | sh || true
 
 # Install some useful VS Code extensions
 RUN code-server --install-extension ms-python.python && \
     code-server --install-extension ritwickdey.LiveServer && \
     code-server --install-extension ms-toolsai.jupyter
 
-# Create a workspace directory
-RUN mkdir -p $HOME/workspace
-
-# Create a startup script to run both services
-USER root
+# Create a startup script
 RUN echo '#!/bin/bash\n\
 # Start ollama in the background\n\
-ollama serve &\n\
+/usr/local/bin/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\
+exec code-server --disable-telemetry --bind-addr 0.0.0.0:7860 /workspace\n\
 ' > /start.sh && \
     chmod +x /start.sh
 
 # Expose ports for both services
 EXPOSE 7860 11434
 
+# Set the workspace as working directory
+WORKDIR /workspace
+
 # Start both services
 CMD ["/start.sh"]