Update Dockerfile

Dockerfile

@@ -28,7 +28,11 @@ RUN apt-get update && apt-get install -y \
 FROM langgenius/dify-web:latest AS web
 FROM langgenius/dify-api:latest AS api
 
-# Final stage
+# After base image setup and before final stage (around line 27)
+# Add PostgreSQL stage
+FROM postgres:15-alpine AS postgres
+
+# Final stage (continuing from line 32)
 FROM base
 
 # Create non-root user
@@ -84,15 +88,32 @@ ENV FLASK_APP=app.py \
 USER user
 EXPOSE 7860 3000
 
+# Install PostgreSQL
+RUN apt-get update && apt-get install -y postgresql postgresql-contrib && \
+    rm -rf /var/lib/apt/lists/* && \
+    mkdir -p /var/lib/postgresql/data && \
+    chown -R postgres:postgres /var/lib/postgresql
+
+# Initialize PostgreSQL database
+USER postgres
+RUN /usr/lib/postgresql/15/bin/initdb -D /var/lib/postgresql/data && \
+    echo "host all  all    0.0.0.0/0  md5" >> /var/lib/postgresql/data/pg_hba.conf && \
+    echo "listen_addresses='*'" >> /var/lib/postgresql/data/postgresql.conf
+
+USER user
+
 # Create startup script with connection retries
 RUN echo '#!/bin/bash\n\
 echo "===== Application Startup at $(date "+%Y-%m-%d %H:%M:%S") ====="\n\
 \n\
-# Wait for database\n\
+# Start PostgreSQL\n\
+sudo -u postgres /usr/lib/postgresql/15/bin/pg_ctl -D /var/lib/postgresql/data start\n\
+\n\
+# Wait for PostgreSQL to start\n\
 max_tries=30\n\
 count=0\n\
 echo "Checking database connection..."\n\
-until PGPASSWORD=$DB_PASSWORD psql -h $DB_HOST -p $DB_PORT -U $DB_USERNAME -d $DB_DATABASE -c "SELECT 1" > /dev/null 2>&1; do\n\
+until PGPASSWORD=$DB_PASSWORD psql -h localhost -p 5432 -U postgres -c "SELECT 1" > /dev/null 2>&1; do\n\
     echo "Waiting for database connection..."\n\
     sleep 2\n\
     count=$((count+1))\n\
@@ -102,6 +123,11 @@ until PGPASSWORD=$DB_PASSWORD psql -h $DB_HOST -p $DB_PORT -U $DB_USERNAME -d $D
     fi\n\
 done\n\
 \n\
+# Create database and user if they dont exist\n\
+PGPASSWORD=$DB_PASSWORD psql -h localhost -U postgres -c "CREATE DATABASE $DB_DATABASE;" || true\n\
+PGPASSWORD=$DB_PASSWORD psql -h localhost -U postgres -c "CREATE USER $DB_USERNAME WITH PASSWORD '\''$DB_PASSWORD'\'';" || true\n\
+PGPASSWORD=$DB_PASSWORD psql -h localhost -U postgres -c "GRANT ALL PRIVILEGES ON DATABASE $DB_DATABASE TO $DB_USERNAME;" || true\n\
+\n\
 echo "Database connection successful"\n\
 \n\
 # Run database migrations\n\
@@ -123,4 +149,4 @@ chmod +x /app/entrypoint.sh
 
 WORKDIR /app
 
-CMD ["./entrypoint.sh"]
+CMD ["./entrypoint.sh"]