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 Hugging Face cache directory within the container
ENV HF_HOME=/code/.huggingface

# Create the directory for Hugging Face cache and ensure proper permissions
RUN mkdir -p /code/.huggingface && chmod -R 777 /code/.huggingface

CMD ["flask", "run"]