Update Dockerfile

Dockerfile

@@ -1,30 +1,46 @@
+# Use Python 3.9 slim image as base
 FROM python:3.9-slim
 
+# Set working directory
 WORKDIR /app
 
 # Install system dependencies
 RUN apt-get update && apt-get install -y \
+    libglib2.0-0 \
+    libsm6 \
+    libxext6 \
+    libxrender-dev \
+    libgomp1 \
     libgl1-mesa-glx \
     libglib2.0-0 \
+    libgtk-3-0 \
+    libgdk-pixbuf2.0-0 \
+    wget \
     && rm -rf /var/lib/apt/lists/*
 
-# Create a directory for the Hugging Face cache and make it writable
-RUN mkdir -p /huggingface_cache && chmod 777 /huggingface_cache
-ENV TRANSFORMERS_CACHE=/huggingface_cache
-ENV HF_HOME=/huggingface_cache
-
 # Copy requirements first to leverage Docker cache
 COPY requirements.txt .
+
+# Install Python dependencies
 RUN pip install --no-cache-dir -r requirements.txt
 
 # Copy application code
 COPY app.py .
 
-# Create uploads directory
-RUN mkdir -p uploads && chmod 777 uploads
+# Create directory for model cache
+RUN mkdir -p /root/.cache
+
+# Set environment variables
+ENV FLASK_APP=app.py
+ENV FLASK_ENV=production
+ENV PYTHONUNBUFFERED=1
 
 # Expose port
 EXPOSE 5000
 
-# Run the application
-CMD ["python", "app.py"]
+# Health check
+HEALTHCHECK --interval=30s --timeout=30s --start-period=60s --retries=3 \
+    CMD curl -f http://localhost:5000/health || exit 1
+
+# Run the application with gunicorn
+CMD ["gunicorn", "--bind", "0.0.0.0:5000", "--workers", "1", "--timeout", "120", "--keep-alive", "2", "app:app"]