FROM python:3.10

WORKDIR /code

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

# Copy code
COPY ./main.py /code/main.py

# Set environment variables
ENV AI_MODEL_ID="TheBloke/Mistral-7B-Instruct-v0.2-GPTQ"
ENV PYTHONUNBUFFERED=1

# Expose port
EXPOSE 7860

# Command to run the application
CMD ["python", "main.py"]