Dockerfile

FROM python:3.9-slim

WORKDIR /app

# Copy only the requirements first to leverage Docker cache
COPY requirements.txt .
RUN pip install --no-cache-dir -r requirements.txt

# Copy the rest of the application
COPY wm_detector/ ./wm_detector/
COPY run.py .

# Create necessary directories
RUN mkdir -p wm_detector/static/hf_cache

# Set environment variables
ENV PYTHONPATH=/app
ENV FLASK_APP=run.py

# Expose the port the app runs on
EXPOSE 7860

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