# Use Python with PyTorch and CUDA support
FROM pytorch/pytorch:2.1.2-cuda12.1-cudnn8-runtime

# Set working directory
WORKDIR /app

COPY requirements.txt .

# Install additional dependencies
RUN pip install python-dotenv datasets huggingface-hub
RUN pip install -r requirements.txt
RUN pip install -U transformers

# Copy source code and data files
COPY src/ ./src/
COPY hf_provider_model_costs.csv ./

# Expose the port that Gradio uses
EXPOSE 7860

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