Update Dockerfile

Dockerfile

@@ -2,20 +2,28 @@
 FROM postgres:latest
 
 # 设置环境变量，用于数据库配置
-ENV POSTGRES_USER=myuser \
-    POSTGRES_PASSWORD=mypassword \
-    POSTGRES_DB=mydatabase \
-    VIRTUAL_ENV=/opt/venv \
-    PATH="$VIRTUAL_ENV/bin:$PATH"
-    
+ENV POSTGRES_USER=n8n
+ENV POSTGRES_PASSWORD=n8n8n
+ENV POSTGRES_DB=n8n_db
+# 设置 n8n 环境变量
 ENV N8N_HOST=0.0.0.0
 ENV N8N_PORT=7860
 ENV N8N_PROTOCOL=https
-ENV WEBHOOK_URL=https://aigenai-db.hf.space/
+ENV WEBHOOK_URL= https://aigenai-db.hf.space/
 ENV GENERIC_TIMEZONE=Asia/Shanghai
 ENV N8N_METRICS=true
 ENV QUEUE_HEALTH_CHECK_ACTIVE=true
 ENV N8N_PAYLOAD_SIZE_MAX=256
+# 设置N8N数据库连接信息
+ENV DB_TYPE=postgresdb
+ENV DB_POSTGRESDB_HOST=localhost
+ENV DB_POSTGRESDB_PORT=5432
+ENV DB_POSTGRESDB_USER=$POSTGRES_USER
+ENV DB_POSTGRESDB_PASSWORD=$POSTGRES_PASSWORD
+ENV DB_POSTGRESDB_DATABASE=$POSTGRES_DB
+# 设置其它环境变量
+ENV VIRTUAL_ENV=/opt/venv
+ENV PATH="$VIRTUAL_ENV/bin:$PATH"
 
 # 将应用程序代码和启动脚本复制到容器中
 COPY run.sh /app/run.sh
@@ -40,7 +48,7 @@ RUN apt-get update && apt-get install -y curl unzip gnupg build-essential sudo v
 # 创建虚拟环境并安装 Python 包
     python3 -m venv $VIRTUAL_ENV && \
     $VIRTUAL_ENV/bin/pip install --upgrade pip && \
-    $VIRTUAL_ENV/bin/pip install psycopg2-binary && \
+    $VIRTUAL_ENV/bin/pip install psycopg2-binary yt-dlp && \
 
 # 更改现有的 postgres 用户 UID 和 GID 为 1000
     usermod -u 1000 postgres && groupmod -g 1000 postgres && \
@@ -65,6 +73,6 @@ WORKDIR /app
 # 启动容器时执行run.sh脚本
 CMD ["./run.sh"]
 
-# 设置健康检查以确保 Flask 应用正常运行
-HEALTHCHECK --interval=30s --timeout=10s --start-period=10s --retries=3 \
+# 设置健康检查以确保应用正常运行
+HEALTHCHECK --interval=120s --timeout=10s --start-period=10s --retries=3 \
     CMD curl -f http://localhost:7860/ || exit 1