# Use an official Python runtime as a base image
FROM python:3.10

# Set the working directory in the container
WORKDIR /app

# Clone the Langchain-Chatchat repository
RUN git clone https://github.com/chatchat-space/Langchain-Chatchat.git

# Change to the Langchain-Chatchat directory
WORKDIR /app/Langchain-Chatchat

# Install project dependencies
RUN pip install -r requirements.txt
RUN pip install -r requirements_api.txt
RUN pip install -r requirements_webui.txt

# Uncomment the following line if you want to use additional vector libraries like milvus/pg_vector
# RUN pip install -r requirements_extra.txt

# # Clone required models using Git LFS
# RUN apt-get update && apt-get install -y git-lfs
# RUN git lfs install
# RUN git clone https://huggingface.co/THUDM/chatglm2-6b
# RUN git clone https://huggingface.co/moka-ai/m3e-base

# Initialize knowledge base and config file
RUN python copy_config_example.py
RUN python init_database.py --recreate-vs

# Expose the required port
EXPOSE 80

# Run the project on container startup
CMD ["python", "startup.py", "-a"]