FROM python:3.9

WORKDIR /code

COPY ./requirements.txt /code/requirements.txt

RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt

COPY . .

# Set the environment variables for Flask
ENV FLASK_APP=app.main
ENV FLASK_RUN_HOST=0.0.0.0
ENV FLASK_RUN_PORT=7860

# Set the Transformers cache directory within the container
ENV TRANSFORMERS_CACHE=/code/.transformers_cache

# Create the directory for Transformers cache
RUN mkdir -p /code/.transformers_cache

CMD ["flask", "run"]