Dockerfile

FROM python:3.12-slim-bookworm

# Set the working directory in the container
WORKDIR /app

# Add the requirements file to the container
ADD requirements.txt .

# Install the app dependencies
RUN pip install --no-cache-dir -r requirements.txt

# Copy the source code into the container
COPY . .

# Set env variables
ENV PYTHONUNBUFFERED=1

RUN --mount=type=secret,id=LLAMA_API_KEY,mode=0444,required=true
RUN --mount=type=secret,id=IMAGE_GEN_API_KEY,mode=0444,required=true

# Command to run the app on container startup
CMD ["python", "-u", "/app/gradio_ui.py", "--host", "0.0.0.0", "--port", "7860"]