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

# Set environment variables for Hugging Face cache
ENV TRANSFORMERS_CACHE=/app/.cache
ENV HF_DATASETS_CACHE=/app/.cache

# Create the cache directory
RUN mkdir -p /app/.cache

# Set the working directory in the container
WORKDIR /app

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

# Install any needed packages specified in requirements.txt
RUN pip install --no-cache-dir -r requirements.txt

# Copy the current directory contents into the container
COPY . .

# Pre-download the model during the build process
RUN python download_model.py

# Expose port 7860 (default port for Gradio)
EXPOSE 7860

# Run the app when the container launches
CMD ["python", "app.py"]