Update Dockerfile

Dockerfile

@@ -1,71 +1,76 @@
-# Multi-stage build for optimization
-FROM python:3.9-slim as builder
-
-# Install system dependencies for building
-RUN apt-get update && apt-get install -y \
-    build-essential \
-    gcc \
-    g++ \
-    && rm -rf /var/lib/apt/lists/*
-
-# Create virtual environment
-RUN python -m venv /opt/venv
-ENV PATH="/opt/venv/bin:$PATH"
-
-# Copy and install requirements
-COPY requirements.txt /tmp/requirements.txt
-COPY webui/requirements.txt /tmp/webui_requirements.txt
-
-# Install main project dependencies
-RUN pip install --no-cache-dir --upgrade pip && \
-    pip install --no-cache-dir -r /tmp/requirements.txt
-
-# Install webui dependencies
-RUN pip install --no-cache-dir -r /tmp/webui_requirements.txt
-
-# Install production WSGI server
-RUN pip install --no-cache-dir gunicorn
-
-# Production stage
-FROM python:3.9-slim
-
-# Install runtime dependencies
-RUN apt-get update && apt-get install -y \
-    curl \
-    && rm -rf /var/lib/apt/lists/*
-
-# Create non-root user
-RUN useradd -m -u 1000 user && \
-    mkdir -p /app && \
-    chown -R user:user /app
-
-# Copy virtual environment from builder stage
-COPY --from=builder /opt/venv /opt/venv
-ENV PATH="/opt/venv/bin:$PATH"
-
-# Set working directory
-WORKDIR /app
-
-# Copy application code
-COPY --chown=user:user . /app
-
-# Make startup script executable
-RUN chmod +x /app/webui/docker_start.sh
-
-# Switch to non-root user
-USER user
-
-# Expose the correct port (Flask app runs on 7070)
-EXPOSE 7860
-
-# Add health check
-HEALTHCHECK --interval=30s --timeout=10s --start-period=5s --retries=3 \
-    CMD curl -f http://localhost:7860/ || exit 1
-
-# Set environment variables
-ENV PYTHONPATH=/app
-ENV FLASK_APP=webui/app.py
-ENV FLASK_ENV=production
-
-# Use the startup script
-CMD ["/app/webui/docker_start.sh"]
+# Multi-stage build for optimization
+FROM python:3.9-slim as builder
+
+# Install system dependencies for building
+RUN apt-get update && apt-get install -y \
+    build-essential \
+    gcc \
+    g++ \
+    && rm -rf /var/lib/apt/lists/*
+
+# Create virtual environment
+RUN python -m venv /opt/venv
+ENV PATH="/opt/venv/bin:$PATH"
+
+# Copy and install requirements
+COPY requirements.txt /tmp/requirements.txt
+COPY webui/requirements.txt /tmp/webui_requirements.txt
+
+# Install main project dependencies
+RUN pip install --no-cache-dir --upgrade pip && \
+    pip install --no-cache-dir -r /tmp/requirements.txt
+
+# Install webui dependencies
+RUN pip install --no-cache-dir -r /tmp/webui_requirements.txt
+
+# Install production WSGI server
+RUN pip install --no-cache-dir gunicorn
+
+# Production stage
+FROM python:3.9-slim
+
+# Install runtime dependencies
+RUN apt-get update && apt-get install -y \
+    curl \
+    && rm -rf /var/lib/apt/lists/*
+
+# Create non-root user
+RUN useradd -m -u 1000 user && \
+    mkdir -p /app && \
+    chown -R user:user /app
+
+# Copy virtual environment from builder stage
+COPY --from=builder /opt/venv /opt/venv
+ENV PATH="/opt/venv/bin:$PATH"
+
+# Set working directory
+WORKDIR /app
+
+# Copy application code
+COPY --chown=user:user . /app
+
+# Make startup script executable
+RUN chmod +x /app/webui/docker_start.sh
+
+# Switch to non-root user
+USER user
+
+# Expose the Flask app port
+EXPOSE 7860
+
+# Add health check
+HEALTHCHECK --interval=30s --timeout=10s --start-period=5s --retries=3 \
+    CMD curl -f http://localhost:7860/ || exit 1
+
+# Set environment variables
+ENV PYTHONPATH=/app
+ENV FLASK_APP=webui/app.py
+ENV FLASK_ENV=production
+ENV PORT=7860
+ENV HF_HOME=/home/user/.cache/huggingface
+ENV HUGGINGFACE_HUB_CACHE=/home/user/.cache/huggingface
+ENV HF_HUB_DISABLE_TELEMETRY=1
+ENV PIP_NO_CACHE_DIR=1
+
+# Run the Flask app directly
+CMD ["python", "webui/app.py"]