feat: optimize Dockerfile for HuggingFace Spaces deployment

Dockerfile

@@ -105,7 +105,8 @@ RUN pip install --no-cache-dir \
     torch \
     tensorflow \
     sentencepiece \
-    tokenizers && \
+    tokenizers \
+    tiktoken && \
     poetry config virtualenvs.create false && \
     poetry install --no-dev --no-interaction --no-ansi
 
@@ -189,7 +190,13 @@ ENV FLASK_APP=app.py \
     APP_WEB_URL=http://127.0.0.1:3000 \
     PYTHONPATH=/app/api \
     PATH="/usr/local/bin:${PATH}" \
-    STORAGE_DIR=/storage
+    STORAGE_DIR=/storage \
+    POSTGRES_HOST=db \
+    POSTGRES_PORT=5432 \
+    POSTGRES_DB=dify \
+    REDIS_HOST=redis \
+    REDIS_PORT=6379 \
+    REDIS_DB=0
 
 # Copy entrypoint script
 COPY docker/entrypoint.sh /app/entrypoint.sh

api/app_factory.py

@@ -37,6 +37,7 @@ from extensions.ext_database import db
 from extensions.ext_login import login_manager
 from libs.passport import PassportService
 from services.account_service import AccountService
+from flask_migrate import Migrate
 
 
 class DifyApp(Flask):
@@ -83,7 +84,7 @@ def initialize_extensions(app):
     ext_compress.init_app(app)
     ext_code_based_extension.init()
     ext_database.init_app(app)
-    ext_migrate.init(app, db)
+    ext_migrate.init(app, ext_database.db)
     ext_redis.init_app(app)
     ext_storage.init_app(app)
     ext_celery.init_app(app)

docker/entrypoint.sh

@@ -3,6 +3,18 @@ set -e
 
 echo "Starting Dify services..."
 
+# Wait for PostgreSQL
+until PGPASSWORD=$POSTGRES_PASSWORD psql -h "$POSTGRES_HOST" -U "$POSTGRES_USER" -d "$POSTGRES_DB" -c '\q'; do
+    echo "PostgreSQL is unavailable - sleeping"
+    sleep 1
+done
+
+# Wait for Redis
+until redis-cli -h "$REDIS_HOST" -p "$REDIS_PORT" ping; do
+    echo "Redis is unavailable - sleeping"
+    sleep 1
+done
+
 # Start API server in background
 cd /app/api
 echo "Starting API server on port 7860..."