# Use a base Python image
FROM python:3.9-slim

# Install system dependencies (e.g., nmap)
RUN apt-get update && apt-get install -y nmap && rm -rf /var/lib/apt/lists/*

# Create directory for the model
RUN mkdir -p /models

# Download the model file into the image
ADD https://huggingface.co/mav23/Pentest_AI-GGUF/resolve/main/pentest_ai.Q4_0.gguf /models/pentest_ai.Q4_0.gguf

# Set working directory
WORKDIR /app

# Copy and install dependencies
COPY requirements.txt .
RUN pip install --no-cache-dir -r requirements.txt

# Copy application files
COPY app.py .

# Expose Streamlit port
EXPOSE 8501

# Run the app
CMD ["streamlit", "run", "app.py", "--server.port=8501", "--server.address=0.0.0.0"]