# Use a CUDA-enabled base image
FROM nvidia/cuda:12.1.1-base-ubuntu22.04

# Install Python & pip
RUN apt update && apt install -y python3 python3-pip

# Install dependencies
COPY requirements.txt /app/requirements.txt
WORKDIR /app
RUN pip install --no-cache-dir -r requirements.txt

# Copy chatbot files
COPY app.py /app/app.py

# Expose FastAPI on port 7860
EXPOSE 7860

# Run FastAPI
CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]