Update Dockerfile

Dockerfile

@@ -1,20 +1,50 @@
 FROM python:3.10-slim
 
+# Set working directory
 WORKDIR /app
 
+# Install system dependencies
 RUN apt-get update && apt-get install -y --no-install-recommends \
     gcc \
+    g++ \
+    git \
+    curl \
     && rm -rf /var/lib/apt/lists/*
 
+# Copy requirements first for better Docker layer caching
 COPY requirements.txt .
-RUN pip install --no-cache-dir --upgrade -r requirements.txt
 
+# Install Python dependencies
+RUN pip install --no-cache-dir --upgrade pip && \
+    pip install --no-cache-dir -r requirements.txt
+
+# Copy application code
 COPY . .
 
+# Create non-root user
 RUN useradd -m -u 1000 user
-RUN mkdir -p /data/.huggingface && chown -R user:user /data
+
+# Create necessary directories with proper permissions
+RUN mkdir -p /data/.huggingface /app/data /app/results && \
+    chown -R user:user /data /app
+
+# Switch to non-root user
 USER user
-ENV HOME=/home/user PATH=/home/user/.local/bin:$PATH
-ENV HF_HOME=/data/.huggingface
 
-CMD ["gunicorn", "--worker-class", "sync", "--workers", "1", "--bind", "0.0.0.0:7860", "app:app"]
+# Set environment variables
+ENV HOME=/home/user \
+    PATH=/home/user/.local/bin:$PATH \
+    HF_HOME=/data/.huggingface \
+    TRANSFORMERS_CACHE=/data/.huggingface \
+    PYTHONPATH=/app \
+    PYTHONUNBUFFERED=1
+
+# Expose port
+EXPOSE 7860
+
+# Health check
+HEALTHCHECK --interval=30s --timeout=30s --start-period=60s --retries=3 \
+    CMD curl -f http://localhost:7860/ || exit 1
+
+# Run the application
+CMD ["gunicorn", "--worker-class", "sync", "--workers", "1", "--bind", "0.0.0.0:7860", "--timeout", "300", "--keep-alive", "2", "app:app"]