Update Dockerfile

Dockerfile

@@ -1,41 +1,42 @@
-# Use an official Python runtime as the base image
-FROM python:3.9-slim
+# Use official Python image
+FROM python:3.10-slim
 
-# Set working directory in the container
-WORKDIR /app
+# Environment variables
+ENV PYTHONUNBUFFERED=1
+ENV TRANSFORMERS_CACHE=/app/cache
+ENV UPLOAD_DIR=/app/uploads
+ENV PORT=7860
 
-# Set environment variables
-ENV PYTHONUNBUFFERED=1 \
-    PYTHONDONTWRITEBYTECODE=1 \
-    HF_HOME=/tmp/huggingface_cache \
-    UPLOAD_DIR=/tmp/uploads \
-    PORT=7860
+# Create directories
+RUN mkdir -p /app/cache /app/uploads /app/static && \
+    chmod -R 777 /app
 
 # Install system dependencies
-RUN apt-get update && apt-get install -y \
+RUN apt-get update && \
+    apt-get install -y --no-install-recommends \
     gcc \
-    libpq-dev \
-    libjpeg-dev \
-    zlib1g-dev \
-    libpng-dev \
-    && apt-get clean \
+    python3-dev \
     && rm -rf /var/lib/apt/lists/*
 
-# Copy the requirements file
+# Set working directory
+WORKDIR /app
+
+# Copy requirements first for better caching
 COPY requirements.txt .
 
-# Install Python dependencies
-RUN pip install --no-cache-dir -r requirements.txt
+# Install Pythonnnn dependencies
+RUN pip install --no-cache-dir -U pip setuptools wheel && \
+    pip install --no-cache-dir -r requirements.txt
 
-# Copy the application code
+# Copy application files
 COPY . .
 
-# Create cache and upload directories and set permissions
-RUN mkdir -p /tmp/huggingface_cache /tmp/uploads \
-    && chmod -R 777 /tmp/huggingface_cache /tmp/uploads
-
-# Expose the port the app runs on
+# Expose the port
 EXPOSE 7860
 
-# Command to run the application
-CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
+# Health check
+HEALTHCHECK --interval=30s --timeout=30s --start-period=5s --retries=3 \
+    CMD curl -f http://localhost:7860/health || exit 1
+
+# Run the application
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860", "--reload"]