# Use an official Python runtime as a parent image
FROM python:3.9

# Set the working directory in the container
WORKDIR /code

# Copy the requirements file into the container
COPY ./requirements.txt /code/requirements.txt

# Install any needed dependencies
RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt

# Install gradio
RUN pip install gradio

# Set the TRANSFORMERS_CACHE environment variable
ENV TRANSFORMERS_CACHE /transformers_cache

# Create the cache directory and adjust permissions
RUN mkdir -p $TRANSFORMERS_CACHE \
    && chmod -R 777 $TRANSFORMERS_CACHE

# Copy the rest of the application code into the container
COPY . .

# Expose the port that the FastAPI application will run on
EXPOSE 7860

# Command to run the FastAPI application with Gradio
CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]