FROM ubuntu:22.04

RUN apt-get update && apt-get install -y \
    unzip wget curl python3 python3-pip

# Sao chép binary LLaMA đã build
COPY llama_bin.zip /app/llama_bin.zip
WORKDIR /app

# Giải nén binaries
RUN unzip llama_bin.zip && rm llama_bin.zip && chmod +x bin/*

# Đặt biến môi trường để tìm `libllama.so`
ENV LD_LIBRARY_PATH=/app/bin:$LD_LIBRARY_PATH

# Sao chép model
RUN mkdir -p /models
RUN wget -O /models/qwen2.5-0.5b-instruct-q5_k_m.gguf \
    https://huggingface.co/Qwen/Qwen2.5-0.5B-Instruct-GGUF/resolve/main/qwen2.5-0.5b-instruct-q5_k_m.gguf

# Mở cổng 8000 cho Hugging Face Space
EXPOSE 3000

# Chạy server với host 0.0.0.0 để có thể truy cập từ bên ngoài
CMD ["./bin/llama-server", "-m", "/models/qwen2.5-0.5b-instruct-q5_k_m.gguf", "--host", "0.0.0.0", "--port", "3000"]