Dockerfile

# Use Python 3.13.2 as base image
FROM python:3.13.2-slim

# Set working directory
WORKDIR /app

# Install system dependencies
RUN apt-get update && apt-get install -y \
    build-essential \
    && rm -rf /var/lib/apt/lists/*

# Create a directory for model caching with proper permissions
RUN mkdir -p /root/.cache/huggingface && \
    chmod -R 777 /root/.cache

# Copy requirements first to leverage Docker cache
COPY requirements.txt .

# Install Python dependencies
RUN pip install --no-cache-dir -r requirements.txt

# Copy the rest of the application
COPY . .

# Expose the port the app runs on
EXPOSE 7860

# Set environment variables
ENV FLASK_APP=main.py
ENV FLASK_ENV=production
ENV TRANSFORMERS_CACHE=/root/.cache/huggingface
ENV HF_HOME=/root/.cache/huggingface

# Command to run the application
CMD ["python", "main.py"]