Update Dockerfile

Dockerfile

@@ -1,25 +1,31 @@
+# Use the PostgresML image
 FROM ghcr.io/postgresml/postgresml:2.7.12
-USER root
-# Set working directory
-WORKDIR /app
 
-# Expose necessary ports
-EXPOSE 5432 8000
+# Create a non-root user
+RUN useradd -m -u 1000 postgresml
 
-# Create volume for PostgreSQL data
-VOLUME ["/var/lib/postgresql"]
+# Set working directory
+WORKDIR /home/postgresml/app
 
 # Install necessary tools
 RUN apt-get update && apt-get install -y sudo
 
-# Create a non-root user
-RUN useradd -m -u 1000 postgresml
-
-# Set permissions
-RUN chown -R postgresml:postgresml /var/lib/postgresql
+# Set permissions for PostgreSQL directories
+RUN mkdir -p /var/run/postgresql /var/lib/postgresql \
+    && chown -R postgresml:postgresml /var/run/postgresql /var/lib/postgresql
 
 # Set environment variables
-ENV HOME=/home/postgresml
+ENV HOME=/home/postgresml \
+    PATH=/home/postgresml/.local/bin:$PATH
+
+# Expose ports
+EXPOSE 5432 8000
+
+# Switch to non-root user
+USER postgresml
+
+# Create README.md
+RUN echo "# PostgresML Space" > README.md
 
 # Default command
-CMD ["sudo", "-u", "postgresml", "psql", "-d", "postgresml"]
+CMD ["bash", "-c", "psql -d postgresml"]