FROM python:3.9

WORKDIR /app

# Install git, git-lfs, and build dependencies for native extensions
RUN apt-get update && \
    apt-get install -y git git-lfs build-essential cmake pkg-config libpq-dev gcc g++ && \
    apt-get clean && \
    rm -rf /var/lib/apt/lists/*

# Copy requirements file
COPY requirements.txt .

# Install dependencies with pip upgrade
RUN pip install --no-cache-dir --upgrade pip && \
    pip install --no-cache-dir --upgrade -r requirements.txt

# Download NLTK data
RUN python -c "import nltk; nltk.download('punkt'); nltk.download('wordnet'); nltk.download('stopwords')"

# Create necessary directories
RUN mkdir -p data/users data/sessions data/conversations data/feedback
RUN mkdir -p mental_health_model_artifacts/chroma_db

# Copy application files
COPY app.py .
COPY .env.example .env
COPY api_mental_health.py .

# Expose the port Hugging Face Spaces expects
EXPOSE 7860

# Command to run the application
CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]