Update Dockerfile

Dockerfile

@@ -1,15 +1,61 @@
-FROM python:3.9-slim
+# Stage 1: Build environment
+FROM python:3.9-slim as builder
 
 WORKDIR /app
 
+# Install system dependencies
+RUN apt-get update && apt-get install -y \
+    gcc \
+    python3-dev \
+    git \
+    && rm -rf /var/lib/apt/lists/*
+
+# Install Python dependencies
 COPY requirements.txt .
-RUN pip install --no-cache-dir -r requirements.txt
+RUN pip install --user --no-cache-dir -r requirements.txt
+
+# Stage 2: Runtime environment
+FROM python:3.9-slim
+
+WORKDIR /app
+
+# Copy Python dependencies from builder
+COPY --from=builder /root/.local /root/.local
 
+# Ensure scripts in .local are usable
+ENV PATH=/root/.local/bin:$PATH
+
+# Install runtime dependencies
+RUN apt-get update && apt-get install -y \
+    libgomp1 \
+    && rm -rf /var/lib/apt/lists/*
+
+# Copy application files
 COPY . .
 
+# Create directory for database
+RUN mkdir -p /app/data && touch /app/data/news.db
+
+# Environment variables
 ENV FLASK_APP=app.py
 ENV FLASK_ENV=production
+ENV DATABASE_URL=/app/data/news.db
+ENV TRANSFORMERS_CACHE=/app/cache
+ENV TORCH_HOME=/app/cache
 
+# Create cache directory for models
+RUN mkdir -p /app/cache
+
+# Pre-download smaller versions of models (optional)
+# RUN python -c "from sentence_transformers import SentenceTransformer; SentenceTransformer('paraphrase-MiniLM-L6-v2', cache_folder='/app/cache')"
+# RUN python -c "from transformers import pipeline; pipeline('summarization', model='facebook/bart-large-cnn', cache_dir='/app/cache')"
+
+# Expose port
 EXPOSE 8080
 
-CMD ["gunicorn", "--bind", "0.0.0.0:8080", "app:app"]
+# Health check
+HEALTHCHECK --interval=30s --timeout=10s \
+  CMD curl -f http://localhost:8080/ || exit 1
+
+# Run application
+CMD ["gunicorn", "--bind", "0.0.0.0:8080", "--timeout", "120", "--workers", "2", "app:app"]