Update Dockerfile

Dockerfile

@@ -1,37 +1,42 @@
-FROM python:3.9-slim
+# Use official Python runtime as base image
+FROM python:3.10-slim
 
+# Set working directory in container
 WORKDIR /app
 
-# Install minimal system dependencies
-RUN apt-get update && apt-get install -y --no-install-recommends \
+# Install system dependencies
+RUN apt-get update && apt-get install -y \
+    git \
+    gcc \
+    g++ \
     libgl1 \
     libglib2.0-0 \
     && rm -rf /var/lib/apt/lists/*
 
-# Copy requirements
+# Copy requirements first to leverage Docker cache
 COPY requirements.txt .
 
 # Install Python dependencies
 RUN pip install --no-cache-dir -r requirements.txt
 
-# Create cache directory
-RUN mkdir -p /app/cache
+# Copy application code
+COPY . .
 
-# Copy application
-COPY app.py .
+# Create a non-root user and switch to it
+RUN useradd -m -u 1000 appuser && chown -R appuser:appuser /app
+USER appuser
+
+# Expose port for Gradio
+EXPOSE 7860
+
+# Health check
+HEALTHCHECK --interval=30s --timeout=10s --start-period=5s --retries=3 \
+    CMD python -c "import requests; requests.get('http://localhost:7860')"
 
 # Set environment variables
 ENV PYTHONUNBUFFERED=1
 ENV GRADIO_SERVER_NAME="0.0.0.0"
 ENV GRADIO_SERVER_PORT=7860
-ENV TRANSFORMERS_CACHE=/app/cache
-ENV HUGGINGFACE_HUB_CACHE=/app/cache
-
-# Expose port
-EXPOSE 7860
-
-# Run as non-root user
-RUN useradd -m appuser && chown -R appuser:appuser /app
-USER appuser
 
+# Command to run the application
 CMD ["python", "app.py"]