# ----------------------------
# PygmyClaw Dockerfile
# ----------------------------
FROM ubuntu:22.04

ENV DEBIAN_FRONTEND=noninteractive
# Dockerfile snippet
ENV MODEL_NAME="hf.co/rahul7star/Qwen3-4B-Thinking-2509-Genius-Coder-AI-Full:Q5_K_M"
ENV OLLAMA_HOST="0.0.0.0:11434"
ENV PYTHONUNBUFFERED=1

# ----------------------------
# Install system dependencies
# ----------------------------
RUN apt-get update && apt-get install -y \
    build-essential \
    libcurl4-openssl-dev \
    libcjson-dev \
    curl \
    python3 \
    python3-pip \
    git \
    zstd \
    sudo \
    && rm -rf /var/lib/apt/lists/*

# ----------------------------
# Install Ollama
# ----------------------------
RUN curl -fsSL https://ollama.com/install.sh | sh

# ----------------------------
# Python dependencies
# ----------------------------
RUN pip3 install --upgrade pip \
    && pip3 install \
        streamlit \
        gradio==4.44.0 \
        huggingface_hub==0.23.5 \
        requests \
        redis \
        huggingface_hub \
        torch \
        torchvision \
        torchaudio

# ----------------------------
# Set working directory
# ----------------------------
WORKDIR /workspace

# ----------------------------
# Copy the PygmyClaw repo
# ----------------------------
COPY . /workspace/

# ----------------------------
# Ensure scripts are executable
# ----------------------------
RUN chmod +x /workspace/entrypoint.sh \
    && chmod +x /workspace/pygmyclaw.py \
    && chmod +x /workspace/pygmyclaw_multitool.py \
    && mkdir -p /workspace/data

# ----------------------------
# Expose UI port for Gradio / Streamlit
# ----------------------------
EXPOSE 7860

# ----------------------------
# Entrypoint
# ----------------------------
CMD ["/workspace/entrypoint.sh"]