FROM python:3.11-slim

WORKDIR /app

# Сначала копируем requirements.txt, чтобы использовать кэш
COPY requirements.txt .

# Устанавливаем зависимости
RUN pip3 install --no-cache-dir -r requirements.txt

# === НОВЫЙ БЛОК: Загрузка моделей при сборке ===
# Загрузка модели и токенизатора transformers
RUN python -c "import spacy; spacy.cli.download('ru_core_news_sm')"
RUN python -c "import nltk; nltk.download('punkt_tab', download_dir='/usr/local/share/nltk_data')"
RUN python -c "import nltk; nltk.download('stopwords')"

# --- НОВАЯ СТРОКА: Загрузка модели spaCy ---
# Убедитесь, что `ru_core_news_lg` доступна в образе при сборке

# Копируем остальные файлы и делаем новую загрузку
COPY . .

EXPOSE 8501

HEALTHCHECK CMD curl --fail http://localhost:8501/_stcore/health

ENTRYPOINT ["streamlit", "run", "app.py", "--server.port=8501", "--server.address=0.0.0.0"]