Update Dockerfile

Dockerfile

@@ -1,18 +1,22 @@
 # Use the PostgresML image
 FROM ghcr.io/postgresml/postgresml:2.7.12
 
-# Create a non-root user
-RUN useradd -m -u 1000 postgresml
+# Switch to root to perform initial setup
+USER root
 
-# Set working directory
-WORKDIR /home/postgresml/app
+# Create necessary directories with correct permissions
+RUN mkdir -p /var/run/postgresql /var/lib/postgresql \
+    && chown -R postgresml:postgresml /var/run/postgresql /var/lib/postgresql
 
-# Install necessary tools
+# Ensure sudo is available (though it's already installed)
 RUN apt-get update && apt-get install -y sudo
 
-# Set permissions for PostgreSQL directories
-RUN mkdir -p /var/run/postgresql /var/lib/postgresql \
-    && chown -R postgresml:postgresml /var/run/postgresql /var/lib/postgresql
+# Create a user with specific permissions
+RUN useradd -m -u 1000 postgresml \
+    && usermod -aG sudo postgresml
+
+# Set working directory
+WORKDIR /home/postgresml/app
 
 # Set environment variables
 ENV HOME=/home/postgresml \
@@ -21,11 +25,12 @@ ENV HOME=/home/postgresml \
 # Expose ports
 EXPOSE 5432 8000
 
+# Create README.md with root permissions
+RUN echo "# PostgresML Space" > /home/postgresml/README.md \
+    && chown postgresml:postgresml /home/postgresml/README.md
+
 # Switch to non-root user
 USER postgresml
 
-# Create README.md
-RUN echo "# PostgresML Space" > README.md
-
 # Default command
 CMD ["bash", "-c", "psql -d postgresml"]